feat: rebind mobile

services/analysis/internal/oss/session.go

@@ -0,0 +1,53 @@
+package oss
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"lone-services/pkg/redis"
+	"lone-services/pkg/utils"
+	pkgws "lone-services/pkg/ws"
+
+	jsoniter "github.com/json-iterator/go"
+)
+
+type UploadSession struct {
+	UserId  int64  `json:"user_id"`
+	BizId   int64  `json:"biz_id"`
+	ObjId   int64  `json:"obj_id"` // 兼容旧缓存
+	DateDir string `json:"date_dir"`
+}
+
+func (s UploadSession) SkinUserId() int64 {
+	if s.BizId > utils.NumberZero {
+		return s.BizId
+	}
+	return s.ObjId
+}
+
+func SaveUploadSession(ctx context.Context, taskId string, sess UploadSession, expireSeconds int) error {
+	if expireSeconds <= 0 {
+		expireSeconds = 3600
+	}
+	raw, err := jsoniter.Marshal(sess)
+	if err != nil {
+		return err
+	}
+	return redis.Client.Set(ctx, pkgws.RedisKeyOssUpload+taskId, raw, time.Duration(expireSeconds)*time.Second).Err()
+}
+
+func LoadUploadSession(ctx context.Context, taskId string) (*UploadSession, error) {
+	val, err := redis.Client.Get(ctx, pkgws.RedisKeyOssUpload+taskId).Result()
+	if err != nil {
+		return nil, err
+	}
+	var sess UploadSession
+	if err = jsoniter.UnmarshalFromString(val, &sess); err != nil {
+		return nil, err
+	}
+	if sess.UserId < utils.NumberOne {
+		return nil, fmt.Errorf("upload session user empty")
+	}
+	return &sess, nil
+}