feat: add storages

backend/app/http/storages/service.go

@@ -0,0 +1,124 @@
+package storages
+
+import (
+	"context"
+	"database/sql"
+
+	"backend/database/models/qvyun_v2/public/model"
+	"backend/database/models/qvyun_v2/public/table"
+	"backend/providers/otel"
+
+	. "github.com/go-jet/jet/v2/postgres"
+	log "github.com/sirupsen/logrus"
+	semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
+)
+
+// @provider:except
+type Service struct {
+	db  *sql.DB
+	log *log.Entry `inject:"false"`
+}
+
+func (svc *Service) Prepare() error {
+	svc.log = log.WithField("module", "storages.service")
+	_ = Int(1)
+	return nil
+}
+
+// GetStorages
+func (svc *Service) GetStorages(ctx context.Context) ([]model.Storages, error) {
+	_, span := otel.Start(ctx, "storages.service.GetStorages")
+	defer span.End()
+
+	tbl := table.Storages
+	stmt := tbl.SELECT(tbl.AllColumns)
+	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
+}
+
+// GetStorageByID
+func (svc *Service) GetStorageByID(ctx context.Context, id int64) (*model.Storages, error) {
+	_, span := otel.Start(ctx, "storages.service.GetStorageByID")
+	defer span.End()
+
+	tbl := table.Storages
+	stmt := tbl.SELECT(tbl.AllColumns).WHERE(tbl.ID.EQ(Int64(id)))
+	span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
+
+	var storage model.Storages
+	if err := stmt.QueryContext(ctx, svc.db, &storage); err != nil {
+		return nil, err
+	}
+	return &storage, nil
+}
+
+// DeleteStorageByID
+func (svc *Service) DeleteStorageByID(ctx context.Context, id int64) error {
+	_, span := otel.Start(ctx, "storages.service.DeleteStorageByID")
+	defer span.End()
+
+	tbl := table.Storages
+	stmt := tbl.DELETE().WHERE(tbl.ID.EQ(Int64(id)))
+	span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
+
+	_, err := stmt.ExecContext(ctx, svc.db)
+	return err
+}
+
+// Create
+func (svc *Service) Create(ctx context.Context, req *model.Storages) error {
+	_, span := otel.Start(ctx, "storages.service.Create")
+	defer span.End()
+
+	tbl := table.Storages
+	stmt := tbl.INSERT(tbl.Name, tbl.Type, tbl.Config).VALUES(req.Name, req.Type, req.Config)
+	span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
+
+	_, err := stmt.ExecContext(ctx, svc.db)
+	return err
+}
+
+// SetDefault
+func (svc *Service) SetDefault(ctx context.Context, id int64) error {
+	_, span := otel.Start(ctx, "storages.service.SetDefault")
+	defer span.End()
+
+	// Start transaction
+	tx, err := svc.db.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	tbl := table.Storages
+
+	// First, set all storages' is_default to false
+	resetStmt := tbl.
+		UPDATE(tbl.IsDefault).
+		SET(tbl.IsDefault.SET(Bool(false)))
+
+	span.SetAttributes(semconv.DBStatementKey.String(resetStmt.DebugSql()))
+	if _, err := resetStmt.ExecContext(ctx, tx); err != nil {
+		return err
+	}
+
+	// Then, set the specified storage's is_default to true
+	setDefaultStmt := tbl.
+		UPDATE(tbl.IsDefault).
+		SET(tbl.IsDefault.SET(Bool(true))).
+		WHERE(tbl.ID.EQ(Int64(id)))
+
+	span.SetAttributes(semconv.DBStatementKey.String(setDefaultStmt.DebugSql()))
+
+	if _, err := setDefaultStmt.ExecContext(ctx, tx); err != nil {
+		return err
+	}
+
+	// Commit transaction
+	return tx.Commit()
+}