124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
package transfer
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"mesh-drop/internal/config"
|
|
"mesh-drop/internal/discovery"
|
|
"sync"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/wailsapp/wails/v3/pkg/application"
|
|
)
|
|
|
|
type Service struct {
|
|
config *config.Config
|
|
app *application.App
|
|
port int
|
|
|
|
// pendingRequests 存储等待用户确认的通道
|
|
// Key: TransferID, Value: *Transfer
|
|
transferList sync.Map
|
|
|
|
discoveryService *discovery.Service
|
|
|
|
// cancelMap 存储取消操作的通道
|
|
// Key: TransferID, Value: context.CancelFunc
|
|
cancelMap sync.Map
|
|
}
|
|
|
|
func NewService(config *config.Config, app *application.App, port int, discoveryService *discovery.Service) *Service {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
|
|
return &Service{
|
|
app: app,
|
|
port: port,
|
|
discoveryService: discoveryService,
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
application.RegisterEvent[application.Void]("transfer:refreshList")
|
|
}
|
|
|
|
func (s *Service) GetPort() int {
|
|
return s.port
|
|
}
|
|
|
|
func (s *Service) Start() {
|
|
r := gin.Default()
|
|
transfer := r.Group("/transfer")
|
|
{
|
|
transfer.POST("/ask", s.handleAsk)
|
|
transfer.PUT("/upload/:id", s.handleUpload)
|
|
}
|
|
|
|
go func() {
|
|
addr := fmt.Sprintf(":%d", s.port)
|
|
slog.Info("Transfer service listening", "address", addr, "component", "transfer")
|
|
if err := r.Run(addr); err != nil {
|
|
slog.Error("Transfer service error", "error", err, "component", "transfer")
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (s *Service) GetTransferList() []*Transfer {
|
|
var requests []*Transfer
|
|
s.transferList.Range(func(key, value any) bool {
|
|
requests = append(requests, value.(*Transfer))
|
|
return true
|
|
})
|
|
return requests
|
|
}
|
|
|
|
func (s *Service) GetTransfer(transferID string) (*Transfer, bool) {
|
|
val, ok := s.transferList.Load(transferID)
|
|
if !ok {
|
|
return nil, false
|
|
}
|
|
return val.(*Transfer), true
|
|
}
|
|
|
|
func (s *Service) CancelTransfer(transferID string) {
|
|
if cancel, ok := s.cancelMap.Load(transferID); ok {
|
|
cancel.(context.CancelFunc)()
|
|
s.cancelMap.Delete(transferID)
|
|
t, ok := s.GetTransfer(transferID)
|
|
if ok {
|
|
t.Status = TransferStatusCanceled
|
|
s.StoreTransferToList(t)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *Service) StoreTransferToList(transfer *Transfer) {
|
|
s.transferList.Store(transfer.ID, transfer)
|
|
s.NotifyTransferListUpdate()
|
|
}
|
|
|
|
func (s *Service) NotifyTransferListUpdate() {
|
|
s.app.Event.Emit("transfer:refreshList")
|
|
}
|
|
|
|
// CleanTransferList 清理完成的 transfer
|
|
func (s *Service) CleanTransferList() {
|
|
s.transferList.Range(func(key, value any) bool {
|
|
task := value.(*Transfer)
|
|
if task.Status == TransferStatusCompleted ||
|
|
task.Status == TransferStatusError ||
|
|
task.Status == TransferStatusCanceled ||
|
|
task.Status == TransferStatusRejected {
|
|
s.transferList.Delete(key)
|
|
}
|
|
return true
|
|
})
|
|
s.NotifyTransferListUpdate()
|
|
}
|
|
|
|
func (s *Service) DeleteTransfer(transferID string) {
|
|
s.transferList.Delete(transferID)
|
|
s.NotifyTransferListUpdate()
|
|
}
|