vscode-helper-js/hasaki-gen-crud/template.go

package main

import (
	"fmt"
	"time"

	"git.ouxuan.net/hasaki-service/hasaki-sdk/hskdb"
	"git.ouxuan.net/hasaki-service/hasaki-sdk/hskgin"
	"git.ouxuan.net/hasaki-service/hasaki-sdk/hsklogger/L"
	"xorm.io/builder"
	"xorm.io/xorm"
)

type Temp3A60 struct {
	Id          int
	service     string
	QueueStatus string
}

// ----------------------------temp------------------------------------

// wherePrimaryKey 主键查询
func (TempLowHead3A60 *Temp3A60) wherePrimaryKey() (q interface{}, args []interface{}) {
	//todo 根据需求修改
	return "`id`=?", []interface{}{
		TempLowHead3A60.Id,
	}
}

// wherePrimaryKeyWithCond 主键查询
func (TempLowHead3A60 *Temp3A60) wherePrimaryKeyWithCond() builder.Cond {
	//todo 根据需求修改
	return nil
}

// whereDataWithCond 列表查询
func (TempLowHead3A60 *Temp3A60) whereDataWithCond() builder.Cond {
	//todo 根据需求修改
	return nil
}

// whereData 列表查询筛选
func (TempLowHead3A60 *Temp3A60) whereData() (q interface{}, args []interface{}) {
	//todo
	return "", nil
}

// orderData 列表查询排序
func (TempLowHead3A60 *Temp3A60) orderData() string {
	//todo
	return "`id` desc"
}

// init 数据初始化,可附加更多数据
func (TempLowHead3A60 *Temp3A60) init() {
	//todo
}

func (TempLowHead3A60 *Temp3A60) joinTable() (types, sqlx, condition, alias string) {
	//todo
	return "LEFT", "", "Temp3A60.id=Temp3A60_id", "tmp"
}

// saveAllCols 保存时是否全量保存
func (TempLowHead3A60 *Temp3A60) isSaveAllCols() bool {
	return false
}

// insertPretreatment 新增前预处理
func (TempLowHead3A60 *Temp3A60) insertPretreatment() error {
	//todo
	return nil
}

// updatePretreatment 修改前预处理
func (TempLowHead3A60 *Temp3A60) updatePretreatment() error {
	//todo
	return nil
}

// deletePretreatment 删除前预处理
func (TempLowHead3A60 *Temp3A60) deletePretreatment() error {
	//todo
	return nil
}

func (TempLowHead3A60 *Temp3A60) getXorm() *xorm.Engine {
	//todo
	return hskdb.GetXormAuto()
}

// 队列状态
const (
	TempLowHead3A60_AutoQueueWait          = 2 * time.Second
	TempLowHead3A60_AutoQueueStatusNone    = "None"
	TempLowHead3A60_AutoQueueStatusWait    = "Wait"
	TempLowHead3A60_AutoQueueStatusIng     = "Ing"
	TempLowHead3A60_AutoQueueStatusSuccess = "Success"
	TempLowHead3A60_AutoQueueStatusFail    = "Fail"
)

// 不同service对应的不同队列状态
var TempLowHead3A60_AutoQueueStatusFieldMap = map[string]string{
	"service": "queue_status",
}

// 队列事件
func (TempLowHead3A60 *Temp3A60) queueEvent() error {
	//todo
	return nil
}

// ----------------------------temp------------------------------------

// Save 保存或者新增,通过wherePrimaryKey查不到数据时,新增数据,否则修改数据,参数:isAllCols...
func (TempLowHead3A60 *Temp3A60) Save(mustCols ...string) (err error) {
	if TempLowHead3A60.service == "" {
		TempLowHead3A60.service = "none"
	}

	tmp := Temp3A60{}

	whereQuery, whereArgs := TempLowHead3A60.wherePrimaryKey()
	whereCond := TempLowHead3A60.wherePrimaryKeyWithCond()
	session := TempLowHead3A60.getXorm().NewSession()
	defer session.Close()
	if whereQuery != "" {
		session = session.Where(whereQuery, whereArgs...)
	}
	if whereCond != nil {
		session = session.Where(whereCond)
	}
	_, err = session.Get(&tmp)
	if err != nil {
		return
	}

	if tmp.Id == 0 {
		err = TempLowHead3A60.insertPretreatment()
		if err != nil {
			return
		}
		_, err = TempLowHead3A60.getXorm().Insert(TempLowHead3A60)
	} else {
		err = TempLowHead3A60.updatePretreatment()
		if err != nil {
			return
		}
		if len(mustCols) > 0 {
			_, err = TempLowHead3A60.getXorm().MustCols(mustCols...).Where(whereQuery, whereArgs...).Update(TempLowHead3A60)
		} else {
			if TempLowHead3A60.isSaveAllCols() {
				_, err = TempLowHead3A60.getXorm().AllCols().Where(whereQuery, whereArgs...).Update(TempLowHead3A60)
			} else {
				_, err = TempLowHead3A60.getXorm().Where(whereQuery, whereArgs...).Update(TempLowHead3A60)
			}
		}
	}
	return err
}

// Load 为实例载入数据,数据来源取决于wherePrimaryKey查找实例中的数据
func (TempLowHead3A60 *Temp3A60) Load() (err error) {
	if TempLowHead3A60.service == "" {
		TempLowHead3A60.service = "none"
	}

	whereQuery, whereArgs := TempLowHead3A60.wherePrimaryKey()
	whereCond := TempLowHead3A60.wherePrimaryKeyWithCond()
	session := TempLowHead3A60.getXorm().NewSession()
	defer session.Close()

	types, sqlx, condition, alias := TempLowHead3A60.joinTable()
	if sqlx != "" {
		session = session.Join(types, sqlx, condition, alias)
	}

	if whereQuery != "" {
		session = session.Where(whereQuery, whereArgs...)
	}
	if whereCond != nil {
		session = session.Where(whereCond)
	}

	_, err = session.Get(TempLowHead3A60)
	if err != nil {
		return
	}

	TempLowHead3A60.init()
	return
}

// Data 不进行分页的数据列表
func (TempLowHead3A60 *Temp3A60) Data() (data []Temp3A60) {
	if TempLowHead3A60.service == "" {
		TempLowHead3A60.service = "none"
	}

	q, args := TempLowHead3A60.whereData()

	_xorm := TempLowHead3A60.getXorm()
	session := _xorm.NewSession()
	defer func() {
		session.Close()
	}()

	types, sqlx, condition, alias := TempLowHead3A60.joinTable()
	if sqlx != "" {
		session = session.Join(types, sqlx, condition, alias)
	}

	if q != "" {
		session = session.Where(q, args...)
	}

	whereCond := TempLowHead3A60.whereDataWithCond()
	if whereCond != nil {
		session = session.Where(whereCond)
	}

	order := TempLowHead3A60.orderData()
	if order != "" {
		session = session.OrderBy(order)
	}

	session.Find(&data)
	for i := range data {
		data[i].init()
	}
	return
}

// PageData 进行分页的数据列表,page从0开始
func (TempLowHead3A60 *Temp3A60) PageData(page, pageSize int) (data []Temp3A60, count int64) {
	if TempLowHead3A60.service == "" {
		TempLowHead3A60.service = "none"
	}

	q, args := TempLowHead3A60.whereData()
	_xorm := TempLowHead3A60.getXorm()
	session := _xorm.NewSession()
	defer func() {
		session.Close()
	}()

	types, sqlx, condition, alias := TempLowHead3A60.joinTable()
	if sqlx != "" {
		session = session.Join(types, sqlx, condition, alias)
	}

	if q != "" {
		session = session.Where(q, args...)
	}

	whereCond := TempLowHead3A60.whereDataWithCond()
	if whereCond != nil {
		session = session.Where(whereCond)
	}

	order := TempLowHead3A60.orderData()
	if order != "" {
		session = session.OrderBy(order)
	}

	count, _ = session.Limit(pageSize, page*pageSize).FindAndCount(&data)
	for i := range data {
		data[i].init()
	}
	return
}

// Delete 删除数据
func (TempLowHead3A60 *Temp3A60) Delete() (err error) {
	if TempLowHead3A60.service == "" {
		TempLowHead3A60.service = "none"
	}

	whereQuery, whereArgs := TempLowHead3A60.wherePrimaryKey()
	whereCond := TempLowHead3A60.wherePrimaryKeyWithCond()
	session := TempLowHead3A60.getXorm().NewSession()
	defer session.Close()
	if whereQuery != "" {
		session = session.Where(whereQuery, whereArgs...)
	}
	if whereCond != nil {
		session = session.Where(whereCond)
	}

	err = TempLowHead3A60.deletePretreatment()
	if err != nil {
		return
	}
	_, err = session.Delete(TempLowHead3A60)
	return
}

// ----------------------------temp------------------------------------

func StartQueueTaskPrefixHigHead3A60Temp3A60() {
	serviceName := "Prefix3A60"
	f := TempLowHead3A60_AutoQueueStatusFieldMap[serviceName]
	go func() {
		defer func() {
			err := recover()
			if err != nil {
				L.Error("StartQueueTaskPrefixHigHead3A60Temp3A60 意外退出:", err)
			}
			time.Sleep(time.Second * 5)

			StartQueueTaskPrefixHigHead3A60Temp3A60()
		}()

		for {
			onec := &Temp3A60{}
			hskdb.GetXormAuto().Where(fmt.Sprintf("`%s`=?", f), TempLowHead3A60_AutoQueueStatusWait).OrderBy("rand()").Get(onec)

			session := onec.getXorm().NewSession()
			defer session.Close()
			session = session.Table(onec)
			whereQuery, whereArgs := onec.wherePrimaryKey()
			whereCond := onec.wherePrimaryKeyWithCond()
			if whereQuery != "" {
				session = session.Where(whereQuery, whereArgs...)
			}
			if whereCond != nil {
				session = session.Where(whereCond)
			}

			i, err := session.Where(fmt.Sprintf("`%s`=?", f), TempLowHead3A60_AutoQueueStatusWait).Update(map[string]interface{}{f: TempLowHead3A60_AutoQueueStatusIng})
			if err == nil && i == 1 {
				func() {
					{
						onec.service = serviceName
						err := onec.queueEvent()
						session := onec.getXorm().NewSession()
						defer session.Close()
						session = session.Table(onec)
						whereQuery, whereArgs := onec.wherePrimaryKey()
						whereCond := onec.wherePrimaryKeyWithCond()
						if whereQuery != "" {
							session = session.Where(whereQuery, whereArgs...)
						}
						if whereCond != nil {
							session = session.Where(whereCond)
						}
						if err != nil {
							session.Update(map[string]interface{}{f: TempLowHead3A60_AutoQueueStatusFail})
						} else {
							session.Update(map[string]interface{}{f: TempLowHead3A60_AutoQueueStatusSuccess})
						}
					}
				}()
			}
			time.Sleep(TempLowHead3A60_AutoQueueWait)
		}

	}()
}

func InitializationHttpCrudPrefixHigHead3A60Temp3A60(router *hskgin.GinHelper, authoritys ...func(ctx *hskgin.GinContextHelper, TempLowHead3A60 *Temp3A60)) {
	var authority = func(ctx *hskgin.GinContextHelper, TempLowHead3A60 *Temp3A60) {
	}
	if len(authoritys) > 0 {
		authority = authoritys[0]
	}

	/*
		@yapi

		@group TempLowHead3A60

		@name 获取(Prefix3A60-TempLowHead3A60)单个数据

		@path /Prefix3A60/TempLowHead3A60/get

		@markdown

		##
		```
		传入主键获取对应数据
		```
		@end
	*/
	router.Any("/Prefix3A60/TempLowHead3A60/get", func(ctx *hskgin.GinContextHelper) {
		req := &Temp3A60{}
		ctx.Bind(req)
		authority(ctx, req)
		req.service = "Prefix3A60"
		err := req.Load()
		ctx.CheckErrDisplayByError(err)
		ctx.DisplayByData(req)
	})

	/*
		@yapi

		@group TempLowHead3A60

		@name 保存(Prefix3A60-TempLowHead3A60)单个数据

		@path /Prefix3A60/TempLowHead3A60/save

		@markdown

		##
		```
		传入整体数据插入或者保存(取决于主键是否存在)
		```
		@end
	*/
	router.Any("/Prefix3A60/TempLowHead3A60/save", func(ctx *hskgin.GinContextHelper) {
		req := &Temp3A60{}
		ctx.Bind(&req)
		authority(ctx, req)
		req.service = "Prefix3A60"
		err := req.Save()
		ctx.CheckErrDisplayByError(err)
		err = req.Load()
		ctx.CheckErrDisplayByError(err)
		ctx.DisplayByData(req)
	})

	/*
		@yapi

		@group TempLowHead3A60

		@name 获取(Prefix3A60-TempLowHead3A60)无分页数据

		@path /Prefix3A60/TempLowHead3A60/data

		@markdown

		##
		```
		获取数据列表,无分页
		```
		@end
	*/
	router.Any("/Prefix3A60/TempLowHead3A60/data", func(ctx *hskgin.GinContextHelper) {
		req := &Temp3A60{}
		ctx.Bind(&req)
		authority(ctx, req)
		req.service = "Prefix3A60"
		data := req.Data()
		ctx.DisplayByData(data)
	})

	/*
		@yapi

		@group TempLowHead3A60

		@name 获取(Prefix3A60-TempLowHead3A60)分页数据

		@path /Prefix3A60/TempLowHead3A60/list

		@markdown

		##
		```
		获取数据列表,分页,page参数
		```
		@end
	*/
	router.Any("/Prefix3A60/TempLowHead3A60/list", func(ctx *hskgin.GinContextHelper) {
		req := &Temp3A60{}
		ctx.Bind(&req)
		authority(ctx, req)
		page := ctx.ParamRequired2Page(true)
		req.service = "Prefix3A60"
		data, count := req.PageData(page, 10)
		ctx.DisplayByPage(data, count)
	})

	/*
		@yapi

		@group TempLowHead3A60

		@name 删除一条(Prefix3A60-TempLowHead3A60)数据

		@path /Prefix3A60/TempLowHead3A60/delete

		@markdown

		##
		```
		传入主键参数,删除一条数据
		```
		@end
	*/
	router.Any("/Prefix3A60/TempLowHead3A60/delete", func(ctx *hskgin.GinContextHelper) {
		req := &Temp3A60{}
		ctx.Bind(&req)
		authority(ctx, req)
		req.service = "Prefix3A60"
		err := req.Delete()
		ctx.CheckErrDisplayByError(err)
		ctx.DisplayBySuccess("删除成功")
	})
}