chore: update Go modules to version 1.25.5, restructure models into pkg/models, and remove unused model files for improved organization and dependency management

pkg/models/file.go

@@ -0,0 +1,65 @@
+package models
+
+import (
+	"encoding/json"
+	"pkg/utils"
+
+	"dario.cat/mergo"
+	"github.com/redis/go-redis/v9"
+)
+
+type FileInfo struct {
+	FileSize  int64  `json:"size"`
+	MimeType  string `json:"mime_type"`
+	FileHash  string `json:"hash"`
+	ChunkSize int64  `json:"chunk_size"`
+}
+
+type FileType string
+
+const (
+	FileTypeInit   FileType = "init"
+	FileTypeUpload FileType = "already"
+)
+
+type RedisFileInfo struct {
+	FileInfo
+	FileType  FileType `json:"type"`
+	CreatedAt int64    `json:"created_at"`
+	Expire    int64    `json:"expire"` // 只有上传文件(init)的时候有这个字段
+}
+
+func GetRedisFileInfo(fileId string) (*RedisFileInfo, error) {
+	rdb, ctx := utils.GetRedisClient()
+	fileInfoUnmarshalData, err := rdb.HGet(ctx, "015:fileInfoMap", fileId).Result()
+	if err == redis.Nil {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	var fileInfoData RedisFileInfo
+	if err := json.Unmarshal([]byte(fileInfoUnmarshalData), &fileInfoData); err != nil {
+		return nil, err
+	}
+	return &fileInfoData, nil
+}
+
+func SetRedisFileInfo(fileId string, fileInfo RedisFileInfo) error {
+	rdb, ctx := utils.GetRedisClient()
+	old_fileInfo, err := GetRedisFileInfo(fileId)
+	if err != nil {
+		return err
+	}
+	if old_fileInfo != nil {
+		mergo.Merge(&fileInfo, old_fileInfo)
+	}
+	jsonData, _ := json.Marshal(fileInfo)
+	_, err = rdb.HSet(ctx, "015:fileInfoMap", fileId, string(jsonData)).Result()
+	return err
+}
+
+func GetRedisFileInfoAll() (map[string]string, error) {
+	rdb, ctx := utils.GetRedisClient()
+	return rdb.HGetAll(ctx, "015:fileInfoMap").Result()
+}

pkg/models/file_share_relational.go

@@ -0,0 +1,31 @@
+package models
+
+import (
+	"encoding/json"
+	"pkg/utils"
+
+	"github.com/redis/go-redis/v9"
+)
+
+func GetRedisFileShareRelational(fileId string) ([]string, error) {
+	rdb, ctx := utils.GetRedisClient()
+	fileShareRelationalUnmarshalData, err := rdb.HGet(ctx, "015:fileShareRelational", fileId).Result()
+	if err == redis.Nil {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	var shareIDs []string
+	if err := json.Unmarshal([]byte(fileShareRelationalUnmarshalData), &shareIDs); err != nil {
+		return nil, err
+	}
+	return shareIDs, nil
+}
+
+func SetRedisFileShareRelational(fileId string, shareIDs []string) error {
+	rdb, ctx := utils.GetRedisClient()
+	jsonData, _ := json.Marshal(shareIDs)
+	_, err := rdb.HSet(ctx, "015:fileShareRelational", fileId, string(jsonData)).Result()
+	return err
+}

pkg/models/go.mod

@@ -0,0 +1,13 @@
+module pkg/models
+
+go 1.25.5
+
+require (
+	dario.cat/mergo v1.0.2
+	github.com/redis/go-redis/v9 v9.17.2
+)
+
+require (
+	github.com/cespare/xxhash/v2 v2.3.0 // indirect
+	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
+)

pkg/models/go.sum

@@ -0,0 +1,7 @@
+dario.cat/mergo v1.0.2 h1:85+piFYR1tMbRrLcDwR18y4UKJ3aH1Tbzi24VRW1TK8=
+dario.cat/mergo v1.0.2/go.mod h1:E/hbnu0NxMFBjpMIE34DRGLWqDy0g5FuKDhCb31ngxA=
+github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
+github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
+github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
+github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
+github.com/redis/go-redis/v9 v9.17.2 h1:P2EGsA4qVIM3Pp+aPocCJ7DguDHhqrXNhVcEp4ViluI=

pkg/models/pickupcode.go

@@ -0,0 +1,28 @@
+package models
+
+import (
+	"fmt"
+	"time"
+
+	"pkg/utils"
+
+	"github.com/redis/go-redis/v9"
+)
+
+func GetRedisPickupData(pickupCode string) (string, error) {
+	rdb, ctx := utils.GetRedisClient()
+	ShareId, err := rdb.Get(ctx, fmt.Sprintf("015:pickupCode:%s", pickupCode)).Result()
+	if err == redis.Nil {
+		return "", nil
+	}
+	if err != nil {
+		return "", err
+	}
+	return ShareId, nil
+}
+
+func SetRedisPickupData(pickupCode string, shareId string) (bool, error) {
+	rdb, ctx := utils.GetRedisClient()
+	ok, err := rdb.SetNX(ctx, fmt.Sprintf("015:pickupCode:%s", pickupCode), shareId, time.Until(time.Now().Add(24*time.Hour))).Result()
+	return ok, err
+}

pkg/models/share.go

@@ -0,0 +1,67 @@
+package models
+
+import (
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"pkg/utils"
+
+	"dario.cat/mergo"
+	"github.com/redis/go-redis/v9"
+)
+
+type RedisShareInfo struct {
+	// Id          string    `json:"id"`
+	CreatedAt   int64     `json:"created_at"`
+	Owner       string    `json:"owner"`
+	Type        ShareType `json:"type"`
+	Data        string    `json:"data"` // 分享数据 文件分享为文件id 文本分享为文本内容
+	ExpireAt    int64     `json:"expire_time"`
+	ViewNum     int64     `json:"download_nums"`
+	Password    string    `json:"password"`
+	NotifyEmail []string  `json:"notify_email"`
+	FileName    string    `json:"file_name"`
+	// PickupCode  bool      `json:"pickup_code"`
+}
+
+type ShareType string
+
+const (
+	ShareTypeFile ShareType = "file"
+	ShareTypeText ShareType = "text"
+)
+
+func GetRedisShareInfo(shareId string) (*RedisShareInfo, error) {
+	rdb, ctx := utils.GetRedisClient()
+	shareInfo := rdb.Get(ctx, fmt.Sprintf("015:shareInfoMap:%s", shareId))
+	shareInfoUnmarshalData, err := shareInfo.Result()
+	ttl, _ := rdb.TTL(ctx, fmt.Sprintf("015:shareInfoMap:%s", shareId)).Result()
+	if err == redis.Nil {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	var shareInfoData RedisShareInfo
+
+	if err := json.Unmarshal([]byte(shareInfoUnmarshalData), &shareInfoData); err != nil {
+		return nil, err
+	}
+	shareInfoData.ExpireAt = time.Now().Add(ttl).Unix()
+	return &shareInfoData, nil
+}
+
+func SetRedisShareInfo(shareId string, shareInfo RedisShareInfo) error {
+	rdb, ctx := utils.GetRedisClient()
+	old_shareInfo, err := GetRedisShareInfo(shareId)
+	if err != nil {
+		return err
+	}
+	if old_shareInfo != nil {
+		mergo.Merge(&shareInfo, old_shareInfo)
+	}
+	jsonData, _ := json.Marshal(shareInfo)
+	_, err = rdb.Set(ctx, fmt.Sprintf("015:shareInfoMap:%s", shareId), string(jsonData), time.Until(time.Unix(shareInfo.ExpireAt, 0))).Result()
+	return err
+}

pkg/models/stat.go

@@ -0,0 +1,53 @@
+package models
+
+import (
+	"encoding/json"
+
+	"pkg/utils"
+
+	"dario.cat/mergo"
+	"github.com/redis/go-redis/v9"
+)
+
+// 统计数据结构
+type StatData struct {
+	FileSize    int64 `json:"file_size"`    // 文件大小
+	FileNum     int64 `json:"file_num"`     // 文件数量
+	ShareNum    int64 `json:"share_num"`    // 分享数量
+	DownloadNum int64 `json:"download_num"` // 下载数量
+}
+
+func GetRedisStat(key string) (*StatData, error) {
+	rdb, ctx := utils.GetRedisClient()
+	statUnmarshalData, err := rdb.HGet(ctx, "015:stat", key).Result()
+	if err == redis.Nil {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	var stat StatData
+	if err := json.Unmarshal([]byte(statUnmarshalData), &stat); err != nil {
+		return nil, err
+	}
+	return &stat, nil
+}
+
+func SetRedisStat(key string, stat StatData) error {
+	rdb, ctx := utils.GetRedisClient()
+	old_stat, err := GetRedisStat(key)
+	if err != nil {
+		return err
+	}
+	if old_stat != nil {
+		mergo.Merge(&stat, old_stat)
+	}
+	jsonData, _ := json.Marshal(stat)
+	_, err = rdb.HSet(ctx, "015:stat", key, string(jsonData)).Result()
+	return err
+}
+
+func GetRedisStatAll() (map[string]string, error) {
+	rdb, ctx := utils.GetRedisClient()
+	return rdb.HGetAll(ctx, "015:stat").Result()
+}

pkg/models/task.go

@@ -0,0 +1,36 @@
+package models
+
+import (
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"pkg/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
+}