mirror of
https://github.com/keven1024/015.git
synced 2026-05-26 15:13:30 +00:00
36 lines
845 B
Go
36 lines
845 B
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
"worker/internal/utils"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
func GetRedisTaskInfo(taskId string) (*map[string]any, error) {
|
|
rdb, ctx := utils.GetRedisClient()
|
|
taskInfo := rdb.Get(ctx, fmt.Sprintf("015:taskInfoMap:%s", taskId))
|
|
taskInfoUnmarshalData, err := taskInfo.Result()
|
|
if err == redis.Nil {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var taskInfoData map[string]any
|
|
|
|
if err := json.Unmarshal([]byte(taskInfoUnmarshalData), &taskInfoData); err != nil {
|
|
return nil, err
|
|
}
|
|
return &taskInfoData, nil
|
|
}
|
|
|
|
func SetRedisTaskInfo(taskId string, taskInfo map[string]any) error {
|
|
rdb, ctx := utils.GetRedisClient()
|
|
jsonData, _ := json.Marshal(taskInfo)
|
|
_, err := rdb.Set(ctx, fmt.Sprintf("015:taskInfoMap:%s", taskId), jsonData, time.Hour).Result()
|
|
return err
|
|
}
|