diff --git a/app/http/controller/web/doc_controller.go b/app/http/controller/web/doc_controller.go index 4d8eb05..a3af29c 100644 --- a/app/http/controller/web/doc_controller.go +++ b/app/http/controller/web/doc_controller.go @@ -14,15 +14,19 @@ type Docs struct { } func (d *Docs) Upload(context *gin.Context) { - // TODO 1. 读取源文件,调用 py API 分块上传。 - path := context.GetString(consts.ValidatorPrefix + "path") - filePath := filepath.Join(variable.ConfigYml.GetString("FileUploadSetting.UploadFileSavePath"), variable.ConfigYml.GetString("FileUploadSetting.DocsRootPath"), path) - _ = filePath - - // STAGE 2. - if ok := model.CreateDocFactory("").InsertDocumentData(context); ok { - response.Success(context, consts.CurdStatusOkMsg, "") - } else { + // STAGE 1. 插入 MySQL 记录。 + var doc_id int64 + ok := false + if doc_id, ok = model.CreateDocFactory("").InsertDocumentData(context); !ok { response.Fail(context, consts.CurdCreatFailCode, consts.CurdCreatFailMsg, "上传文档错误") } + + // STAGE 2. 调用 python API + path := context.GetString(consts.ValidatorPrefix + "path") + filePath := filepath.Join(variable.ConfigYml.GetString("FileUploadSetting.UploadFileSavePath"), variable.ConfigYml.GetString("FileUploadSetting.DocsRootPath"), path) + + // TODO + _ = filePath + _ = doc_id + response.Success(context, consts.CurdStatusOkMsg, "") } diff --git a/app/model/doc.go b/app/model/doc.go index 871b274..7e6d679 100644 --- a/app/model/doc.go +++ b/app/model/doc.go @@ -22,16 +22,16 @@ type Doc struct { func (d *Doc) TableName() string { return "docs" } -func (d *Doc) InsertDocumentData(c *gin.Context) bool { +func (d *Doc) InsertDocumentData(c *gin.Context) (int64, bool) { var tmp Doc if err := data_bind.ShouldBindFormDataToModel(c, &tmp); err == nil { if res := d.Create(&tmp); res.Error == nil { - return true + return tmp.Id, true } else { variable.ZapLog.Error("Doc 数据新增出错", zap.Error(res.Error)) } } else { variable.ZapLog.Error("Doc 数据绑定出错", zap.Error(err)) } - return false + return 0, false }