feat: update
This commit is contained in:
@@ -2,7 +2,6 @@ package model
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"quyun/app/requests"
|
||||
"quyun/database/fields"
|
||||
@@ -10,55 +9,34 @@ import (
|
||||
|
||||
. "github.com/go-jet/jet/v2/postgres"
|
||||
"github.com/samber/lo"
|
||||
log "github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
func (m *Medias) log() *log.Entry {
|
||||
return log.WithField("model", "MediasModel")
|
||||
var mediasUpdateExcludeColumns = []Column{
|
||||
table.Medias.CreatedAt,
|
||||
}
|
||||
|
||||
func (m *Medias) BuildConditionWithKey(key *string) BoolExpression {
|
||||
tbl := table.Medias
|
||||
func (m *Medias) Like(key *string) Cond {
|
||||
return func(cond BoolExpression) BoolExpression {
|
||||
tbl := table.Medias
|
||||
if key == nil || *key == "" {
|
||||
return cond
|
||||
}
|
||||
|
||||
cond := Bool(true)
|
||||
cond = cond.AND(
|
||||
tbl.Name.LIKE(String("%" + *key + "%")),
|
||||
)
|
||||
|
||||
if key == nil || *key == "" {
|
||||
return cond
|
||||
}
|
||||
|
||||
cond = cond.AND(
|
||||
tbl.Name.LIKE(String("%" + *key + "%")),
|
||||
)
|
||||
|
||||
return cond
|
||||
}
|
||||
|
||||
// countByCond
|
||||
func (m *Medias) countByCondition(ctx context.Context, expr BoolExpression) (int64, error) {
|
||||
var cnt struct {
|
||||
Cnt int64
|
||||
}
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := SELECT(COUNT(tbl.ID).AS("cnt")).FROM(tbl).WHERE(expr)
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
err := stmt.QueryContext(ctx, db, &cnt)
|
||||
if err != nil {
|
||||
m.log().Errorf("error counting media items: %v", err)
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return cnt.Cnt, nil
|
||||
}
|
||||
|
||||
func (m *Medias) List(ctx context.Context, pagination *requests.Pagination, expr BoolExpression) (*requests.Pager, error) {
|
||||
func (m *Medias) List(ctx context.Context, pagination *requests.Pagination, conds ...Cond) (*requests.Pager, error) {
|
||||
pagination.Format()
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
SELECT(tbl.AllColumns).
|
||||
WHERE(expr).
|
||||
WHERE(CondTrue(conds...)).
|
||||
ORDER_BY(tbl.ID.DESC()).
|
||||
LIMIT(pagination.Limit).
|
||||
OFFSET(pagination.Offset)
|
||||
@@ -71,7 +49,7 @@ func (m *Medias) List(ctx context.Context, pagination *requests.Pagination, expr
|
||||
return nil, err
|
||||
}
|
||||
|
||||
count, err := m.countByCondition(ctx, expr)
|
||||
count, err := m.Count(ctx, conds...)
|
||||
if err != nil {
|
||||
m.log().Errorf("error getting media count: %v", err)
|
||||
return nil, err
|
||||
@@ -84,33 +62,6 @@ func (m *Medias) List(ctx context.Context, pagination *requests.Pagination, expr
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (m *Medias) BatchCreate(ctx context.Context, models []*Medias) error {
|
||||
stmt := table.Medias.INSERT(table.Medias.MutableColumns).MODELS(models)
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
if _, err := stmt.ExecContext(ctx, db); err != nil {
|
||||
m.log().Errorf("error creating media item: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
m.log().Infof("media item created successfully")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Medias) Create(ctx context.Context) error {
|
||||
m.CreatedAt = time.Now()
|
||||
stmt := table.Medias.INSERT(table.Medias.MutableColumns).MODEL(m).RETURNING(table.Medias.AllColumns)
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
if err := stmt.QueryContext(ctx, db, m); err != nil {
|
||||
m.log().Errorf("error creating media item: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
m.log().Infof("media item created successfully")
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetByIds
|
||||
func (m *Medias) GetByIds(ctx context.Context, ids []int64) ([]*Medias, error) {
|
||||
if len(ids) == 0 {
|
||||
@@ -157,59 +108,6 @@ func (m *Medias) GetByHash(ctx context.Context, hash string) (*Medias, error) {
|
||||
return &media, nil
|
||||
}
|
||||
|
||||
// Update
|
||||
func (m *Medias) Update(ctx context.Context, hash string, model *Medias) error {
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
UPDATE(tbl.MutableColumns.Except(tbl.CreatedAt)).
|
||||
MODEL(model).
|
||||
WHERE(table.Medias.Hash.EQ(String(hash)))
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
if _, err := stmt.ExecContext(ctx, db); err != nil {
|
||||
m.log().WithField("hash", hash).Errorf("error updating media item: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
m.log().Infof("media item updated successfully")
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetByID
|
||||
func (m *Medias) GetByID(ctx context.Context, id int64) (*Medias, error) {
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
SELECT(tbl.AllColumns).
|
||||
WHERE(tbl.ID.EQ(Int64(id)))
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
var media Medias
|
||||
|
||||
if err := stmt.QueryContext(ctx, db, &media); err != nil {
|
||||
m.log().Errorf("error querying media item by ID: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &media, nil
|
||||
}
|
||||
|
||||
// Delete
|
||||
func (m *Medias) Delete(ctx context.Context, id int64) error {
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
DELETE().
|
||||
WHERE(tbl.ID.EQ(Int64(id)))
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
if _, err := stmt.ExecContext(ctx, db); err != nil {
|
||||
m.log().Errorf("error deleting media item: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
m.log().Infof("media item deleted successfully")
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateMetas
|
||||
func (m *Medias) UpdateMetas(ctx context.Context, id int64, metas fields.MediaMetas) error {
|
||||
meta := fields.ToJson(metas)
|
||||
@@ -250,21 +148,3 @@ func (m *Medias) GetRelations(ctx context.Context, hash string) ([]*Medias, erro
|
||||
return &media
|
||||
}), nil
|
||||
}
|
||||
|
||||
// Count
|
||||
func (m *Medias) Count(ctx context.Context) (int64, error) {
|
||||
tbl := table.Medias
|
||||
stmt := tbl.SELECT(COUNT(tbl.ID).AS("count"))
|
||||
m.log().Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
var count struct {
|
||||
Count int64
|
||||
}
|
||||
|
||||
if err := stmt.QueryContext(ctx, db, &count); err != nil {
|
||||
m.log().Errorf("error counting media items: %v", err)
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return count.Count, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user