complete backend
This commit is contained in:
@@ -92,6 +92,16 @@ func (r *Routes) Register(router fiber.Router) {
|
||||
r.uploads.Token,
|
||||
))
|
||||
|
||||
router.Get("/v1/admin/uploads/pre-uploaded-check/:md5", Func1(
|
||||
r.uploads.PreUploadCheck,
|
||||
PathParam[string]("md5"),
|
||||
))
|
||||
|
||||
router.Post("/v1/admin/uploads/post-uploaded-action", Func1(
|
||||
r.uploads.PostUploadedAction,
|
||||
Body[PostUploadedForm]("body"),
|
||||
))
|
||||
|
||||
// 注册路由组: users
|
||||
router.Get("/v1/admin/users", DataFunc2(
|
||||
r.users.List,
|
||||
|
||||
@@ -14,10 +14,13 @@ import (
|
||||
"quyun/providers/ali"
|
||||
"quyun/providers/app"
|
||||
|
||||
"github.com/go-jet/jet/v2/qrm"
|
||||
"github.com/gofiber/fiber/v3"
|
||||
log "github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
const UPLOAD_PATH = "quyun"
|
||||
|
||||
// @provider
|
||||
type uploads struct {
|
||||
app *app.Config
|
||||
@@ -143,5 +146,42 @@ func (up *uploads) Complete(ctx fiber.Ctx, md5 string, body *UploadFileInfo) err
|
||||
// Token
|
||||
// @Router /v1/admin/uploads/token [get]
|
||||
func (up *uploads) Token(ctx fiber.Ctx) (*ali.PolicyToken, error) {
|
||||
return up.ali.GetToken("quyun")
|
||||
return up.ali.GetToken(UPLOAD_PATH)
|
||||
}
|
||||
|
||||
// PreUploadCheck
|
||||
// @Router /v1/admin/uploads/pre-uploaded-check/:md5 [get]
|
||||
// @Bind md5 path
|
||||
func (up *uploads) PreUploadCheck(ctx fiber.Ctx, md5 string) error {
|
||||
_, err := models.Medias.GetByHash(ctx.Context(), md5)
|
||||
if err != nil && errors.Is(err, qrm.ErrNoRows) {
|
||||
return ctx.SendString("ok")
|
||||
}
|
||||
return ctx.SendString("exists")
|
||||
}
|
||||
|
||||
type PostUploadedForm struct {
|
||||
OriginalName string `json:"originalName"`
|
||||
Md5 string `json:"md5"`
|
||||
MimeType string `json:"mimeType"`
|
||||
Size int64 `json:"size"`
|
||||
}
|
||||
|
||||
// PostUploadedAction
|
||||
// @Router /v1/admin/uploads/post-uploaded-action [post]
|
||||
// @Bind body body
|
||||
func (up *uploads) PostUploadedAction(ctx fiber.Ctx, body *PostUploadedForm) error {
|
||||
m, err := models.Medias.GetByHash(ctx.Context(), body.Md5)
|
||||
if err != nil && !errors.Is(err, qrm.ErrNoRows) {
|
||||
return err
|
||||
}
|
||||
|
||||
m = &model.Medias{
|
||||
Name: body.OriginalName,
|
||||
MimeType: body.MimeType,
|
||||
Size: body.Size,
|
||||
Hash: body.Md5,
|
||||
Path: filepath.Join(UPLOAD_PATH, body.Md5+filepath.Ext(body.OriginalName)),
|
||||
}
|
||||
return models.Medias.Create(ctx.Context(), m)
|
||||
}
|
||||
|
||||
@@ -2,6 +2,7 @@ package models
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"quyun/app/requests"
|
||||
"quyun/database/fields"
|
||||
@@ -127,6 +128,7 @@ func (m *mediasModel) BatchCreate(ctx context.Context, models []*model.Medias) e
|
||||
}
|
||||
|
||||
func (m *mediasModel) Create(ctx context.Context, model *model.Medias) error {
|
||||
model.CreatedAt = time.Now()
|
||||
stmt := table.Medias.INSERT(table.Medias.MutableColumns).MODEL(model)
|
||||
m.log.Infof("sql: %s", stmt.DebugSql())
|
||||
|
||||
@@ -193,3 +195,39 @@ func (m *mediasModel) GetByIds(ctx context.Context, ids []int64) ([]*model.Media
|
||||
return &media
|
||||
}), nil
|
||||
}
|
||||
|
||||
// GetByHash
|
||||
func (m *mediasModel) GetByHash(ctx context.Context, hash string) (*model.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 model.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 *mediasModel) Update(ctx context.Context, hash string, model *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.Errorf("error updating media item: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
m.log.Infof("media item updated successfully")
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -6,7 +6,8 @@ CREATE TABLE medias(
|
||||
name varchar(255) NOT NULL DEFAULT '',
|
||||
mime_type varchar(128) NOT NULL DEFAULT '',
|
||||
size int8 NOT NULL DEFAULT 0,
|
||||
path varchar(255) NOT NULL DEFAULT ''
|
||||
path varchar(255) NOT NULL DEFAULT '',
|
||||
hash varchar(64) NOT NULL DEFAULT ''
|
||||
);
|
||||
|
||||
-- +goose StatementEnd
|
||||
|
||||
@@ -18,4 +18,5 @@ type Medias struct {
|
||||
MimeType string `json:"mime_type"`
|
||||
Size int64 `json:"size"`
|
||||
Path string `json:"path"`
|
||||
Hash string `json:"hash"`
|
||||
}
|
||||
|
||||
@@ -23,6 +23,7 @@ type mediasTable struct {
|
||||
MimeType postgres.ColumnString
|
||||
Size postgres.ColumnInteger
|
||||
Path postgres.ColumnString
|
||||
Hash postgres.ColumnString
|
||||
|
||||
AllColumns postgres.ColumnList
|
||||
MutableColumns postgres.ColumnList
|
||||
@@ -69,8 +70,9 @@ func newMediasTableImpl(schemaName, tableName, alias string) mediasTable {
|
||||
MimeTypeColumn = postgres.StringColumn("mime_type")
|
||||
SizeColumn = postgres.IntegerColumn("size")
|
||||
PathColumn = postgres.StringColumn("path")
|
||||
allColumns = postgres.ColumnList{IDColumn, CreatedAtColumn, NameColumn, MimeTypeColumn, SizeColumn, PathColumn}
|
||||
mutableColumns = postgres.ColumnList{CreatedAtColumn, NameColumn, MimeTypeColumn, SizeColumn, PathColumn}
|
||||
HashColumn = postgres.StringColumn("hash")
|
||||
allColumns = postgres.ColumnList{IDColumn, CreatedAtColumn, NameColumn, MimeTypeColumn, SizeColumn, PathColumn, HashColumn}
|
||||
mutableColumns = postgres.ColumnList{CreatedAtColumn, NameColumn, MimeTypeColumn, SizeColumn, PathColumn, HashColumn}
|
||||
)
|
||||
|
||||
return mediasTable{
|
||||
@@ -83,6 +85,7 @@ func newMediasTableImpl(schemaName, tableName, alias string) mediasTable {
|
||||
MimeType: MimeTypeColumn,
|
||||
Size: SizeColumn,
|
||||
Path: PathColumn,
|
||||
Hash: HashColumn,
|
||||
|
||||
AllColumns: allColumns,
|
||||
MutableColumns: mutableColumns,
|
||||
|
||||
@@ -101,7 +101,7 @@ func (c *Config) GetToken(path string) (*PolicyToken, error) {
|
||||
|
||||
var callbackParam CallbackParam
|
||||
callbackParam.CallbackUrl = c.CallbackURL
|
||||
callbackParam.CallbackBody = "filename=${object}&size=${size}&mimeType=${mimeType}&height=${imageInfo.height}&width=${imageInfo.width}"
|
||||
callbackParam.CallbackBody = "filename=${object}&size=${size}&mimeType=${mimeType}"
|
||||
callbackParam.CallbackBodyType = "application/x-www-form-urlencoded"
|
||||
callback_str, err := json.Marshal(callbackParam)
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user