wss-pool/internal/model/contractMarket.go

package model

import (
	"errors"
	"gorm.io/gorm"
	"time"
	"wss-pool/internal/data"
)

const (
	AlreadyReceived int = 2 //已领取
	Unclaimed       int = 1 //未领取
	ModifyContract  int = 0 //插针
	Market          int = 2 //现货
	Contract        int = 1 //合约
)

type ContractMarket struct {
	ID          int64    `gorm:"column:id;"json:"id"`
	TradeName   string   `gorm:"column:trade_name;"  json:"name"`
	BeginTime   string   `gorm:"column:begin_time;" json:"begin_time"`
	Step        int      `gorm:"column:step;" json:"step"`
	EndTime     string   `gorm:"column:end_time;" json:"end_time"`
	MaxPrice    string   `gorm:"column:max_price" json:"max_price"`
	IsType      int      `gorm:"column:type;not null;type:tinyint(2);COMMENT:'0 插针 1 自发合约 2 自发现货 '"json:"type"`
	IsGet       int      `gorm:"column:is_get;not null;type:tinyint(2);DEFAULT:1;COMMENT:'是否完成（1 未领取 2 已领取 ） '"json:"is_get"`
	KeepDecimal int      `gorm:"column:keep_decimal; '"json:"keep_decimal"`
	Gorm        *gorm.DB `gorm:"-" json:"-"`
}

func (this *ContractMarket) TableName() string {
	return "bot_contract_market"
}

func NewContractMarket() *ContractMarket {
	ContractMarket := &ContractMarket{}
	ContractMarket.NewGorm()
	return ContractMarket
}

func (this *ContractMarket) NewGorm() *ContractMarket {
	this.Gorm = data.WebGorm.Table(this.TableName())
	return this
}

func (this *ContractMarket) WhereBegin() *ContractMarket {

	this.Gorm = this.Gorm.Table(this.TableName()).Where("begin_time >= ?", time.Now().Format("2006-01-02 15:04:05"))
	return this
}

func (this *ContractMarket) WhereID() *ContractMarket {
	if this.ID == 0 {
		this.Gorm.Error = errors.New("param is null")
		return this
	}
	this.Gorm = this.Gorm.Table(this.TableName()).Where(&ContractMarket{ID: this.ID})
	return this
}

func (this *ContractMarket) WhereName() *ContractMarket {
	if this.TradeName == "" {
		this.Gorm.Error = errors.New("param is null")
		return this
	}
	this.Gorm = this.Gorm.Table(this.TableName()).Where(&ContractMarket{TradeName: this.TradeName})
	return this
}

func (this *ContractMarket) WhereModifyContract() *ContractMarket {
	this.Gorm = this.Gorm.Table(this.TableName()).Where("`bot_contract_market`.`type` = ?", ModifyContract)
	return this
}

func (this *ContractMarket) WhereInID(ids []int64) *ContractMarket {
	if len(ids) <= 0 {
		this.Gorm.Error = errors.New("param is null")
		return this
	}
	this.Gorm = this.Gorm.Table(this.TableName()).Where("id in (?)", ids)
	return this
}

func (this *ContractMarket) WhereIsTyep() *ContractMarket {
	if this.IsType <= 0 {
		this.Gorm.Error = errors.New("param is null")
		return this
	}
	this.Gorm = this.Gorm.Table(this.TableName()).Where(&ContractMarket{IsType: this.IsType})
	return this
}

func (this *ContractMarket) WhereIsGet() *ContractMarket {
	if this.IsGet == 0 {
		this.Gorm.Error = errors.New("param is null")
		return this
	}
	this.Gorm = this.Gorm.Table(this.TableName()).Where(&ContractMarket{IsGet: this.IsGet})
	return this
}

func (this *ContractMarket) Update(record *ContractMarket) *ContractMarket {
	this.Gorm = this.Gorm.Updates(record)
	return this
}

func (this *ContractMarket) Create() *ContractMarket {
	this.Gorm = this.Gorm.Create(&this)
	return this
}
func (this *ContractMarket) Joins(sql string) *ContractMarket {
	this.Gorm = this.Gorm.Joins(sql)
	return this
}

func (this *ContractMarket) Limit(limit int) *ContractMarket {
	this.Gorm = this.Gorm.Limit(limit)
	return this
}
func (this *ContractMarket) Offset(Offset int) *ContractMarket {
	this.Gorm = this.Gorm.Offset(Offset)
	return this
}
func (this *ContractMarket) First() *ContractMarket {
	Accounts := ContractMarket{}
	this.Gorm = this.Gorm.First(&Accounts)
	return &Accounts
}

func (this *ContractMarket) Assign(ContractMarket *ContractMarket) *ContractMarket {
	this.Gorm = this.Gorm.Assign(ContractMarket)
	return this
}

func (this *ContractMarket) FirstOrCreate() *ContractMarket {
	this.Gorm = this.Gorm.FirstOrCreate(&this)
	return this
}

func (this *ContractMarket) Order(value interface{}) *ContractMarket {
	this.Gorm = this.Gorm.Order(value)
	return this
}

func (this *ContractMarket) Count() int64 {
	var num int64
	this.Gorm = this.Gorm.Count(&num)
	return num
}
func (this *ContractMarket) Select(column string) *ContractMarket {
	this.Gorm = this.Gorm.Select(column)
	return this
}

func (this *ContractMarket) WhereTime(LoginAt string) *ContractMarket {
	if LoginAt == "" {
		this.Gorm.Error = errors.New("date require ")
		return this
	}
	this.Gorm = this.Gorm.Table(this.TableName()).Where("login_at = ?", LoginAt)
	return this
}

func (this *ContractMarket) Pluck(column string, value interface{}) *ContractMarket {
	this.Gorm = this.Gorm.Pluck(column, value)
	return this
}

func (this *ContractMarket) Find(list *[]ContractMarket) *ContractMarket {
	this.Gorm = this.Gorm.Find(&list)
	return this
}

func (this *ContractMarket) List() (result []ContractMarket) {
	this.IsGet = Unclaimed
	this.WhereIsTyep().WhereIsGet().WhereBegin().WhereName().Order("id asc").Find(&result)
	return result
}

func (this *ContractMarket) ListModifyContract() (result []ContractMarket) {
	this.IsGet = Unclaimed
	this.Select("l.keep_decimal,bot_contract_market.*").WhereModifyContract().Joins("inner join bot_contract_list  l on l.trade_name = bot_contract_market.trade_name").WhereIsGet().Order("id asc").Find(&result)
	return result
}

func (this *ContractMarket) UpdateIsGet(ids []int64) error {
	this.IsGet = Unclaimed
	return this.WhereInID(ids).WhereIsGet().Update(&ContractMarket{IsGet: AlreadyReceived}).Gorm.Error
}

func (this *ContractMarket) UpdateIsGetOne() error {
	this.IsGet = Unclaimed
	return this.WhereID().WhereIsGet().Update(&ContractMarket{IsGet: AlreadyReceived}).Gorm.Error
}