123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500 |
- package Task
- import (
- "ColdVerify_server/conf"
- "ColdVerify_server/lib"
- "ColdVerify_server/logs"
- "encoding/json"
- "fmt"
- "github.com/astaxie/beego/cache"
- "github.com/beego/beego/v2/adapter/orm"
- orm2 "github.com/beego/beego/v2/client/orm"
- _ "github.com/go-sql-driver/mysql"
- "time"
- )
- type Task struct {
- Id int `orm:"column(ID);size(11);auto;pk"`
- T_class int `orm:"size(200);default(0)"`
- T_task_id string `orm:"size(256);null"`
- T_uuid string `orm:"size(256);null"`
- T_name string `orm:"size(256);null"`
- T_VerifyTemplate_class string `orm:"size(256);null"`
- T_VerifyTemplate_id string `orm:"size(256);null"`
- T_deadline string `orm:"size(256);null"`
- T_scheme string `orm:"size(256);null"`
- T_collection string `orm:"size(256);null"`
- T_reporting string `orm:"size(256);null"`
- T_delivery string `orm:"size(256);null"`
- T_scheme_state int `orm:"size(2);default(0)"`
- T_collection_state int `orm:"size(2);default(0)"`
- T_reporting_state int `orm:"size(2);default(0)"`
- T_delivery_state int `orm:"size(2);default(0)"`
- T_VerifyDeviceDataStartTime string `orm:"size(256);null"`
- T_VerifyDeviceDataEndTime string `orm:"size(256);null"`
- T_BindDeviceDataStartTime string `orm:"size(256);null"`
- T_BindDeviceDataEndTime string `orm:"size(256);null"`
- T_doc1 string `orm:"type(text);null"`
- T_pdf1 string `orm:"type(text);null"`
- T_doc2 string `orm:"type(text);null"`
- T_pdf2 string `orm:"type(text);null"`
- T_doc3 string `orm:"type(text);null"`
- T_pdf3 string `orm:"type(text);null"`
- T_Show int `orm:"size(2);default(1)"`
- T_Visit int `orm:"size(200);default(0)"`
- T_State int `orm:"size(2);default(1)"`
- T_step int `orm:"size(2);default(-1)"`
- CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"`
- UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"`
- }
- type Task_R struct {
- T_task_id string
- T_name string
- T_doc1 string
- T_pdf1 string
- T_doc2 string
- T_pdf2 string
- T_doc3 string
- T_pdf3 string
- T_Show int
- T_Visit int
- T_State int
- T_scheme_state int
- T_collection_state int
- T_reporting_state int
- T_delivery_state int
- T_VerifyDeviceDataTime [2]string
- T_BindDeviceDataTime [2]string
- CreateTime string
- UpdateTime string
- }
- type Task_ struct {
- Id int
- T_class int
- T_task_id string
- T_uuid string
- T_user_name string
- T_name string
- T_VerifyTemplate_class string
- T_VerifyTemplate_id string
- T_deadline string
- T_scheme string
- T_collection string
- T_reporting string
- T_delivery string
- T_scheme_state int
- T_collection_state int
- T_reporting_state int
- T_delivery_state int
- T_scheme_name string
- T_collection_name string
- T_reporting_name string
- T_delivery_name string
- T_VerifyDeviceDataTime [2]string
- T_BindDeviceDataTime [2]string
- T_doc1 string
- T_pdf1 string
- T_doc2 string
- T_pdf2 string
- T_doc3 string
- T_pdf3 string
- T_Show int
- T_Visit int
- T_State int
- }
- func (t *Task) TableName() string {
- return "task"
- }
- var redisCache_Task cache.Cache
- func init() {
-
- orm.RegisterModel(new(Task))
- config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
- "redis_"+"Task", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
- logs.Println(config)
- var err error
- redisCache_Task, err = cache.NewCache("redis", config)
- if err != nil || redisCache_Task == nil {
- errMsg := "failed to init redis"
- logs.Println(errMsg, err)
- }
- }
- func TaskToTask_R(T Task) (T_r Task_R) {
- T_r.T_task_id = T.T_task_id
- T_r.T_name = T.T_name
- T_r.T_doc1 = T.T_doc1
- T_r.T_pdf1 = T.T_pdf1
- T_r.T_doc2 = T.T_doc2
- T_r.T_pdf2 = T.T_pdf2
- T_r.T_doc3 = T.T_doc3
- T_r.T_pdf3 = T.T_pdf3
- T_r.T_Show = T.T_Show
- T_r.T_State = T.T_State
- T_r.T_Visit = T.T_Visit
- T_r.T_scheme_state = T.T_scheme_state
- T_r.T_collection_state = T.T_collection_state
- T_r.T_reporting_state = T.T_reporting_state
- T_r.T_delivery_state = T.T_delivery_state
- T_r.T_VerifyDeviceDataTime = [2]string{T.T_VerifyDeviceDataStartTime, T.T_VerifyDeviceDataEndTime}
- T_r.T_BindDeviceDataTime = [2]string{T.T_BindDeviceDataStartTime, T.T_BindDeviceDataEndTime}
- T_r.UpdateTime = T.UpdateTime.Format("2006-01-02 15:04:05")
- T_r.CreateTime = T.CreateTime.Format("2006-01-02 15:04:05")
-
- return T_r
- }
- func TaskToTask_(T Task, userMap, adminMap map[string]string) (T_ Task_) {
- T_.Id = T.Id
- T_.T_class = T.T_class
- T_.T_task_id = T.T_task_id
- T_.T_uuid = T.T_uuid
- T_.T_user_name = userMap[T.T_uuid]
- T_.T_name = T.T_name
- T_.T_VerifyTemplate_class = T.T_VerifyTemplate_class
- T_.T_VerifyTemplate_id = T.T_VerifyTemplate_id
- T_.T_deadline = T.T_deadline
- T_.T_scheme = T.T_scheme
- T_.T_collection = T.T_collection
- T_.T_reporting = T.T_reporting
- T_.T_delivery = T.T_delivery
- T_.T_scheme_state = T.T_scheme_state
- T_.T_collection_state = T.T_collection_state
- T_.T_reporting_state = T.T_reporting_state
- T_.T_delivery_state = T.T_delivery_state
- T_.T_scheme_name = adminMap[T.T_scheme]
- T_.T_collection_name = adminMap[T.T_collection]
- T_.T_reporting_name = adminMap[T.T_reporting]
- T_.T_delivery_name = adminMap[T.T_delivery]
- T_.T_VerifyDeviceDataTime = [2]string{T.T_VerifyDeviceDataStartTime, T.T_VerifyDeviceDataEndTime}
- T_.T_BindDeviceDataTime = [2]string{T.T_BindDeviceDataStartTime, T.T_BindDeviceDataEndTime}
- T_.T_doc1 = T.T_doc1
- T_.T_pdf1 = T.T_pdf1
- T_.T_doc2 = T.T_doc2
- T_.T_pdf2 = T.T_pdf2
- T_.T_doc3 = T.T_doc3
- T_.T_pdf3 = T.T_pdf3
- T_.T_Show = T.T_Show
- T_.T_Visit = T.T_Visit
- T_.T_State = T.T_State
-
- return T_
- }
- func Redis_Task_Set(key string, r Task) (err error) {
-
- str, err := json.Marshal(r)
- if err != nil {
- logs.Error(lib.FuncName(), err)
- return
- }
- err = redisCache_Task.Put(key, str, 24*time.Hour)
- if err != nil {
- logs.Println("set key:", key, ",value:", str, err)
- }
- return
- }
- func Redis_Task_Get(key string) (r Task, is bool) {
- if redisCache_Task.IsExist(key) {
- logs.Println("找到key:", key)
- v := redisCache_Task.Get(key)
- json.Unmarshal(v.([]byte), &r)
- return r, true
- }
- logs.Println("没有 找到key:", key)
- return Task{}, false
- }
- func Redis_Task_DelK(key string) (err error) {
- err = redisCache_Task.Delete(key)
- return
- }
- func Read_Task_ById(id int) (r Task, is bool) {
- o := orm.NewOrm()
- r = Task{Id: id}
- err := o.Read(&r)
- if err != nil {
- logs.Error(lib.FuncName(), err)
- return r, false
- }
- return r, true
- }
- func Read_Task(T_task_id string) (r Task, is bool) {
- if r, is = Redis_Task_Get(T_task_id); is == true {
- return r, true
- }
- o := orm.NewOrm()
- qs := o.QueryTable(new(Task))
-
- err := qs.Filter("T_task_id", T_task_id).Filter("T_State", 1).One(&r)
- if err != nil {
- return r, false
- }
- Redis_Task_Set(T_task_id, r)
- return r, true
- }
- func Add_Task(r Task) (string, bool) {
- o := orm.NewOrm()
-
- rand_x := 0
- for true {
- r.T_task_id = lib.GetRandstring(12, "abcdefghijklmnopqrstuvwxyz0123456789", int64(rand_x))
- err := o.Read(&r, "T_task_id")
- if err != nil {
- break
- }
- rand_x += 1
- }
- _, err := o.Insert(&r)
- if err != nil {
- logs.Error(lib.FuncName(), err)
- return "", false
- }
- if !CREATE_TaskData(r.T_task_id) {
- return "", false
- }
- Redis_Task_Set(r.T_task_id, r)
- return r.T_task_id, true
- }
- func Add_Task_Tool(r Task) (string, bool) {
- o := orm.NewOrm()
- _, err := o.Insert(&r)
- if err != nil {
- logs.Error(lib.FuncName(), err)
- return "", false
- }
- return r.T_task_id, true
- }
- func Delete_Task(v Task) bool {
- o := orm.NewOrm()
- if num, err := o.Delete(&v); err == nil {
- logs.Println("Number of records deleted in database:", num)
- } else {
- logs.Error(lib.FuncName(), err)
- return false
- }
- Redis_Task_DelK(v.T_task_id)
- return true
- }
- func Delete_Task_(v Task) bool {
- o := orm.NewOrm()
- v.T_State = 0
- if num, err := o.Update(&v, "T_State"); err == nil {
- logs.Println("Number of records updated in database:", num)
- } else {
- logs.Error(lib.FuncName(), err)
- return false
- }
- Redis_Task_DelK(v.T_task_id)
- return true
- }
- func Update_Task(m Task, cols ...string) bool {
- o := orm.NewOrm()
- if num, err := o.Update(&m, cols...); err == nil {
- logs.Println("Number of records updated in database:", num)
- Redis_Task_Set(m.T_task_id, m)
- return true
- } else {
- logs.Error(lib.FuncName(), err)
- }
- return false
- }
- func Add_Task_Visit(m Task) bool {
- o := orm.NewOrm()
- m.T_Visit += 1
- if num, err := o.Update(&m, "T_Visit"); err == nil {
- logs.Println("Number of records updated in database:", num)
- Redis_Task_Set(m.T_task_id, m)
- return true
- } else {
- logs.Error(lib.FuncName(), err)
- }
- return false
- }
- func Read_UserTask_List(T_uuid string, T_name string, userMap, adminMap map[string]string, page int, page_z int) ([]Task_, int) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(Task))
- var r []Task
- var offset int64
- if page <= 1 {
- offset = 0
- } else {
- offset = int64((page - 1) * page_z)
- }
- cond := orm.NewCondition()
-
- cond1 := cond.And("T_name__icontains", T_name).And("T_Show", 1).And("T_State", 1)
- if len(T_uuid) > 0 {
- cond1 = cond1.And("T_uuid", T_uuid)
- }
- qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&r)
- cnt, _ := qs.SetCond((*orm2.Condition)(cond1)).Count()
-
- var TaskList []Task_
- for _, v := range r {
- TaskList = append(TaskList, TaskToTask_(v, userMap, adminMap))
- }
- return TaskList, int(cnt)
- }
- func Read_Task_List(T_uuid, T_admin string, T_name string, T_company_list []string, userMap, adminMap map[string]string, page int, page_z int) ([]Task_, int) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(Task))
- var r []Task
- var offset int64
- if page <= 1 {
- offset = 0
- } else {
- offset = int64((page - 1) * page_z)
- }
- cond := orm.NewCondition()
-
- cond1 := cond.AndCond(cond.Or("T_name__icontains", T_name).Or("T_task_id", T_name)).And("T_State", 1)
- if len(T_uuid) > 0 {
- cond1 = cond1.And("T_uuid", T_uuid)
- }
- if len(T_admin) > 0 {
- cond1 = cond1.AndCond(cond.Or("T_scheme", T_admin).Or("T_collection", T_admin).
- Or("T_reporting", T_admin).Or("T_delivery", T_admin))
- }
- if len(T_company_list) > 0 {
- cond1 = cond1.And("T_uuid__in", T_company_list)
- }
- qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&r)
- cnt, _ := qs.SetCond((*orm2.Condition)(cond1)).Count()
-
- var TaskList []Task_
- for _, v := range r {
- TaskList = append(TaskList, TaskToTask_(v, userMap, adminMap))
- }
- return TaskList, int(cnt)
- }
- func Read_Task_List_Time(page int, T_sn string, T_Title string, Time_start_ string, Time_end_ string) ([]Task_R, int64) {
- o := orm.NewOrm()
-
- var r []Task
- var cnt int64
- qs := o.QueryTable(new(Task))
- var offset int64
- if page <= 1 {
- offset = 0
- } else {
- offset = int64((page - 1) * conf.Page_size)
- }
- if len(Time_start_) > 1 && len(Time_end_) > 1 {
- logs.Println(Time_start_, Time_end_)
- qs.Limit(conf.Page_size, offset).Filter("T_Ut__lte", Time_end_+" 23:59:59").Filter("T_Ut__gte", Time_start_+" 00:00:00").Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).OrderBy("-Id").Filter("T_State", 1).All(&r)
- cnt, _ = qs.Filter("T_Ut__lte", Time_end_+" 23:59:59").Filter("T_Ut__gte", Time_start_+" 00:00:00").Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).Filter("T_State", 1).Count()
- return []Task_R{}, cnt
- }
- if len(Time_start_) > 1 {
- qs.Limit(conf.Page_size, offset).Filter("T_Ut__gte", Time_start_+" 00:00:00").Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).OrderBy("-Id").Filter("T_State", 1).All(&r)
- cnt, _ = qs.Filter("T_Ut__gte", Time_start_+" 00:00:00").Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).Filter("T_State", 1).Count()
- return []Task_R{}, cnt
- } else if len(Time_end_) > 1 {
- qs.Limit(conf.Page_size, offset).Filter("T_Ut__lte", Time_end_+" 23:59:59").Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).OrderBy("-Id").Filter("T_State", 1).All(&r)
- cnt, _ = qs.Filter("T_Ut__lte", Time_end_+" 23:59:59").Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).Filter("T_State", 1).Count()
- return []Task_R{}, cnt
- }
- qs.Limit(conf.Page_size, offset).Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).OrderBy("-Id").Filter("T_State", 1).All(&r)
- cnt, _ = qs.Filter("T_Title__icontains", T_Title).Filter("T_sn__icontains", T_sn).Filter("T_State", 1).Count()
- var Task_r []Task_R
- for _, v := range r {
- Task_r = append(Task_r, TaskToTask_R(v))
- }
- return Task_r, cnt
- }
|