zdravko/internal/models/query/workers.gen.go

// 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 query

import (
	"context"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
	"gorm.io/gorm/schema"

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

	"gorm.io/plugin/dbresolver"

	"code.tjo.space/mentos1386/zdravko/internal/models"
)

func newWorker(db *gorm.DB, opts ...gen.DOOption) worker {
	_worker := worker{}

	_worker.workerDo.UseDB(db, opts...)
	_worker.workerDo.UseModel(&models.Worker{})

	tableName := _worker.workerDo.TableName()
	_worker.ALL = field.NewAsterisk(tableName)
	_worker.ID = field.NewUint(tableName, "id")
	_worker.CreatedAt = field.NewTime(tableName, "created_at")
	_worker.UpdatedAt = field.NewTime(tableName, "updated_at")
	_worker.DeletedAt = field.NewField(tableName, "deleted_at")
	_worker.Name = field.NewString(tableName, "name")
	_worker.Slug = field.NewString(tableName, "slug")
	_worker.Group = field.NewString(tableName, "group")
	_worker.Status = field.NewString(tableName, "status")

	_worker.fillFieldMap()

	return _worker
}

type worker struct {
	workerDo workerDo

	ALL       field.Asterisk
	ID        field.Uint
	CreatedAt field.Time
	UpdatedAt field.Time
	DeletedAt field.Field
	Name      field.String
	Slug      field.String
	Group     field.String
	Status    field.String

	fieldMap map[string]field.Expr
}

func (w worker) Table(newTableName string) *worker {
	w.workerDo.UseTable(newTableName)
	return w.updateTableName(newTableName)
}

func (w worker) As(alias string) *worker {
	w.workerDo.DO = *(w.workerDo.As(alias).(*gen.DO))
	return w.updateTableName(alias)
}

func (w *worker) updateTableName(table string) *worker {
	w.ALL = field.NewAsterisk(table)
	w.ID = field.NewUint(table, "id")
	w.CreatedAt = field.NewTime(table, "created_at")
	w.UpdatedAt = field.NewTime(table, "updated_at")
	w.DeletedAt = field.NewField(table, "deleted_at")
	w.Name = field.NewString(table, "name")
	w.Slug = field.NewString(table, "slug")
	w.Group = field.NewString(table, "group")
	w.Status = field.NewString(table, "status")

	w.fillFieldMap()

	return w
}

func (w *worker) WithContext(ctx context.Context) IWorkerDo { return w.workerDo.WithContext(ctx) }

func (w worker) TableName() string { return w.workerDo.TableName() }

func (w worker) Alias() string { return w.workerDo.Alias() }

func (w worker) Columns(cols ...field.Expr) gen.Columns { return w.workerDo.Columns(cols...) }

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

func (w *worker) fillFieldMap() {
	w.fieldMap = make(map[string]field.Expr, 8)
	w.fieldMap["id"] = w.ID
	w.fieldMap["created_at"] = w.CreatedAt
	w.fieldMap["updated_at"] = w.UpdatedAt
	w.fieldMap["deleted_at"] = w.DeletedAt
	w.fieldMap["name"] = w.Name
	w.fieldMap["slug"] = w.Slug
	w.fieldMap["group"] = w.Group
	w.fieldMap["status"] = w.Status
}

func (w worker) clone(db *gorm.DB) worker {
	w.workerDo.ReplaceConnPool(db.Statement.ConnPool)
	return w
}

func (w worker) replaceDB(db *gorm.DB) worker {
	w.workerDo.ReplaceDB(db)
	return w
}

type workerDo struct{ gen.DO }

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

func (w workerDo) Debug() IWorkerDo {
	return w.withDO(w.DO.Debug())
}

func (w workerDo) WithContext(ctx context.Context) IWorkerDo {
	return w.withDO(w.DO.WithContext(ctx))
}

func (w workerDo) ReadDB() IWorkerDo {
	return w.Clauses(dbresolver.Read)
}

func (w workerDo) WriteDB() IWorkerDo {
	return w.Clauses(dbresolver.Write)
}

func (w workerDo) Session(config *gorm.Session) IWorkerDo {
	return w.withDO(w.DO.Session(config))
}

func (w workerDo) Clauses(conds ...clause.Expression) IWorkerDo {
	return w.withDO(w.DO.Clauses(conds...))
}

func (w workerDo) Returning(value interface{}, columns ...string) IWorkerDo {
	return w.withDO(w.DO.Returning(value, columns...))
}

func (w workerDo) Not(conds ...gen.Condition) IWorkerDo {
	return w.withDO(w.DO.Not(conds...))
}

func (w workerDo) Or(conds ...gen.Condition) IWorkerDo {
	return w.withDO(w.DO.Or(conds...))
}

func (w workerDo) Select(conds ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.Select(conds...))
}

func (w workerDo) Where(conds ...gen.Condition) IWorkerDo {
	return w.withDO(w.DO.Where(conds...))
}

func (w workerDo) Order(conds ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.Order(conds...))
}

func (w workerDo) Distinct(cols ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.Distinct(cols...))
}

func (w workerDo) Omit(cols ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.Omit(cols...))
}

func (w workerDo) Join(table schema.Tabler, on ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.Join(table, on...))
}

func (w workerDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.LeftJoin(table, on...))
}

func (w workerDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.RightJoin(table, on...))
}

func (w workerDo) Group(cols ...field.Expr) IWorkerDo {
	return w.withDO(w.DO.Group(cols...))
}

func (w workerDo) Having(conds ...gen.Condition) IWorkerDo {
	return w.withDO(w.DO.Having(conds...))
}

func (w workerDo) Limit(limit int) IWorkerDo {
	return w.withDO(w.DO.Limit(limit))
}

func (w workerDo) Offset(offset int) IWorkerDo {
	return w.withDO(w.DO.Offset(offset))
}

func (w workerDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkerDo {
	return w.withDO(w.DO.Scopes(funcs...))
}

func (w workerDo) Unscoped() IWorkerDo {
	return w.withDO(w.DO.Unscoped())
}

func (w workerDo) Create(values ...*models.Worker) error {
	if len(values) == 0 {
		return nil
	}
	return w.DO.Create(values)
}

func (w workerDo) CreateInBatches(values []*models.Worker, batchSize int) error {
	return w.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 (w workerDo) Save(values ...*models.Worker) error {
	if len(values) == 0 {
		return nil
	}
	return w.DO.Save(values)
}

func (w workerDo) First() (*models.Worker, error) {
	if result, err := w.DO.First(); err != nil {
		return nil, err
	} else {
		return result.(*models.Worker), nil
	}
}

func (w workerDo) Take() (*models.Worker, error) {
	if result, err := w.DO.Take(); err != nil {
		return nil, err
	} else {
		return result.(*models.Worker), nil
	}
}

func (w workerDo) Last() (*models.Worker, error) {
	if result, err := w.DO.Last(); err != nil {
		return nil, err
	} else {
		return result.(*models.Worker), nil
	}
}

func (w workerDo) Find() ([]*models.Worker, error) {
	result, err := w.DO.Find()
	return result.([]*models.Worker), err
}

func (w workerDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*models.Worker, err error) {
	buf := make([]*models.Worker, 0, batchSize)
	err = w.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 (w workerDo) FindInBatches(result *[]*models.Worker, batchSize int, fc func(tx gen.Dao, batch int) error) error {
	return w.DO.FindInBatches(result, batchSize, fc)
}

func (w workerDo) Attrs(attrs ...field.AssignExpr) IWorkerDo {
	return w.withDO(w.DO.Attrs(attrs...))
}

func (w workerDo) Assign(attrs ...field.AssignExpr) IWorkerDo {
	return w.withDO(w.DO.Assign(attrs...))
}

func (w workerDo) Joins(fields ...field.RelationField) IWorkerDo {
	for _, _f := range fields {
		w = *w.withDO(w.DO.Joins(_f))
	}
	return &w
}

func (w workerDo) Preload(fields ...field.RelationField) IWorkerDo {
	for _, _f := range fields {
		w = *w.withDO(w.DO.Preload(_f))
	}
	return &w
}

func (w workerDo) FirstOrInit() (*models.Worker, error) {
	if result, err := w.DO.FirstOrInit(); err != nil {
		return nil, err
	} else {
		return result.(*models.Worker), nil
	}
}

func (w workerDo) FirstOrCreate() (*models.Worker, error) {
	if result, err := w.DO.FirstOrCreate(); err != nil {
		return nil, err
	} else {
		return result.(*models.Worker), nil
	}
}

func (w workerDo) FindByPage(offset int, limit int) (result []*models.Worker, count int64, err error) {
	result, err = w.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 = w.Offset(-1).Limit(-1).Count()
	return
}

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

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

func (w workerDo) Scan(result interface{}) (err error) {
	return w.DO.Scan(result)
}

func (w workerDo) Delete(models ...*models.Worker) (result gen.ResultInfo, err error) {
	return w.DO.Delete(models)
}

func (w *workerDo) withDO(do gen.Dao) *workerDo {
	w.DO = *do.(*gen.DO)
	return w
}