// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.

package dbgen

import (
	"context"
	"database/sql"

	"git.mstar.dev/mstar/linstrom/storage-new/models"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
	"gorm.io/gorm/schema"

	"gorm.io/gen"
	"gorm.io/gen/field"

	"gorm.io/plugin/dbresolver"
)

func newCollection(db *gorm.DB, opts ...gen.DOOption) collection {
	_collection := collection{}

	_collection.collectionDo.UseDB(db, opts...)
	_collection.collectionDo.UseModel(&models.Collection{})

	tableName := _collection.collectionDo.TableName()
	_collection.ALL = field.NewAsterisk(tableName)
	_collection.Id = field.NewString(tableName, "id")
	_collection.TargetId = field.NewString(tableName, "target_id")
	_collection.TargetType = field.NewString(tableName, "target_type")

	_collection.fillFieldMap()

	return _collection
}

type collection struct {
	collectionDo

	ALL        field.Asterisk
	Id         field.String
	TargetId   field.String
	TargetType field.String

	fieldMap map[string]field.Expr
}

func (c collection) Table(newTableName string) *collection {
	c.collectionDo.UseTable(newTableName)
	return c.updateTableName(newTableName)
}

func (c collection) As(alias string) *collection {
	c.collectionDo.DO = *(c.collectionDo.As(alias).(*gen.DO))
	return c.updateTableName(alias)
}

func (c *collection) updateTableName(table string) *collection {
	c.ALL = field.NewAsterisk(table)
	c.Id = field.NewString(table, "id")
	c.TargetId = field.NewString(table, "target_id")
	c.TargetType = field.NewString(table, "target_type")

	c.fillFieldMap()

	return c
}

func (c *collection) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
	_f, ok := c.fieldMap[fieldName]
	if !ok || _f == nil {
		return nil, false
	}
	_oe, ok := _f.(field.OrderExpr)
	return _oe, ok
}

func (c *collection) fillFieldMap() {
	c.fieldMap = make(map[string]field.Expr, 3)
	c.fieldMap["id"] = c.Id
	c.fieldMap["target_id"] = c.TargetId
	c.fieldMap["target_type"] = c.TargetType
}

func (c collection) clone(db *gorm.DB) collection {
	c.collectionDo.ReplaceConnPool(db.Statement.ConnPool)
	return c
}

func (c collection) replaceDB(db *gorm.DB) collection {
	c.collectionDo.ReplaceDB(db)
	return c
}

type collectionDo struct{ gen.DO }

type ICollectionDo interface {
	gen.SubQuery
	Debug() ICollectionDo
	WithContext(ctx context.Context) ICollectionDo
	WithResult(fc func(tx gen.Dao)) gen.ResultInfo
	ReplaceDB(db *gorm.DB)
	ReadDB() ICollectionDo
	WriteDB() ICollectionDo
	As(alias string) gen.Dao
	Session(config *gorm.Session) ICollectionDo
	Columns(cols ...field.Expr) gen.Columns
	Clauses(conds ...clause.Expression) ICollectionDo
	Not(conds ...gen.Condition) ICollectionDo
	Or(conds ...gen.Condition) ICollectionDo
	Select(conds ...field.Expr) ICollectionDo
	Where(conds ...gen.Condition) ICollectionDo
	Order(conds ...field.Expr) ICollectionDo
	Distinct(cols ...field.Expr) ICollectionDo
	Omit(cols ...field.Expr) ICollectionDo
	Join(table schema.Tabler, on ...field.Expr) ICollectionDo
	LeftJoin(table schema.Tabler, on ...field.Expr) ICollectionDo
	RightJoin(table schema.Tabler, on ...field.Expr) ICollectionDo
	Group(cols ...field.Expr) ICollectionDo
	Having(conds ...gen.Condition) ICollectionDo
	Limit(limit int) ICollectionDo
	Offset(offset int) ICollectionDo
	Count() (count int64, err error)
	Scopes(funcs ...func(gen.Dao) gen.Dao) ICollectionDo
	Unscoped() ICollectionDo
	Create(values ...*models.Collection) error
	CreateInBatches(values []*models.Collection, batchSize int) error
	Save(values ...*models.Collection) error
	First() (*models.Collection, error)
	Take() (*models.Collection, error)
	Last() (*models.Collection, error)
	Find() ([]*models.Collection, error)
	FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*models.Collection, err error)
	FindInBatches(result *[]*models.Collection, batchSize int, fc func(tx gen.Dao, batch int) error) error
	Pluck(column field.Expr, dest interface{}) error
	Delete(...*models.Collection) (info gen.ResultInfo, err error)
	Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
	UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
	Updates(value interface{}) (info gen.ResultInfo, err error)
	UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
	UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
	UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
	UpdateFrom(q gen.SubQuery) gen.Dao
	Attrs(attrs ...field.AssignExpr) ICollectionDo
	Assign(attrs ...field.AssignExpr) ICollectionDo
	Joins(fields ...field.RelationField) ICollectionDo
	Preload(fields ...field.RelationField) ICollectionDo
	FirstOrInit() (*models.Collection, error)
	FirstOrCreate() (*models.Collection, error)
	FindByPage(offset int, limit int) (result []*models.Collection, count int64, err error)
	ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
	Rows() (*sql.Rows, error)
	Row() *sql.Row
	Scan(result interface{}) (err error)
	Returning(value interface{}, columns ...string) ICollectionDo
	UnderlyingDB() *gorm.DB
	schema.Tabler
}

func (c collectionDo) Debug() ICollectionDo {
	return c.withDO(c.DO.Debug())
}

func (c collectionDo) WithContext(ctx context.Context) ICollectionDo {
	return c.withDO(c.DO.WithContext(ctx))
}

func (c collectionDo) ReadDB() ICollectionDo {
	return c.Clauses(dbresolver.Read)
}

func (c collectionDo) WriteDB() ICollectionDo {
	return c.Clauses(dbresolver.Write)
}

func (c collectionDo) Session(config *gorm.Session) ICollectionDo {
	return c.withDO(c.DO.Session(config))
}

func (c collectionDo) Clauses(conds ...clause.Expression) ICollectionDo {
	return c.withDO(c.DO.Clauses(conds...))
}

func (c collectionDo) Returning(value interface{}, columns ...string) ICollectionDo {
	return c.withDO(c.DO.Returning(value, columns...))
}

func (c collectionDo) Not(conds ...gen.Condition) ICollectionDo {
	return c.withDO(c.DO.Not(conds...))
}

func (c collectionDo) Or(conds ...gen.Condition) ICollectionDo {
	return c.withDO(c.DO.Or(conds...))
}

func (c collectionDo) Select(conds ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.Select(conds...))
}

func (c collectionDo) Where(conds ...gen.Condition) ICollectionDo {
	return c.withDO(c.DO.Where(conds...))
}

func (c collectionDo) Order(conds ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.Order(conds...))
}

func (c collectionDo) Distinct(cols ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.Distinct(cols...))
}

func (c collectionDo) Omit(cols ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.Omit(cols...))
}

func (c collectionDo) Join(table schema.Tabler, on ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.Join(table, on...))
}

func (c collectionDo) LeftJoin(table schema.Tabler, on ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.LeftJoin(table, on...))
}

func (c collectionDo) RightJoin(table schema.Tabler, on ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.RightJoin(table, on...))
}

func (c collectionDo) Group(cols ...field.Expr) ICollectionDo {
	return c.withDO(c.DO.Group(cols...))
}

func (c collectionDo) Having(conds ...gen.Condition) ICollectionDo {
	return c.withDO(c.DO.Having(conds...))
}

func (c collectionDo) Limit(limit int) ICollectionDo {
	return c.withDO(c.DO.Limit(limit))
}

func (c collectionDo) Offset(offset int) ICollectionDo {
	return c.withDO(c.DO.Offset(offset))
}

func (c collectionDo) Scopes(funcs ...func(gen.Dao) gen.Dao) ICollectionDo {
	return c.withDO(c.DO.Scopes(funcs...))
}

func (c collectionDo) Unscoped() ICollectionDo {
	return c.withDO(c.DO.Unscoped())
}

func (c collectionDo) Create(values ...*models.Collection) error {
	if len(values) == 0 {
		return nil
	}
	return c.DO.Create(values)
}

func (c collectionDo) CreateInBatches(values []*models.Collection, batchSize int) error {
	return c.DO.CreateInBatches(values, batchSize)
}

// Save : !!! underlying implementation is different with GORM
// The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
func (c collectionDo) Save(values ...*models.Collection) error {
	if len(values) == 0 {
		return nil
	}
	return c.DO.Save(values)
}

func (c collectionDo) First() (*models.Collection, error) {
	if result, err := c.DO.First(); err != nil {
		return nil, err
	} else {
		return result.(*models.Collection), nil
	}
}

func (c collectionDo) Take() (*models.Collection, error) {
	if result, err := c.DO.Take(); err != nil {
		return nil, err
	} else {
		return result.(*models.Collection), nil
	}
}

func (c collectionDo) Last() (*models.Collection, error) {
	if result, err := c.DO.Last(); err != nil {
		return nil, err
	} else {
		return result.(*models.Collection), nil
	}
}

func (c collectionDo) Find() ([]*models.Collection, error) {
	result, err := c.DO.Find()
	return result.([]*models.Collection), err
}

func (c collectionDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*models.Collection, err error) {
	buf := make([]*models.Collection, 0, batchSize)
	err = c.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
		defer func() { results = append(results, buf...) }()
		return fc(tx, batch)
	})
	return results, err
}

func (c collectionDo) FindInBatches(result *[]*models.Collection, batchSize int, fc func(tx gen.Dao, batch int) error) error {
	return c.DO.FindInBatches(result, batchSize, fc)
}

func (c collectionDo) Attrs(attrs ...field.AssignExpr) ICollectionDo {
	return c.withDO(c.DO.Attrs(attrs...))
}

func (c collectionDo) Assign(attrs ...field.AssignExpr) ICollectionDo {
	return c.withDO(c.DO.Assign(attrs...))
}

func (c collectionDo) Joins(fields ...field.RelationField) ICollectionDo {
	for _, _f := range fields {
		c = *c.withDO(c.DO.Joins(_f))
	}
	return &c
}

func (c collectionDo) Preload(fields ...field.RelationField) ICollectionDo {
	for _, _f := range fields {
		c = *c.withDO(c.DO.Preload(_f))
	}
	return &c
}

func (c collectionDo) FirstOrInit() (*models.Collection, error) {
	if result, err := c.DO.FirstOrInit(); err != nil {
		return nil, err
	} else {
		return result.(*models.Collection), nil
	}
}

func (c collectionDo) FirstOrCreate() (*models.Collection, error) {
	if result, err := c.DO.FirstOrCreate(); err != nil {
		return nil, err
	} else {
		return result.(*models.Collection), nil
	}
}

func (c collectionDo) FindByPage(offset int, limit int) (result []*models.Collection, count int64, err error) {
	result, err = c.Offset(offset).Limit(limit).Find()
	if err != nil {
		return
	}

	if size := len(result); 0 < limit && 0 < size && size < limit {
		count = int64(size + offset)
		return
	}

	count, err = c.Offset(-1).Limit(-1).Count()
	return
}

func (c collectionDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
	count, err = c.Count()
	if err != nil {
		return
	}

	err = c.Offset(offset).Limit(limit).Scan(result)
	return
}

func (c collectionDo) Scan(result interface{}) (err error) {
	return c.DO.Scan(result)
}

func (c collectionDo) Delete(models ...*models.Collection) (result gen.ResultInfo, err error) {
	return c.DO.Delete(models)
}

func (c *collectionDo) withDO(do gen.Dao) *collectionDo {
	c.DO = *do.(*gen.DO)
	return c
}