feat: update

2025-05-23 20:07:34 +08:00
parent 9b38699764
commit 57cb0a750b
53 changed files with 741 additions and 982 deletions
--- a/backend/app/model/medias.go
+++ b/backend/app/model/medias.go
@@ -0,0 +1,270 @@
+package model
+
+import (
+	"context"
+	"time"
+
+	"quyun/app/requests"
+	"quyun/database/fields"
+	"quyun/database/table"
+
+	. "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")
+}
+
+func (m *Medias) BuildConditionWithKey(key *string) BoolExpression {
+	tbl := table.Medias
+
+	cond := Bool(true)
+
+	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) {
+	pagination.Format()
+
+	tbl := table.Medias
+	stmt := tbl.
+		SELECT(tbl.AllColumns).
+		WHERE(expr).
+		ORDER_BY(tbl.ID.DESC()).
+		LIMIT(pagination.Limit).
+		OFFSET(pagination.Offset)
+	m.log().Infof("sql: %s", stmt.DebugSql())
+
+	var medias []Medias
+	err := stmt.QueryContext(ctx, db, &medias)
+	if err != nil {
+		m.log().Errorf("error querying media items: %v", err)
+		return nil, err
+	}
+
+	count, err := m.countByCondition(ctx, expr)
+	if err != nil {
+		m.log().Errorf("error getting media count: %v", err)
+		return nil, err
+	}
+
+	return &requests.Pager{
+		Items:      medias,
+		Total:      count,
+		Pagination: *pagination,
+	}, 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, medias *Medias) error {
+	medias.CreatedAt = time.Now()
+	stmt := table.Medias.INSERT(table.Medias.MutableColumns).MODEL(medias).RETURNING(table.Medias.AllColumns)
+	m.log().Infof("sql: %s", stmt.DebugSql())
+
+	if err := stmt.QueryContext(ctx, db, medias); 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 {
+		return nil, nil
+	}
+
+	condIds := lo.Map(ids, func(id int64, _ int) Expression {
+		return Int64(id)
+	})
+
+	tbl := table.Medias
+	stmt := tbl.
+		SELECT(tbl.AllColumns).
+		WHERE(tbl.ID.IN(condIds...))
+	m.log().Infof("sql: %s", stmt.DebugSql())
+
+	var medias []Medias
+	err := stmt.QueryContext(ctx, db, &medias)
+	if err != nil {
+		m.log().Errorf("error querying media items: %v", err)
+		return nil, err
+	}
+
+	return lo.Map(medias, func(media Medias, _ int) *Medias {
+		return &media
+	}), nil
+}
+
+// GetByHash
+func (m *Medias) GetByHash(ctx context.Context, hash string) (*Medias, error) {
+	tbl := table.Medias
+	stmt := tbl.
+		SELECT(tbl.AllColumns).
+		WHERE(tbl.Hash.EQ(String(hash)))
+	m.log().Infof("sql: %s", stmt.DebugSql())
+
+	var media Medias
+	err := stmt.QueryContext(ctx, db, &media)
+	if err != nil {
+		m.log().Errorf("error querying media item by hash: %v", err)
+		return nil, err
+	}
+
+	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)
+
+	tbl := table.Medias
+	stmt := tbl.
+		UPDATE(tbl.Metas).
+		SET(meta).
+		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 updating media metas: %v", err)
+		return err
+	}
+
+	m.log().Infof("media (%d) metas updated successfully", id)
+	return nil
+}
+
+// GetRelationMedias
+func (m *Medias) GetRelations(ctx context.Context, hash string) ([]*Medias, error) {
+	tbl := table.Medias
+	stmt := tbl.
+		SELECT(tbl.AllColumns).
+		WHERE(
+			RawBool("metas->>'parent_hash' = ?", RawArgs{"?": hash}),
+		)
+	m.log().Infof("sql: %s", stmt.DebugSql())
+
+	var medias []Medias
+	if err := stmt.QueryContext(ctx, db, &medias); err != nil {
+		m.log().Errorf("error querying media items: %v", err)
+		return nil, err
+	}
+
+	return lo.Map(medias, func(media Medias, _ int) *Medias {
+		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
+}