complete backend

This commit is contained in:
yanghao05
2025-04-11 14:48:36 +08:00
parent 736991e3ea
commit 79972e963c
10 changed files with 131 additions and 21 deletions

View File

@@ -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,

View File

@@ -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)
}

View File

@@ -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
}

View File

@@ -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

View File

@@ -18,4 +18,5 @@ type Medias struct {
MimeType string `json:"mime_type"`
Size int64 `json:"size"`
Path string `json:"path"`
Hash string `json:"hash"`
}

View File

@@ -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,

View File

@@ -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 {