add storages

backend/app/http/medias/service.go

@@ -108,3 +108,19 @@ func (svc *Service) GetMediaByHash(ctx context.Context, tenantID, userID int64,
 	}
 	return &ret, nil
 }
+
+func (svc *Service) DeleteByID(ctx context.Context, id ...int64) error {
+	if len(id) == 0 {
+		return nil
+	}
+
+	_, span := otel.Start(ctx, "medias.service.DeleteByID")
+	defer span.End()
+
+	tbl := table.Medias
+	stmt := tbl.DELETE().WHERE(tbl.ID.IN(lo.Map(id, func(item int64, _ int) Expression { return Int64(item) })...))
+	span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
+
+	_, err := stmt.ExecContext(ctx, svc.db)
+	return err
+}

backend/app/http/storages/service.go

@@ -3,13 +3,17 @@ package storages
 import (
 	"context"
 	"database/sql"
+	"errors"
 
+	"backend/database/fields"
 	"backend/database/models/qvyun_v2/public/model"
 	"backend/database/models/qvyun_v2/public/table"
 	"backend/providers/otel"
 
 	. "github.com/go-jet/jet/v2/postgres"
+	"github.com/samber/lo"
 	log "github.com/sirupsen/logrus"
+	"github.com/spf13/afero"
 	semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
 )
 
@@ -138,3 +142,74 @@ func (svc *Service) GetDefault(ctx context.Context) (*model.Storages, error) {
 	}
 	return &storage, nil
 }
+
+// GetStoragesByID
+func (svc *Service) GetStoragesByID(ctx context.Context, ids ...int64) ([]model.Storages, error) {
+	if len(ids) == 0 {
+		return nil, nil
+	}
+	ids = lo.Uniq(ids)
+
+	_, span := otel.Start(ctx, "storages.service.GetStoragesByID")
+	defer span.End()
+
+	tbl := table.Storages
+	stmt := tbl.SELECT(tbl.AllColumns).WHERE(tbl.ID.IN(lo.Map(ids, func(i int64, _ int) Expression { return Int64(i) })...))
+	span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
+
+	var storages []model.Storages
+	if err := stmt.QueryContext(ctx, svc.db, &storages); err != nil {
+		return nil, err
+	}
+	return storages, nil
+}
+
+func (svc *Service) GetDefaultFS(ctx context.Context) (afero.Fs, error) {
+	st, err := svc.GetDefault(ctx)
+	if err != nil {
+		return nil, err
+	}
+	return svc.BuildFS(st)
+}
+
+func (svc *Service) GetFSMapByIDs(ctx context.Context, ids ...int64) (map[int64]afero.Fs, error) {
+	_, span := otel.Start(ctx, "storages.service.GetFSByID")
+	defer span.End()
+
+	storages, err := svc.GetStoragesByID(ctx, ids...)
+	if err != nil {
+		return nil, err
+	}
+
+	maps := make(map[int64]afero.Fs)
+	for _, s := range storages {
+		fs, err := svc.BuildFS(&s)
+		if err != nil {
+			return nil, err
+		}
+		maps[s.ID] = fs
+	}
+
+	return maps, nil
+}
+
+// GetFSByID
+func (svc *Service) GetFSByID(ctx context.Context, id int64) (afero.Fs, error) {
+	_, span := otel.Start(ctx, "storages.service.GetFSByID")
+	defer span.End()
+
+	storage, err := svc.GetStorageByID(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+
+	return svc.BuildFS(storage)
+}
+
+func (svc *Service) BuildFS(storage *model.Storages) (afero.Fs, error) {
+	switch storage.Type {
+	case fields.StorageTypeLocal:
+		return afero.NewBasePathFs(afero.NewOsFs(), storage.Config.Local.Path), nil
+	}
+	return nil, errors.New("invalid storage type")
+}