141 lines
3.6 KiB
Go
141 lines
3.6 KiB
Go
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()
|
|
}
|
|
|
|
// GetDefault
|
|
func (svc *Service) GetDefault(ctx context.Context) (*model.Storages, error) {
|
|
_, span := otel.Start(ctx, "storages.service.GetDefault")
|
|
defer span.End()
|
|
|
|
tbl := table.Storages
|
|
stmt := tbl.SELECT(tbl.AllColumns).WHERE(tbl.IsDefault.EQ(Bool(true)))
|
|
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
|
|
}
|