143 lines
3.1 KiB
Go
143 lines
3.1 KiB
Go
package model
|
|
|
|
import (
|
|
"context"
|
|
|
|
"quyun/app/requests"
|
|
"quyun/database/fields"
|
|
|
|
. "github.com/go-jet/jet/v2/postgres"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
var tblMediasUpdateMutableColumns = tblMedias.MutableColumns.Except(
|
|
tblMedias.CreatedAt,
|
|
)
|
|
|
|
func (m *Medias) Like(key *string) Cond {
|
|
return func(cond BoolExpression) BoolExpression {
|
|
tbl := tblMedias
|
|
if key == nil || *key == "" {
|
|
return cond
|
|
}
|
|
|
|
cond = cond.AND(
|
|
tbl.Name.LIKE(String("%" + *key + "%")),
|
|
)
|
|
|
|
return cond
|
|
}
|
|
}
|
|
|
|
func (m *Medias) List(ctx context.Context, pagination *requests.Pagination, conds ...Cond) (*requests.Pager, error) {
|
|
pagination.Format()
|
|
|
|
tbl := tblMedias
|
|
stmt := tbl.
|
|
SELECT(tbl.AllColumns).
|
|
WHERE(CondTrue(conds...)).
|
|
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.Count(ctx, conds...)
|
|
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
|
|
}
|
|
|
|
// GetByIds
|
|
func (m *Medias) GetByIds(ctx context.Context, ids []int64) ([]*Medias, error) {
|
|
if len(ids) == 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
tbl := tblMedias
|
|
stmt := tbl.
|
|
SELECT(tbl.AllColumns).
|
|
WHERE(tbl.ID.IN(IntExprSlice(ids)...))
|
|
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) {
|
|
stmt := tblMedias.
|
|
SELECT(tblMedias.AllColumns).
|
|
WHERE(tblMedias.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
|
|
}
|
|
|
|
// UpdateMetas
|
|
func (m *Medias) UpdateMetas(ctx context.Context, id int64, metas fields.MediaMetas) error {
|
|
meta := fields.ToJson(metas)
|
|
|
|
stmt := tblMedias.
|
|
UPDATE(tblMedias.Metas).
|
|
SET(meta).
|
|
WHERE(tblMedias.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) {
|
|
stmt := tblMedias.
|
|
SELECT(tblMedias.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
|
|
}
|