feat: 自动更新数据库的拓展参数

DokiDoki1103 · Jun 27, 2024 · 51c314c · 51c314c
1 parent 4666b77
commit 51c314c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/internal/controller/search.go b/internal/controller/search.go
@@ -23,6 +23,7 @@ func Search(c *gin.Context) {
 		c.JSON(http.StatusBadRequest, global.ErrorParam)
 		return
 	}
+	req.Extra = c.Query("extra")
 
 	var result [][]string                          // 最后所有的答案的二维数组
 	var localAnswer [][]string                     // 本地答案

diff --git a/internal/search/db.go b/internal/search/db.go
@@ -7,6 +7,7 @@ import (
 	"github.com/itihey/tikuAdapter/internal/dao"
 	"github.com/itihey/tikuAdapter/pkg/model"
 	"github.com/itihey/tikuAdapter/pkg/util"
+	"log"
 )
 
 // DB mysql 或者sqlite3
@@ -39,6 +40,13 @@ func (in *dBSearch) SearchAnswer(req model.SearchRequest) (answer [][]string, er
 		}
 		find = find2
 	}
+	//如果数据库中没有extra那么自动补全他
+	if len(find) == 1 && find[0].Extra == "" {
+		_, err1 := tiku.Where(tiku.ID.Eq(find[0].ID)).Update(tiku.Extra, req.Extra)
+		if err1 != nil {
+			log.Println("更新extra失败", err1)
+		}
+	}
 	for i := range find {
 		var answers []string // 最后所有的答案的二维数组
 		err := json.Unmarshal([]byte(find[i].Answer), &answers)

diff --git a/pkg/model/search.go b/pkg/model/search.go
@@ -7,6 +7,8 @@ type SearchRequest struct {
 	Question string   `json:"question"`
 	Options  []string `json:"options"`
 	Type     uint     `json:"type"`
+
+	Extra string `json:"extra"`
 }
 
 // SearchResponse 搜题响应体