github.com/Ali-iotechsys/sqlboiler/v4@v4.0.0-20221208124957-6aec9a5f1f71/templates/test/singleton/boil_queries_test.go.tpl (about)

     1  var dbNameRand *rand.Rand
     2  
     3  {{if .NoContext -}}
     4  func MustTx(transactor boil.Transactor, err error) boil.Transactor {
     5  	if err != nil {
     6  		panic(fmt.Sprintf("Cannot create a transactor: %s", err))
     7  	}
     8  	return transactor
     9  }
    10  {{- else -}}
    11  func MustTx(transactor boil.ContextTransactor, err error) boil.ContextTransactor {
    12  	if err != nil {
    13  		panic(fmt.Sprintf("Cannot create a transactor: %s", err))
    14  	}
    15  	return transactor
    16  }
    17  {{- end}}
    18  
    19  func newFKeyDestroyer(regex *regexp.Regexp, reader io.Reader) io.Reader {
    20  	return &fKeyDestroyer{
    21  		reader: reader,
    22  		rgx:    regex,
    23  	}
    24  }
    25  
    26  type fKeyDestroyer struct {
    27  	reader io.Reader
    28  	buf    *bytes.Buffer
    29  	rgx    *regexp.Regexp
    30  }
    31  
    32  func (f *fKeyDestroyer) Read(b []byte) (int, error) {
    33  	if f.buf == nil {
    34  		all, err := io.ReadAll(f.reader)
    35  		if err != nil {
    36  			return 0, err
    37  		}
    38  
    39  		all = bytes.Replace(all, []byte{'\r', '\n'}, []byte{'\n'}, -1)
    40  		all = f.rgx.ReplaceAll(all, []byte{})
    41  		f.buf = bytes.NewBuffer(all)
    42  	}
    43  
    44  	return f.buf.Read(b)
    45  }
    46