datastore.go

package goquery

import (
	"io"

	"github.com/jackc/pgconn"
)

type RecordHandler func(interface{}) error

type BindParamTemplateFunction func(field string, i int) string
type SequenceTemplateFunction func(sequence string) string
type UrlTemplateFunction func(config *RdbmsConfig) string

const (
	DEST OutputFormat = iota
	JSON
	CSV
)

type DbDialect struct {
	TableExistsStmt string
	Bind            BindParamTemplateFunction
	Seq             SequenceTemplateFunction
	Url             UrlTemplateFunction
}

type QueryInput struct {
	DataSet      DataSet
	StatementKey string
	Statement    string
	Suffix       string
	BindParams   []interface{}
	StmtAppends  []interface{}
	PanicOnErr   bool
	LogSql       bool
}

type QueryOutput struct {
	OutputFormat OutputFormat
	Writer       io.Writer
	Options      OutputOptions
	rowFunction  RowFunction
}

type InsertInput struct {
	Dataset    DataSet
	Records    interface{}
	Batch      bool
	BatchSize  int
	PanicOnErr bool
}

type OutputOptions struct {
	ToCamelCase    bool
	IsArray        bool
	DateFormat     string
	OmitNull       bool
	CsvPrintHeader bool
}

type DataStore interface {
	Connection() interface{}
	NewTransaction() (Tx, error)
	Transaction(tf TransactionFunction) error
	Fetch(tx *Tx, input QueryInput, output QueryOutput, dest any) error
	FetchRows(tx *Tx, input QueryInput) (Rows, error)
	GetJSON(writer io.Writer, input QueryInput, jo OutputOptions) error
	GetCSV(input QueryInput, co OutputOptions) (string, error)
	Select(stmt ...string) *FluentSelect
	Insert(ds DataSet) *FluentInsert
	//InsertRecs(ds DataSet, recs interface{}, batch bool, batchSize int, tx *Tx) error
	InsertRecs(tx *Tx, input InsertInput) error
	//UpdateRecs(ds DataSet, recs interface{}, batch bool, batchSize int, tx *Tx) error
	Exec(tx *Tx, stmt string, params ...interface{}) error
	Execr(tx *Tx, stmt string, params ...interface{}) (ExecResult, error)
	MustExec(tx *Tx, stmt string, params ...interface{})
	MustExecr(tx *Tx, stmt string, params ...interface{}) ExecResult
	//RecordsetIterator(s Select, handler RecordHandler)
}

type Batch interface {
	Queue(stmt string, params ...interface{})
	Len() int
}

type BatchResult interface {
	Exec() (pgconn.CommandTag, error)
	Close() error
}

type ExecResult interface {
	RowsAffected() int64
}