zhangqian
2023-10-08 d4243f398b84a8b8cdcd27c35ace72f6a30a6452
service/cache_store.go
@@ -5,6 +5,7 @@
   "apsClient/model"
   "fmt"
   "github.com/spf13/cast"
   "gorm.io/gorm"
   "sync"
   "time"
)
@@ -79,18 +80,25 @@
}
func TaskFlagSet(channel int32) {
   defaultCacheStore.Add(fmt.Sprintf(CurrentTaskCacheKey, channel), struct{}{})
   defaultCacheStore.Add(fmt.Sprintf(CurrentTaskCacheKey, channel), true)
}
func TaskFlagUnset(channel int32) {
   defaultCacheStore.Remove(fmt.Sprintf(CurrentTaskCacheKey, channel))
   defaultCacheStore.Add(fmt.Sprintf(CurrentTaskCacheKey, channel), false)
}
func TaskFlagGet(channel int32) bool {
   if _, ok := defaultCacheStore.Get(fmt.Sprintf(CurrentTaskCacheKey, channel)); ok {
   if v, ok := defaultCacheStore.Get(fmt.Sprintf(CurrentTaskCacheKey, channel)); ok {
      return v.(bool)
   }
   _, err := model.NewProceduresSearch(nil).SetStatus(model.ProcedureStatusProcessing).SetChannel(channel).First()
   if err == gorm.ErrRecordNotFound {
      defaultCacheStore.Add(fmt.Sprintf(CurrentTaskCacheKey, channel), false)
      return false
   } else {
      defaultCacheStore.Add(fmt.Sprintf(CurrentTaskCacheKey, channel), true)
      return true
   }
   return false
}
func ProgressCacheGet(channel int32) (*model.ProductionProgress, bool) {