127 lines
2.9 KiB
Go
127 lines
2.9 KiB
Go
package medias
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"time"
|
|
|
|
"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"
|
|
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", "medias.service")
|
|
_ = Int(1)
|
|
return nil
|
|
}
|
|
|
|
// Create
|
|
func (svc *Service) Create(ctx context.Context, m *model.Medias) (*model.Medias, error) {
|
|
_, span := otel.Start(ctx, "medias.service.Create")
|
|
defer span.End()
|
|
|
|
if m.CreatedAt.IsZero() {
|
|
m.CreatedAt = time.Now()
|
|
}
|
|
|
|
if m.UpdatedAt.IsZero() {
|
|
m.UpdatedAt = time.Now()
|
|
}
|
|
|
|
tbl := table.Medias
|
|
stmt := tbl.INSERT(tbl.MutableColumns).MODEL(m).RETURNING(tbl.AllColumns)
|
|
span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
|
|
|
|
var ret model.Medias
|
|
if err := stmt.QueryContext(ctx, svc.db, &ret); err != nil {
|
|
return nil, err
|
|
}
|
|
return &ret, nil
|
|
}
|
|
|
|
// GetMediasByHash
|
|
func (svc *Service) GetMediasByHash(ctx context.Context, tenantID, userID int64, hashes []string) ([]*model.Medias, error) {
|
|
_, span := otel.Start(ctx, "medias.service.GetMediasByHash")
|
|
defer span.End()
|
|
|
|
hashExpr := lo.Map(hashes, func(item string, index int) Expression { return String(item) })
|
|
|
|
tbl := table.Medias
|
|
stmt := tbl.
|
|
SELECT(tbl.AllColumns).
|
|
WHERE(
|
|
tbl.TenantID.
|
|
EQ(Int64(tenantID)).
|
|
AND(
|
|
tbl.UserID.EQ(Int64(userID)),
|
|
).
|
|
AND(
|
|
tbl.Hash.IN(hashExpr...),
|
|
),
|
|
)
|
|
span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
|
|
|
|
var ret []model.Medias
|
|
if err := stmt.QueryContext(ctx, svc.db, &ret); err != nil {
|
|
return nil, err
|
|
}
|
|
return lo.Map(ret, func(item model.Medias, _ int) *model.Medias {
|
|
return &item
|
|
}), nil
|
|
}
|
|
|
|
func (svc *Service) GetMediaByHash(ctx context.Context, tenantID, userID int64, hash string) (*model.Medias, error) {
|
|
_, span := otel.Start(ctx, "medias.service.GetMediasByHash")
|
|
defer span.End()
|
|
|
|
tbl := table.Medias
|
|
stmt := tbl.
|
|
SELECT(tbl.AllColumns).
|
|
LIMIT(1).
|
|
WHERE(
|
|
tbl.TenantID.
|
|
EQ(Int64(tenantID)).
|
|
AND(
|
|
tbl.UserID.EQ(Int64(userID)),
|
|
).
|
|
AND(
|
|
tbl.Hash.EQ(String(hash)),
|
|
),
|
|
)
|
|
span.SetAttributes(semconv.DBStatementKey.String(stmt.DebugSql()))
|
|
|
|
var ret model.Medias
|
|
if err := stmt.QueryContext(ctx, svc.db, &ret); err != nil {
|
|
return nil, err
|
|
}
|
|
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
|
|
}
|