getui

pkg/common/db/RedisModel.go

@@ -28,6 +28,7 @@ const (
 	uidPidToken                   = "UID_PID_TOKEN_STATUS:"
 	conversationReceiveMessageOpt = "CON_RECV_MSG_OPT:"
 	getuiToken                    = "GETUI_TOKEN"
+	getuiTaskID                   = "GETUI_TASK_ID"
 	messageCache                  = "MESSAGE_CACHE:"
 	SignalCache                   = "SIGNAL_CACHE:"
 	SignalListCache               = "SIGNAL_LIST_CACHE:"
@@ -397,6 +398,15 @@ func (d *DataBases) GetGetuiToken() (string, error) {
 	return result, err
 }
 
+func (d *DataBases) SetGetuiTaskID(taskID string, expireTime int64) error {
+	return d.RDB.Set(context.Background(), getuiTaskID, taskID, time.Duration(expireTime)*time.Second).Err()
+}
+
+func (d *DataBases) GetGetuiTaskID() (string, error) {
+	result, err := d.RDB.Get(context.Background(), getuiTaskID).Result()
+	return result, err
+}
+
 func (d *DataBases) SetSendMsgStatus(status int32, operationID string) error {
 	return d.RDB.Set(context.Background(), sendMsgFailedFlag+operationID, status, time.Hour*24).Err()
 }