From c34923031da16df9c8a22db57c3b03db63f4f510 Mon Sep 17 00:00:00 2001 From: LinkinStars Date: Wed, 20 Sep 2023 11:53:23 +0800 Subject: [PATCH 1/2] feat(plugin): update plugin url --- script/plugin_list | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/script/plugin_list b/script/plugin_list index b3ea84c2..a7cfc670 100644 --- a/script/plugin_list +++ b/script/plugin_list @@ -1 +1 @@ -github.com/answerdev/answer-basic-connector@latest \ No newline at end of file +github.com/answerdev/plugins/connector/basic@latest \ No newline at end of file From 5ce51874228842bf21af13201e89765c782318c4 Mon Sep 17 00:00:00 2001 From: Lev Zakharov Date: Fri, 6 Oct 2023 19:14:53 +0300 Subject: [PATCH 2/2] refactor: close used resources --- internal/base/conf/conf.go | 1 + internal/cli/dump.go | 1 + internal/cli/install_check.go | 2 ++ internal/migrations/migrations.go | 1 + internal/service/uploader/upload.go | 21 ++++++++++++--------- 5 files changed, 17 insertions(+), 9 deletions(-) diff --git a/internal/base/conf/conf.go b/internal/base/conf/conf.go index b0401e1e..b441c471 100644 --- a/internal/base/conf/conf.go +++ b/internal/base/conf/conf.go @@ -60,6 +60,7 @@ func ReadConfig(configFilePath string) (c *AllConfig, err error) { func RewriteConfig(configFilePath string, allConfig *AllConfig) error { buf := bytes.Buffer{} enc := yaml.NewEncoder(&buf) + defer enc.Close() enc.SetIndent(2) if err := enc.Encode(allConfig); err != nil { return err diff --git a/internal/cli/dump.go b/internal/cli/dump.go index 7b84d862..3696a432 100644 --- a/internal/cli/dump.go +++ b/internal/cli/dump.go @@ -15,6 +15,7 @@ func DumpAllData(dataConf *data.Database, dumpDataPath string) error { if err != nil { return err } + defer db.Close() if err = db.Ping(); err != nil { return err } diff --git a/internal/cli/install_check.go b/internal/cli/install_check.go index 6cb25987..a8926060 100644 --- a/internal/cli/install_check.go +++ b/internal/cli/install_check.go @@ -23,6 +23,7 @@ func CheckDBConnection(dataConf *data.Database) bool { fmt.Printf("connection database failed: %s\n", err) return false } + defer db.Close() if err = db.Ping(); err != nil { fmt.Printf("connection ping database failed: %s\n", err) return false @@ -38,6 +39,7 @@ func CheckDBTableExist(dataConf *data.Database) bool { fmt.Printf("connection database failed: %s\n", err) return false } + defer db.Close() if err = db.Ping(); err != nil { fmt.Printf("connection ping database failed: %s\n", err) return false diff --git a/internal/migrations/migrations.go b/internal/migrations/migrations.go index a4ebeba8..de28c0c5 100644 --- a/internal/migrations/migrations.go +++ b/internal/migrations/migrations.go @@ -116,6 +116,7 @@ func Migrate(debug bool, dbConf *data.Database, cacheConf *data.CacheConf, upgra fmt.Println("new database failed: ", err.Error()) return err } + defer engine.Close() currentDBVersion, err := GetCurrentDBVersion(engine) if err != nil { diff --git a/internal/service/uploader/upload.go b/internal/service/uploader/upload.go index 19bf6322..01abec43 100644 --- a/internal/service/uploader/upload.go +++ b/internal/service/uploader/upload.go @@ -91,18 +91,19 @@ func (us *uploaderService) UploadAvatarFile(ctx *gin.Context) (url string, err e // max size ctx.Request.Body = http.MaxBytesReader(ctx.Writer, ctx.Request.Body, 5*1024*1024) - _, file, err := ctx.Request.FormFile("file") + file, fileHeader, err := ctx.Request.FormFile("file") if err != nil { return "", errors.BadRequest(reason.RequestFormatError).WithError(err) } - fileExt := strings.ToLower(path.Ext(file.Filename)) + file.Close() + fileExt := strings.ToLower(path.Ext(fileHeader.Filename)) if _, ok := FormatExts[fileExt]; !ok { return "", errors.BadRequest(reason.RequestFormatError).WithError(err) } newFilename := fmt.Sprintf("%s%s", uid.IDStr12(), fileExt) avatarFilePath := path.Join(avatarSubPath, newFilename) - return us.uploadFile(ctx, file, avatarFilePath) + return us.uploadFile(ctx, fileHeader, avatarFilePath) } func (us *uploaderService) AvatarThumbFile(ctx *gin.Context, fileName string, size int) (url string, err error) { @@ -170,18 +171,19 @@ func (us *uploaderService) UploadPostFile(ctx *gin.Context) ( // max size ctx.Request.Body = http.MaxBytesReader(ctx.Writer, ctx.Request.Body, 10*1024*1024) - _, file, err := ctx.Request.FormFile("file") + file, fileHeader, err := ctx.Request.FormFile("file") if err != nil { return "", errors.BadRequest(reason.RequestFormatError).WithError(err) } - fileExt := strings.ToLower(path.Ext(file.Filename)) + defer file.Close() + fileExt := strings.ToLower(path.Ext(fileHeader.Filename)) if _, ok := FormatExts[fileExt]; !ok { return "", errors.BadRequest(reason.RequestFormatError).WithError(err) } newFilename := fmt.Sprintf("%s%s", uid.IDStr12(), fileExt) avatarFilePath := path.Join(postSubPath, newFilename) - return us.uploadFile(ctx, file, avatarFilePath) + return us.uploadFile(ctx, fileHeader, avatarFilePath) } func (us *uploaderService) UploadBrandingFile(ctx *gin.Context) ( @@ -196,11 +198,12 @@ func (us *uploaderService) UploadBrandingFile(ctx *gin.Context) ( // max size ctx.Request.Body = http.MaxBytesReader(ctx.Writer, ctx.Request.Body, 10*1024*1024) - _, file, err := ctx.Request.FormFile("file") + file, fileHeader, err := ctx.Request.FormFile("file") if err != nil { return "", errors.BadRequest(reason.RequestFormatError).WithError(err) } - fileExt := strings.ToLower(path.Ext(file.Filename)) + file.Close() + fileExt := strings.ToLower(path.Ext(fileHeader.Filename)) _, ok := FormatExts[fileExt] if !ok && fileExt != ".ico" { return "", errors.BadRequest(reason.RequestFormatError).WithError(err) @@ -208,7 +211,7 @@ func (us *uploaderService) UploadBrandingFile(ctx *gin.Context) ( newFilename := fmt.Sprintf("%s%s", uid.IDStr12(), fileExt) avatarFilePath := path.Join(brandingSubPath, newFilename) - return us.uploadFile(ctx, file, avatarFilePath) + return us.uploadFile(ctx, fileHeader, avatarFilePath) } func (us *uploaderService) uploadFile(ctx *gin.Context, file *multipart.FileHeader, fileSubPath string) (