platform/web/queries/edge.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 queries

import (
	"context"

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

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

	"gorm.io/plugin/dbresolver"

	"platform/web/models"
)

func newEdge(db *gorm.DB, opts ...gen.DOOption) edge {
	_edge := edge{}

	_edge.edgeDo.UseDB(db, opts...)
	_edge.edgeDo.UseModel(&models.Edge{})

	tableName := _edge.edgeDo.TableName()
	_edge.ALL = field.NewAsterisk(tableName)
	_edge.ID = field.NewInt32(tableName, "id")
	_edge.CreatedAt = field.NewTime(tableName, "created_at")
	_edge.UpdatedAt = field.NewTime(tableName, "updated_at")
	_edge.DeletedAt = field.NewField(tableName, "deleted_at")
	_edge.Type = field.NewInt(tableName, "type")
	_edge.Version = field.NewInt32(tableName, "version")
	_edge.Mac = field.NewString(tableName, "mac")
	_edge.IP = field.NewField(tableName, "ip")
	_edge.Port = field.NewUint16(tableName, "port")
	_edge.ISP = field.NewInt(tableName, "isp")
	_edge.AreaID = field.NewInt32(tableName, "area_id")
	_edge.Status = field.NewInt(tableName, "status")
	_edge.RTT = field.NewInt32(tableName, "rtt")
	_edge.Loss = field.NewInt32(tableName, "loss")
	_edge.Area = edgeBelongsToArea{
		db: db.Session(&gorm.Session{}),

		RelationField: field.NewRelation("Area", "models.Area"),
		Parent: struct {
			field.RelationField
		}{
			RelationField: field.NewRelation("Area.Parent", "models.Area"),
		},
	}

	_edge.fillFieldMap()

	return _edge
}

type edge struct {
	edgeDo

	ALL       field.Asterisk
	ID        field.Int32
	CreatedAt field.Time
	UpdatedAt field.Time
	DeletedAt field.Field
	Type      field.Int
	Version   field.Int32
	Mac       field.String
	IP        field.Field
	Port      field.Uint16
	ISP       field.Int
	AreaID    field.Int32
	Status    field.Int
	RTT       field.Int32
	Loss      field.Int32
	Area      edgeBelongsToArea

	fieldMap map[string]field.Expr
}

func (e edge) Table(newTableName string) *edge {
	e.edgeDo.UseTable(newTableName)
	return e.updateTableName(newTableName)
}

func (e edge) As(alias string) *edge {
	e.edgeDo.DO = *(e.edgeDo.As(alias).(*gen.DO))
	return e.updateTableName(alias)
}

func (e *edge) updateTableName(table string) *edge {
	e.ALL = field.NewAsterisk(table)
	e.ID = field.NewInt32(table, "id")
	e.CreatedAt = field.NewTime(table, "created_at")
	e.UpdatedAt = field.NewTime(table, "updated_at")
	e.DeletedAt = field.NewField(table, "deleted_at")
	e.Type = field.NewInt(table, "type")
	e.Version = field.NewInt32(table, "version")
	e.Mac = field.NewString(table, "mac")
	e.IP = field.NewField(table, "ip")
	e.Port = field.NewUint16(table, "port")
	e.ISP = field.NewInt(table, "isp")
	e.AreaID = field.NewInt32(table, "area_id")
	e.Status = field.NewInt(table, "status")
	e.RTT = field.NewInt32(table, "rtt")
	e.Loss = field.NewInt32(table, "loss")

	e.fillFieldMap()

	return e
}

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

func (e *edge) fillFieldMap() {
	e.fieldMap = make(map[string]field.Expr, 15)
	e.fieldMap["id"] = e.ID
	e.fieldMap["created_at"] = e.CreatedAt
	e.fieldMap["updated_at"] = e.UpdatedAt
	e.fieldMap["deleted_at"] = e.DeletedAt
	e.fieldMap["type"] = e.Type
	e.fieldMap["version"] = e.Version
	e.fieldMap["mac"] = e.Mac
	e.fieldMap["ip"] = e.IP
	e.fieldMap["port"] = e.Port
	e.fieldMap["isp"] = e.ISP
	e.fieldMap["area_id"] = e.AreaID
	e.fieldMap["status"] = e.Status
	e.fieldMap["rtt"] = e.RTT
	e.fieldMap["loss"] = e.Loss

}

func (e edge) clone(db *gorm.DB) edge {
	e.edgeDo.ReplaceConnPool(db.Statement.ConnPool)
	e.Area.db = db.Session(&gorm.Session{Initialized: true})
	e.Area.db.Statement.ConnPool = db.Statement.ConnPool
	return e
}

func (e edge) replaceDB(db *gorm.DB) edge {
	e.edgeDo.ReplaceDB(db)
	e.Area.db = db.Session(&gorm.Session{})
	return e
}

type edgeBelongsToArea struct {
	db *gorm.DB

	field.RelationField

	Parent struct {
		field.RelationField
	}
}

func (a edgeBelongsToArea) Where(conds ...field.Expr) *edgeBelongsToArea {
	if len(conds) == 0 {
		return &a
	}

	exprs := make([]clause.Expression, 0, len(conds))
	for _, cond := range conds {
		exprs = append(exprs, cond.BeCond().(clause.Expression))
	}
	a.db = a.db.Clauses(clause.Where{Exprs: exprs})
	return &a
}

func (a edgeBelongsToArea) WithContext(ctx context.Context) *edgeBelongsToArea {
	a.db = a.db.WithContext(ctx)
	return &a
}

func (a edgeBelongsToArea) Session(session *gorm.Session) *edgeBelongsToArea {
	a.db = a.db.Session(session)
	return &a
}

func (a edgeBelongsToArea) Model(m *models.Edge) *edgeBelongsToAreaTx {
	return &edgeBelongsToAreaTx{a.db.Model(m).Association(a.Name())}
}

func (a edgeBelongsToArea) Unscoped() *edgeBelongsToArea {
	a.db = a.db.Unscoped()
	return &a
}

type edgeBelongsToAreaTx struct{ tx *gorm.Association }

func (a edgeBelongsToAreaTx) Find() (result *models.Area, err error) {
	return result, a.tx.Find(&result)
}

func (a edgeBelongsToAreaTx) Append(values ...*models.Area) (err error) {
	targetValues := make([]interface{}, len(values))
	for i, v := range values {
		targetValues[i] = v
	}
	return a.tx.Append(targetValues...)
}

func (a edgeBelongsToAreaTx) Replace(values ...*models.Area) (err error) {
	targetValues := make([]interface{}, len(values))
	for i, v := range values {
		targetValues[i] = v
	}
	return a.tx.Replace(targetValues...)
}

func (a edgeBelongsToAreaTx) Delete(values ...*models.Area) (err error) {
	targetValues := make([]interface{}, len(values))
	for i, v := range values {
		targetValues[i] = v
	}
	return a.tx.Delete(targetValues...)
}

func (a edgeBelongsToAreaTx) Clear() error {
	return a.tx.Clear()
}

func (a edgeBelongsToAreaTx) Count() int64 {
	return a.tx.Count()
}

func (a edgeBelongsToAreaTx) Unscoped() *edgeBelongsToAreaTx {
	a.tx = a.tx.Unscoped()
	return &a
}

type edgeDo struct{ gen.DO }

func (e edgeDo) Debug() *edgeDo {
	return e.withDO(e.DO.Debug())
}

func (e edgeDo) WithContext(ctx context.Context) *edgeDo {
	return e.withDO(e.DO.WithContext(ctx))
}

func (e edgeDo) ReadDB() *edgeDo {
	return e.Clauses(dbresolver.Read)
}

func (e edgeDo) WriteDB() *edgeDo {
	return e.Clauses(dbresolver.Write)
}

func (e edgeDo) Session(config *gorm.Session) *edgeDo {
	return e.withDO(e.DO.Session(config))
}

func (e edgeDo) Clauses(conds ...clause.Expression) *edgeDo {
	return e.withDO(e.DO.Clauses(conds...))
}

func (e edgeDo) Returning(value interface{}, columns ...string) *edgeDo {
	return e.withDO(e.DO.Returning(value, columns...))
}

func (e edgeDo) Not(conds ...gen.Condition) *edgeDo {
	return e.withDO(e.DO.Not(conds...))
}

func (e edgeDo) Or(conds ...gen.Condition) *edgeDo {
	return e.withDO(e.DO.Or(conds...))
}

func (e edgeDo) Select(conds ...field.Expr) *edgeDo {
	return e.withDO(e.DO.Select(conds...))
}

func (e edgeDo) Where(conds ...gen.Condition) *edgeDo {
	return e.withDO(e.DO.Where(conds...))
}

func (e edgeDo) Order(conds ...field.Expr) *edgeDo {
	return e.withDO(e.DO.Order(conds...))
}

func (e edgeDo) Distinct(cols ...field.Expr) *edgeDo {
	return e.withDO(e.DO.Distinct(cols...))
}

func (e edgeDo) Omit(cols ...field.Expr) *edgeDo {
	return e.withDO(e.DO.Omit(cols...))
}

func (e edgeDo) Join(table schema.Tabler, on ...field.Expr) *edgeDo {
	return e.withDO(e.DO.Join(table, on...))
}

func (e edgeDo) LeftJoin(table schema.Tabler, on ...field.Expr) *edgeDo {
	return e.withDO(e.DO.LeftJoin(table, on...))
}

func (e edgeDo) RightJoin(table schema.Tabler, on ...field.Expr) *edgeDo {
	return e.withDO(e.DO.RightJoin(table, on...))
}

func (e edgeDo) Group(cols ...field.Expr) *edgeDo {
	return e.withDO(e.DO.Group(cols...))
}

func (e edgeDo) Having(conds ...gen.Condition) *edgeDo {
	return e.withDO(e.DO.Having(conds...))
}

func (e edgeDo) Limit(limit int) *edgeDo {
	return e.withDO(e.DO.Limit(limit))
}

func (e edgeDo) Offset(offset int) *edgeDo {
	return e.withDO(e.DO.Offset(offset))
}

func (e edgeDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *edgeDo {
	return e.withDO(e.DO.Scopes(funcs...))
}

func (e edgeDo) Unscoped() *edgeDo {
	return e.withDO(e.DO.Unscoped())
}

func (e edgeDo) Create(values ...*models.Edge) error {
	if len(values) == 0 {
		return nil
	}
	return e.DO.Create(values)
}

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

func (e edgeDo) First() (*models.Edge, error) {
	if result, err := e.DO.First(); err != nil {
		return nil, err
	} else {
		return result.(*models.Edge), nil
	}
}

func (e edgeDo) Take() (*models.Edge, error) {
	if result, err := e.DO.Take(); err != nil {
		return nil, err
	} else {
		return result.(*models.Edge), nil
	}
}

func (e edgeDo) Last() (*models.Edge, error) {
	if result, err := e.DO.Last(); err != nil {
		return nil, err
	} else {
		return result.(*models.Edge), nil
	}
}

func (e edgeDo) Find() ([]*models.Edge, error) {
	result, err := e.DO.Find()
	return result.([]*models.Edge), err
}

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

func (e edgeDo) Attrs(attrs ...field.AssignExpr) *edgeDo {
	return e.withDO(e.DO.Attrs(attrs...))
}

func (e edgeDo) Assign(attrs ...field.AssignExpr) *edgeDo {
	return e.withDO(e.DO.Assign(attrs...))
}

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

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

func (e edgeDo) FirstOrInit() (*models.Edge, error) {
	if result, err := e.DO.FirstOrInit(); err != nil {
		return nil, err
	} else {
		return result.(*models.Edge), nil
	}
}

func (e edgeDo) FirstOrCreate() (*models.Edge, error) {
	if result, err := e.DO.FirstOrCreate(); err != nil {
		return nil, err
	} else {
		return result.(*models.Edge), nil
	}
}

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

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

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

func (e edgeDo) Scan(result interface{}) (err error) {
	return e.DO.Scan(result)
}

func (e edgeDo) Delete(models ...*models.Edge) (result gen.ResultInfo, err error) {
	return e.DO.Delete(models)
}

func (e *edgeDo) withDO(do gen.Dao) *edgeDo {
	e.DO = *do.(*gen.DO)
	return e
}