sq

type API interface {
	// Ping 检查连通性
	Ping(ctx context.Context) error
	// Close 关闭数据库连接
	Close() error

	// Insert 插入数据
	Insert(ctx context.Context, qb QB) (result Result, err error)
	// InsertAffected 插入数据(返回影响行数)
	InsertAffected(ctx context.Context, qb QB) (affected int64, err error)
	// InsertModel 基于 Model 创建数据, 根据 Model 字段自动填充 qb.Insert
	InsertModel(ctx context.Context, ptr Model, qb QB) (err error)
	// InsertModelAffected 基于 Model 创建数据 (返回影响行数)
	InsertModelAffected(ctx context.Context, ptr Model, qb QB) (affected int64, err error)
	// QueryRow 查询单行多列 类似 sql.Row{}.Scan()
	QueryRow(ctx context.Context, qb QB, desc []interface{}) (has bool, err error)
	// Query 查询单行多列并转换为结构体
	Query(ctx context.Context, ptr Tabler, qb QB) (has bool, err error)
	// QuerySlice 查询多行并转换为结构体
	QuerySlice(ctx context.Context, slicePtr interface{}, qb QB) (err error)
	// QuerySliceScaner 查询多行多列(自定义扫描)
	QuerySliceScaner(ctx context.Context, qb QB, scaner Scaner) (err error)

	QueryRelation(ctx context.Context, ptr Relation, qb QB) (has bool, err error)
	// QueryRelationSlice 查询多条数据并转换为 Relation slice
	QueryRelationSlice(ctx context.Context, relationSlicePtr interface{}, qb QB) (err error)
	// Count SELECT count(*)
	Count(ctx context.Context, from Tabler, qb QB) (count uint64, err error)
	// Has 查询数据是否存在(单条数据是否存在不建议使用 count 而是使用 Exist)
	Has(ctx context.Context, from Tabler, qb QB) (has bool, err error)
	SumInt64(ctx context.Context, from Tabler, column Column, qb QB) (value sql.NullInt64, err error)
	SumFloat64(ctx context.Context, from Tabler, column Column, qb QB) (value sql.NullFloat64, err error)

	// Update 更新
	Update(ctx context.Context, from Tabler, qb QB) (err error)
	// UpdateAffected 更新(返回影响行数)
	UpdateAffected(ctx context.Context, from Tabler, qb QB) (affected int64, err error)
	// ClearTestData 删除测试数据库的数据，只能运行在 test_ 为前缀的数据库中
	ClearTestData(ctx context.Context, form Tabler, qb QB) (err error)

	// HardDelete 硬删除（不可恢复）
	HardDelete(ctx context.Context, form Tabler, qb QB) (err error)
	// HardDeleteAffected 硬删除（不可恢复）(返回影响行数)
	HardDeleteAffected(ctx context.Context, form Tabler, qb QB) (affected int64, err error)
	// SoftDelete 软删除（可恢复）
	SoftDelete(ctx context.Context, form Tabler, qb QB) (err error)
	// SoftDeleteAffected 软删除（可恢复）(返回影响行数)
	SoftDeleteAffected(ctx context.Context, form Tabler, qb QB) (affected int64, err error)

	// ExecQB 执行QB
	ExecQB(ctx context.Context, qb QB, statement Statement) (result Result, err error)
	// ExecQBAffected 执行QB(返回影响行数)
	ExecQBAffected(ctx context.Context, qb QB, statement Statement) (affected int64, err error)
	// Exec 执行
	Exec(ctx context.Context, query string, values []interface{}) (result Result, err error)

	// Begin 开启事务
	Begin(ctx context.Context, level sql.IsolationLevel, handle func(tx *T) TxResult) (rollbackNoError bool, err error)
	BeginOpt(ctx context.Context, opt sql.TxOptions, handle func(tx *T) TxResult) (rollbackNoError bool, err error)
	// LastQueryCost show status like "last_query_cost"
	LastQueryCost(ctx context.Context) (lastQueryCost float64, err error)
	// PrintLastQueryCost 打印 show status like "last_query_cost" 的结果
	PrintLastQueryCost(ctx context.Context)
	// PublishMessage 发布消息
	PublishMessage(ctx context.Context, queueName string, publish Publish) (message Message, err error)
	// ConsumeMessage 消费消息
	ConsumeMessage(ctx context.Context, consume Consume) error
}

type BytesIDLister interface {
	Append(data []byte)
}

type Column string

type Condition struct {
	Column Column
	OP     OP
}

type Consume struct {
	QueueName       string
	HandleError     func(err error)
	HandleMessage   func(message Message) MessageResult
	NextConsumeTime func(consumeChance uint16, maxConsumeChance uint16) time.Duration
	// contains filtered or unexported fields
}

type CreateTimeUpdateTime struct {
	CreateTime time.Time `db:"create_time"`
	UpdateTime time.Time `db:"update_time"`
}

type CreatedAtUpdatedAt struct {
	CreatedAt time.Time `db:"created_at"`
	UpdatedAt time.Time `db:"updated_at"`
}

type Database struct {
	Core              *sqlx.DB
	SQLChecker        SQLChecker
	QueueTimeLocation *time.Location
}

type DeadLetterHandler interface {

	// HandleDeadLetter 将死信消息标记为已处理。
	HandleDeadLetter(ctx context.Context, id uint64, remark string) (err error)
	// RequeueDeadLetter 将死信消息重新入队以重新处理。
	RequeueDeadLetter(ctx context.Context, id uint64, publish Publish) (err error)
	// ArchiveHandledDeadLetter 将已处理过的死信消息归档以备将来分析
	ArchiveHandledDeadLetter(ctx context.Context, ago time.Duration) (cleanCount bool, err error)
}

type DeadLetterQueueMessage struct {
	QueueName     string
	ID            uint64    `db:"id" sq:"ignoreInsert"`
	BusinessID    uint64    `db:"business_id"`
	Reason        string    `db:"reason"`
	Handled       bool      `db:"handled"`
	HandledResult string    `db:"handled_result"`
	CreateTime    time.Time `db:"create_time"`
	DefaultLifeCycle
	WithoutSoftDelete
}

type DefaultLifeCycle struct {
}

type DefaultLog struct {
	// contains filtered or unexported fields
}

type DefaultSQLChecker struct {
}

type FromRaw struct {
	TableName       Raw
	SoftDeleteWhere Raw
}

type GMTCreateGMTModified struct {
	GMTCreate   time.Time `db:"gmt_create"`
	GMTModified time.Time `db:"gmt_modified"`
}

type Insert struct {
	Column Column
	Value  interface{}
}

type InsertMultiple struct {
	Column []Column
	Values [][]interface{}
}

type IntLister interface {
	Append(i int)
}

type Join struct {
	Type      JoinType
	TableName string
	On        string
}

type JoinType string

type Logger interface {
	Debug(message string, keysAndValues ...interface{})
	Info(message string, keysAndValues ...interface{})
	Warn(message string, keysAndValues ...interface{})
	Error(message string, keysAndValues ...interface{})
	Sync() error
}

type Message struct {
	QueueName        string
	ID               uint64    `db:"id" sq:"ignoreInsert"`
	BusinessID       uint64    `db:"business_id"`
	NextConsumeTime  time.Time `db:"next_consume_time"`
	ConsumeChance    uint16    `db:"consume_chance"`
	MaxConsumeChance uint16    `db:"max_consume_chance"`
	UpdateID         string    `db:"update_id"`
	Priority         uint8     `db:"priority"`
	CreateTime       time.Time `db:"create_time"`

	DefaultLifeCycle
	WithoutSoftDelete
	// contains filtered or unexported fields
}

type MessageResult struct {
	// contains filtered or unexported fields
}

type Model interface {
	Tabler
	BeforeInsert() error
	AfterInsert(result Result) error
	BeforeUpdate() error
	AfterUpdate() error
}

type MysqlDataSource struct {
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Host     string `yaml:"host"`
	Port     string `yaml:"port"`
	DB       string `yaml:"db"`
	// DefaultQuery
	// 	map[string]string{
	// 	"charset": "utf8",
	// 	"parseTime": "True",
	// 	"loc": "Local",
	// }
	Query map[string]string `yaml:"query"`
}

type OP struct {
	Query       string
	Values      []interface{}
	Symbol      string
	Placeholder string
	Multiple    []OP
	OrGroup     []Condition
	Ignore      bool
}

type OrderBy struct {
	Column Column
	Type   orderByType
}

type Publish struct {
	BusinessID       uint64
	NextConsumeTime  time.Duration
	MaxConsumeChance uint16
	Priority         uint8 `default:"100"`
}

type QB struct {
	Select    []Column
	SelectRaw []Raw

	From Tabler

	FromRaw FromRaw

	DisableSoftDelete bool

	UnionTable UnionTable

	Index string

	Set []Update
	// UPDATE IGNORE
	UseUpdateIgnore bool
	Insert          Values
	InsertMultiple  InsertMultiple
	// INSERT IGNORE INTO
	UseInsertIgnoreInto bool

	Where           []Condition
	WhereOR         [][]Condition
	WhereRaw        Raw
	WhereAllowEmpty bool

	OrderBy    []OrderBy
	OrderByRaw Raw

	GroupBy    []Column
	GroupByRaw Raw

	Having    []Condition
	HavingRaw Raw

	Limit uint64

	Offset uint64

	Lock SelectLock

	Join []Join
	Raw  Raw

	Debug bool

	PrintSQL bool
	Explain  bool
	RunTime  bool

	LastQueryCost bool

	Review     string
	Reviews    []string
	SQLChecker SQLChecker
	// contains filtered or unexported fields
}

type Raw struct {
	Query  string
	Values []interface{}
}

type Relation interface {
	TableName() string
	SoftDeleteWhere() Raw
	RelationJoin() []Join
}

type Result struct {
	// contains filtered or unexported fields
}

type SQLChecker interface {
	Check(checkSQL []string, execSQL string) (pass bool, refs string, err error)
	TrackFail(debugID uint64, err error, reviews []string, query string, refs string)
}

type Scaner func(rows *sqlx.Rows) error

type SelectLock string

type SoftDeleteTime struct{}

type SoftDeletedAt struct{}

type SoftIsDeleted struct{}

type Statement string

type Storager interface {
	// contains filtered or unexported methods
}

type StoragerCore interface {
	sqlx.Queryer
	sqlx.QueryerContext
	sqlx.Execer
	sqlx.ExecerContext
	sqlx.Preparer
	sqlx.PreparerContext
	SelectContext(ctx context.Context, dest interface{}, query string, args ...interface{}) error
}

type StringLister interface {
	Append(s string)
}

type T struct {
	Core *sqlx.Tx
	// contains filtered or unexported fields
}

type Tabler interface {
	TableName() string
	SoftDeleteWhere() Raw
	SoftDeleteSet() Raw
}

type Tag struct {
	Value string
}

type TxResult struct {
	// contains filtered or unexported fields
}

type UintLister interface {
	Append(i uint)
}

type UnionTable struct {
	Tables   []QB
	UnionAll bool
}

type Update struct {
	Column Column
	Value  interface{}
	Raw    Raw
}

type Values []Insert

type WithoutSoftDelete struct{}