项目初始化
This commit is contained in:
86
internal/service/batch_service.go
Normal file
86
internal/service/batch_service.go
Normal file
@@ -0,0 +1,86 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"FileRelay/internal/bootstrap"
|
||||
"FileRelay/internal/model"
|
||||
"FileRelay/internal/storage"
|
||||
"context"
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type BatchService struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewBatchService() *BatchService {
|
||||
return &BatchService{db: bootstrap.DB}
|
||||
}
|
||||
|
||||
func (s *BatchService) GetBatchByPickupCode(code string) (*model.FileBatch, error) {
|
||||
var batch model.FileBatch
|
||||
err := s.db.Preload("FileItems").Where("pickup_code = ? AND status = ?", code, "active").First(&batch).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 检查是否过期
|
||||
if s.IsExpired(&batch) {
|
||||
s.MarkAsExpired(&batch)
|
||||
return nil, errors.New("batch expired")
|
||||
}
|
||||
|
||||
return &batch, nil
|
||||
}
|
||||
|
||||
func (s *BatchService) IsExpired(batch *model.FileBatch) bool {
|
||||
if batch.Status != "active" {
|
||||
return true
|
||||
}
|
||||
|
||||
switch batch.ExpireType {
|
||||
case "time":
|
||||
if batch.ExpireAt != nil && time.Now().After(*batch.ExpireAt) {
|
||||
return true
|
||||
}
|
||||
case "download":
|
||||
if batch.MaxDownloads > 0 && batch.DownloadCount >= batch.MaxDownloads {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (s *BatchService) MarkAsExpired(batch *model.FileBatch) error {
|
||||
return s.db.Model(batch).Update("status", "expired").Error
|
||||
}
|
||||
|
||||
func (s *BatchService) DeleteBatch(ctx context.Context, batchID uint) error {
|
||||
var batch model.FileBatch
|
||||
if err := s.db.Preload("FileItems").First(&batch, batchID).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 删除物理文件
|
||||
for _, item := range batch.FileItems {
|
||||
_ = storage.GlobalStorage.Delete(ctx, item.StoragePath)
|
||||
}
|
||||
|
||||
// 删除数据库记录 (软删除 Batch)
|
||||
return s.db.Transaction(func(tx *gorm.DB) error {
|
||||
if err := tx.Where("batch_id = ?", batch.ID).Delete(&model.FileItem{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := tx.Delete(&batch).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func (s *BatchService) IncrementDownloadCount(batchID uint) error {
|
||||
return s.db.Model(&model.FileBatch{}).Where("id = ?", batchID).
|
||||
UpdateColumn("download_count", gorm.Expr("download_count + ?", 1)).Error
|
||||
}
|
||||
83
internal/service/token_service.go
Normal file
83
internal/service/token_service.go
Normal file
@@ -0,0 +1,83 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"FileRelay/internal/bootstrap"
|
||||
"FileRelay/internal/model"
|
||||
"crypto/sha256"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type TokenService struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewTokenService() *TokenService {
|
||||
return &TokenService{db: bootstrap.DB}
|
||||
}
|
||||
|
||||
func (s *TokenService) CreateToken(name string, scope string, expireAt *time.Time) (string, *model.APIToken, error) {
|
||||
rawToken := uuid.New().String()
|
||||
hash := s.hashToken(rawToken)
|
||||
|
||||
token := &model.APIToken{
|
||||
Name: name,
|
||||
TokenHash: hash,
|
||||
Scope: scope,
|
||||
ExpireAt: expireAt,
|
||||
}
|
||||
|
||||
if err := s.db.Create(token).Error; err != nil {
|
||||
return "", nil, err
|
||||
}
|
||||
|
||||
return rawToken, token, nil
|
||||
}
|
||||
|
||||
func (s *TokenService) ValidateToken(rawToken string, requiredScope string) (*model.APIToken, error) {
|
||||
hash := s.hashToken(rawToken)
|
||||
var token model.APIToken
|
||||
if err := s.db.Where("token_hash = ? AND revoked = ?", hash, false).First(&token).Error; err != nil {
|
||||
return nil, errors.New("invalid token")
|
||||
}
|
||||
|
||||
if token.ExpireAt != nil && time.Now().After(*token.ExpireAt) {
|
||||
return nil, errors.New("token expired")
|
||||
}
|
||||
|
||||
// 检查 Scope (简单包含判断)
|
||||
// 在实际应用中可以实现更复杂的逻辑
|
||||
if requiredScope != "" && !s.checkScope(token.Scope, requiredScope) {
|
||||
return nil, errors.New("insufficient scope")
|
||||
}
|
||||
|
||||
// 更新最后使用时间
|
||||
now := time.Now()
|
||||
s.db.Model(&token).Update("last_used_at", &now)
|
||||
|
||||
return &token, nil
|
||||
}
|
||||
|
||||
func (s *TokenService) hashToken(token string) string {
|
||||
h := sha256.New()
|
||||
h.Write([]byte(token))
|
||||
return hex.EncodeToString(h.Sum(nil))
|
||||
}
|
||||
|
||||
func (s *TokenService) checkScope(tokenScope, requiredScope string) bool {
|
||||
if requiredScope == "" {
|
||||
return true
|
||||
}
|
||||
scopes := strings.Split(tokenScope, ",")
|
||||
for _, s := range scopes {
|
||||
if strings.TrimSpace(s) == requiredScope {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
121
internal/service/upload_service.go
Normal file
121
internal/service/upload_service.go
Normal file
@@ -0,0 +1,121 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"FileRelay/internal/bootstrap"
|
||||
"FileRelay/internal/config"
|
||||
"FileRelay/internal/model"
|
||||
"FileRelay/internal/storage"
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"fmt"
|
||||
"math/big"
|
||||
"mime/multipart"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type UploadService struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewUploadService() *UploadService {
|
||||
return &UploadService{db: bootstrap.DB}
|
||||
}
|
||||
|
||||
func (s *UploadService) CreateBatch(ctx context.Context, files []*multipart.FileHeader, remark string, expireType string, expireValue interface{}) (*model.FileBatch, error) {
|
||||
// 1. 生成取件码
|
||||
pickupCode, err := s.generatePickupCode(config.GlobalConfig.Security.PickupCodeLength)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 2. 准备 Batch
|
||||
batch := &model.FileBatch{
|
||||
PickupCode: pickupCode,
|
||||
Remark: remark,
|
||||
ExpireType: expireType,
|
||||
Status: "active",
|
||||
}
|
||||
|
||||
switch expireType {
|
||||
case "time":
|
||||
if days, ok := expireValue.(int); ok {
|
||||
expireAt := time.Now().Add(time.Duration(days) * 24 * time.Hour)
|
||||
batch.ExpireAt = &expireAt
|
||||
}
|
||||
case "download":
|
||||
if max, ok := expireValue.(int); ok {
|
||||
batch.MaxDownloads = max
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 处理文件上传
|
||||
return batch, s.db.Transaction(func(tx *gorm.DB) error {
|
||||
if err := tx.Create(batch).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, fileHeader := range files {
|
||||
fileItem, err := s.processFile(ctx, tx, batch.ID, fileHeader)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
batch.FileItems = append(batch.FileItems, *fileItem)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func (s *UploadService) processFile(ctx context.Context, tx *gorm.DB, batchID uint, fileHeader *multipart.FileHeader) (*model.FileItem, error) {
|
||||
file, err := fileHeader.Open()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// 生成唯一存储路径
|
||||
ext := filepath.Ext(fileHeader.Filename)
|
||||
storagePath := fmt.Sprintf("%d/%s%s", batchID, uuid.New().String(), ext)
|
||||
|
||||
// 保存到存储层
|
||||
if err := storage.GlobalStorage.Save(ctx, storagePath, file); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 创建数据库记录
|
||||
item := &model.FileItem{
|
||||
BatchID: batchID,
|
||||
OriginalName: fileHeader.Filename,
|
||||
StoragePath: storagePath,
|
||||
Size: fileHeader.Size,
|
||||
MimeType: fileHeader.Header.Get("Content-Type"),
|
||||
}
|
||||
|
||||
if err := tx.Create(item).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return item, nil
|
||||
}
|
||||
|
||||
func (s *UploadService) generatePickupCode(length int) (string, error) {
|
||||
const charset = "0123456789"
|
||||
b := make([]byte, length)
|
||||
for i := range b {
|
||||
num, err := rand.Int(rand.Reader, big.NewInt(int64(len(charset))))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
b[i] = charset[num.Int64()]
|
||||
}
|
||||
// 检查是否冲突
|
||||
var count int64
|
||||
s.db.Model(&model.FileBatch{}).Where("pickup_code = ? AND status = ?", string(b), "active").Count(&count)
|
||||
if count > 0 {
|
||||
return s.generatePickupCode(length) // 递归生成
|
||||
}
|
||||
return string(b), nil
|
||||
}
|
||||
Reference in New Issue
Block a user