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 } // 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() 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() }() 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() }() 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] } 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) }) 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) }) 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) }) 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) }) 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("删除成功") }) }