feat: complte media store
This commit is contained in:
Rogee
@@ -5,13 +5,13 @@ import (
|
||||
"database/sql"
|
||||
"path/filepath"
|
||||
|
||||
"backend/common/media_store"
|
||||
"backend/database/models/qvyun/public/table"
|
||||
"backend/pkg/path"
|
||||
"backend/pkg/pg"
|
||||
"backend/providers/storage"
|
||||
|
||||
. "github.com/go-jet/jet/v2/postgres"
|
||||
"github.com/google/uuid"
|
||||
"github.com/pkg/errors"
|
||||
"github.com/samber/lo"
|
||||
"github.com/sirupsen/logrus"
|
||||
@@ -162,75 +162,40 @@ func (svc *Service) HasUserBought(ctx context.Context, tenantId, userId, mediaId
|
||||
return mediaID > 0, nil
|
||||
}
|
||||
|
||||
// UnPublishTenantWithNotInUUIDs
|
||||
func (svc *Service) UnPublishTenantWithNotInUUIDs(ctx context.Context, tenantId int64, uuids []string) error {
|
||||
log := svc.log.WithField("method", "UnPublishTenantWithNotInUUIDs")
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
UPDATE().
|
||||
SET(
|
||||
tbl.Publish.SET(Bool(false)),
|
||||
).
|
||||
WHERE(
|
||||
tbl.TenantID.EQ(Int(tenantId)).AND(
|
||||
tbl.UUID.NOT_IN(lo.Map(uuids, func(item string, _ int) Expression {
|
||||
return String(item)
|
||||
})...),
|
||||
),
|
||||
)
|
||||
log.Debug(stmt.DebugSql())
|
||||
|
||||
if _, err := stmt.ExecContext(ctx, svc.db); err != nil {
|
||||
return errors.Wrap(err, "unpublish tenant with not in uuids")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetTenantUUIDs
|
||||
func (svc *Service) GetTenantUUIDs(ctx context.Context, tenantId int64) ([]string, error) {
|
||||
log := svc.log.WithField("method", "GetTenantUUIDs")
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
SELECT(tbl.UUID).
|
||||
WHERE(tbl.TenantID.EQ(Int(tenantId)))
|
||||
log.Debug(stmt.DebugSql())
|
||||
|
||||
var uuids []string
|
||||
if err := stmt.QueryContext(ctx, svc.db, &uuids); err != nil {
|
||||
return nil, errors.Wrap(err, "query tenant uuids")
|
||||
}
|
||||
|
||||
return uuids, nil
|
||||
}
|
||||
|
||||
// PublishTenant
|
||||
func (svc *Service) PublishTenantMedia(ctx context.Context, tenantId int64, uuid uuid.UUID, name string, price uint) error {
|
||||
log := svc.log.WithField("method", "PublishTenant")
|
||||
// Upsert
|
||||
func (svc *Service) Upsert(ctx context.Context, tenantId int64, item media_store.VideoInfo) error {
|
||||
log := svc.log.WithField("method", "Upsert")
|
||||
|
||||
resources := pg.MediaResources{}
|
||||
if path.DirExists(filepath.Join(svc.storageConfig.Path, uuid.String(), pg.MediaTypeVideo.String())) {
|
||||
if path.DirExists(filepath.Join(svc.storageConfig.Path, item.Hash, pg.MediaTypeVideo.String())) {
|
||||
resources = append(resources, pg.MediaTypeVideo)
|
||||
}
|
||||
|
||||
if path.DirExists(filepath.Join(svc.storageConfig.Path, uuid.String(), pg.MediaTypeAudio.String())) {
|
||||
if path.DirExists(filepath.Join(svc.storageConfig.Path, item.Hash, pg.MediaTypeAudio.String())) {
|
||||
resources = append(resources, pg.MediaTypeAudio)
|
||||
}
|
||||
|
||||
if path.DirExists(filepath.Join(svc.storageConfig.Path, uuid.String(), pg.MediaTypePdf.String())) {
|
||||
if path.DirExists(filepath.Join(svc.storageConfig.Path, item.Hash, pg.MediaTypePdf.String())) {
|
||||
resources = append(resources, pg.MediaTypePdf)
|
||||
}
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
INSERT(tbl.TenantID, tbl.UUID, tbl.Title, tbl.Price, tbl.Resources, tbl.Publish).
|
||||
VALUES(Int(tenantId), UUID(uuid), String(name), Int(int64(price)), Json(resources.MustValue()), Bool(true))
|
||||
INSERT(tbl.TenantID, tbl.Hash, tbl.Title, tbl.Price, tbl.Resources, tbl.Publish).
|
||||
VALUES(Int(tenantId), String(item.Hash), String(item.Name), Int(item.Price()), Json(resources), Bool(true)).
|
||||
ON_CONFLICT(tbl.Hash).
|
||||
DO_UPDATE(
|
||||
SET(
|
||||
tbl.Title.SET(String(item.Name)),
|
||||
tbl.Price.SET(Int(item.Price())),
|
||||
tbl.Resources.SET(Json(resources)),
|
||||
tbl.Publish.SET(Bool(true)),
|
||||
),
|
||||
)
|
||||
log.Debug(stmt.DebugSql())
|
||||
|
||||
if _, err := stmt.ExecContext(ctx, svc.db); err != nil {
|
||||
return errors.Wrap(err, "publish tenant")
|
||||
return errors.Wrapf(err, "upsert media: %s %s", item.Hash, item.Name)
|
||||
}
|
||||
|
||||
return nil
|
||||
|
||||
@@ -8,10 +8,7 @@ import (
|
||||
"backend/database/models/qvyun/public/table"
|
||||
"backend/fixtures"
|
||||
dbUtil "backend/pkg/db"
|
||||
"backend/providers/storage"
|
||||
|
||||
. "github.com/go-jet/jet/v2/postgres"
|
||||
"github.com/google/uuid"
|
||||
"github.com/samber/lo"
|
||||
. "github.com/smartystreets/goconvey/convey"
|
||||
)
|
||||
@@ -52,158 +49,3 @@ func TestService_GetUserBoughtMedias(t *testing.T) {
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestService_List(t *testing.T) {
|
||||
Convey("TestService_list", t, func() {
|
||||
db, err := fixtures.GetDB()
|
||||
So(err, ShouldBeNil)
|
||||
defer db.Close()
|
||||
|
||||
Convey("truncate all tables", func() {
|
||||
So(dbUtil.TruncateAllTables(context.TODO(), db, "medias", "media_resources", "user_medias"), ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("insert user_medias data", func() {
|
||||
items := []model.UserMedias{
|
||||
{UserID: 1, TenantID: 1, MediaID: 1, Price: 10},
|
||||
}
|
||||
|
||||
tbl := table.UserMedias
|
||||
stmt := tbl.INSERT(tbl.UserID, tbl.TenantID, tbl.MediaID, tbl.Price).MODELS(items)
|
||||
|
||||
_, err := stmt.Exec(db)
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("insert medias data", func() {
|
||||
items := []model.Medias{
|
||||
{
|
||||
UUID: uuid.New(),
|
||||
TenantID: 1,
|
||||
Title: "title1",
|
||||
Description: "hello",
|
||||
Price: 100,
|
||||
Publish: true,
|
||||
},
|
||||
{
|
||||
UUID: uuid.New(),
|
||||
TenantID: 1,
|
||||
Title: "title2",
|
||||
Description: "hello",
|
||||
Price: 100,
|
||||
Publish: true,
|
||||
},
|
||||
{
|
||||
UUID: uuid.New(),
|
||||
TenantID: 1,
|
||||
Title: "title3",
|
||||
Description: "hello",
|
||||
Price: 100,
|
||||
Publish: false,
|
||||
},
|
||||
}
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := tbl.INSERT(
|
||||
tbl.UUID,
|
||||
tbl.TenantID,
|
||||
tbl.Title,
|
||||
tbl.Description,
|
||||
tbl.Price,
|
||||
tbl.Publish,
|
||||
).MODELS(items)
|
||||
t.Log(stmt.DebugSql())
|
||||
|
||||
_, err := stmt.Exec(db)
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("get list", func() {
|
||||
svc := &Service{db: db}
|
||||
So(svc.Prepare(), ShouldBeNil)
|
||||
|
||||
items, err := svc.List(context.TODO(), 1, 1, &ListFilter{
|
||||
Bought: lo.ToPtr(true),
|
||||
})
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
t.Logf("items: %+v", items)
|
||||
So(items, ShouldHaveLength, 1)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestService_List1(t *testing.T) {
|
||||
Convey("TestService_list", t, func() {
|
||||
db, err := fixtures.GetDB()
|
||||
So(err, ShouldBeNil)
|
||||
defer db.Close()
|
||||
|
||||
tbl := table.Medias
|
||||
stmt := tbl.
|
||||
SELECT(tbl.AllColumns).
|
||||
WHERE(
|
||||
tbl.Publish.EQ(Bool(true)).AND(
|
||||
tbl.TenantID.EQ(Int(1)),
|
||||
),
|
||||
).
|
||||
WHERE(tbl.ID.EQ(Int(1))).
|
||||
ORDER_BY(tbl.ID.DESC())
|
||||
|
||||
t.Log(stmt.DebugSql())
|
||||
|
||||
var dest []ListItem
|
||||
err = stmt.QueryContext(context.TODO(), db, &dest)
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
t.Logf("dest: %+v", dest)
|
||||
})
|
||||
}
|
||||
|
||||
func TestService_PublishTenant(t *testing.T) {
|
||||
Convey("TestService_PublishTenant", t, func() {
|
||||
db, err := fixtures.GetDB()
|
||||
So(err, ShouldBeNil)
|
||||
defer db.Close()
|
||||
|
||||
Convey("truncate all tables", func() {
|
||||
So(dbUtil.TruncateAllTables(context.TODO(), db, "medias"), ShouldBeNil)
|
||||
})
|
||||
|
||||
u1, err := uuid.Parse("6c63c4d8-b2cb-4588-8dd3-9a2276918d17")
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
u2, err := uuid.Parse("8c183427-02b1-4426-ad65-18c6e7735072")
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
u3, err := uuid.Parse("24aa56a1-a85b-4a03-9a1b-c39a36103ddf")
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
Convey("insert some data", func() {
|
||||
items := []model.Medias{
|
||||
{UUID: u1, TenantID: 1, Title: "title1", Description: "hello", Price: 100, Publish: true},
|
||||
{UUID: u2, TenantID: 1, Title: "title2", Description: "hello", Price: 100, Publish: true},
|
||||
{UUID: u3, TenantID: 1, Title: "title3", Description: "hello", Price: 100, Publish: true},
|
||||
}
|
||||
|
||||
Convey("publish tenant", func() {
|
||||
svc := &Service{
|
||||
db: db,
|
||||
storageConfig: &storage.Config{
|
||||
Path: "/projects/mp-qvyun/backend/fixtures/medias",
|
||||
},
|
||||
}
|
||||
So(svc.Prepare(), ShouldBeNil)
|
||||
|
||||
err := svc.PublishTenantMedia(context.TODO(), 1, items[0].UUID, items[0].Title, 100)
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
err = svc.PublishTenantMedia(context.TODO(), 1, items[1].UUID, items[1].Title, 200)
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
err = svc.PublishTenantMedia(context.TODO(), 1, items[2].UUID, items[2].Title, 300)
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user