dbutil: add censor pass type

NekoWheel · Jan 10, 2024 · ee10847 · ee10847
1 parent d038028
commit ee10847
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 14 deletions.
diff --git a/internal/cmd/censor.go b/internal/cmd/censor.go
@@ -31,7 +31,7 @@ func runCensor(ctx *cli.Context) error {
 		return errors.New("text censor is disabled")
 	}
 
-	dsn := fmt.Sprintf("%s:%s@%s:%s/%s?charset=utf8mb4&parseTime=True&loc=Local",
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
 		conf.Database.User,
 		conf.Database.Password,
 		conf.Database.Host,

diff --git a/internal/cmd/web.go b/internal/cmd/web.go
@@ -51,7 +51,7 @@ func runWeb(ctx *cli.Context) error {
 	var dsn string
 	switch dbType {
 	case "mysql", "":
-		dsn = fmt.Sprintf("%s:%s@%s:%s/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		dsn = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
 			conf.Database.User,
 			conf.Database.Password,
 			conf.Database.Host,

diff --git a/internal/db/questions.go b/internal/db/questions.go
@@ -43,18 +43,18 @@ type questions struct {
 
 type Question struct {
 	dbutil.Model
-	FromIP                string         `json:"-"`
-	UserID                uint           `gorm:"index:idx_question_user_id" json:"-"`
-	Content               string         `json:"content"`
-	ContentCensorMetadata datatypes.JSON `json:"-"`
-	ContentCensorPass     bool           `gorm:"->;type:boolean GENERATED ALWAYS AS (IFNULL(content_censor_metadata->'$.pass' = true, false)) STORED NOT NULL" json:"-"`
-	Token                 string         `json:"-"`
-	Answer                string         `json:"answer"`
-	AnswerCensorMetadata  datatypes.JSON `json:"-"`
-	AnswerCensorPass      bool           `gorm:"->;type:boolean GENERATED ALWAYS AS (IFNULL(answer_censor_metadata->'$.pass' = true, false)) STORED NOT NULL" json:"-"`
-	ReceiveReplyEmail     string         `json:"-"`
-	AskerUserID           uint           `json:"-"`
-	IsPrivate             bool           `gorm:"default: FALSE; NOT NULL" json:"-"`
+	FromIP                string                   `json:"-"`
+	UserID                uint                     `gorm:"index:idx_question_user_id" json:"-"`
+	Content               string                   `json:"content"`
+	ContentCensorMetadata datatypes.JSON           `json:"-"`
+	ContentCensorPass     dbutil.ContentCensorPass `json:"-"`
+	Token                 string                   `json:"-"`
+	Answer                string                   `json:"answer"`
+	AnswerCensorMetadata  datatypes.JSON           `json:"-"`
+	AnswerCensorPass      dbutil.AnswerCensorPass  `json:"-"`
+	ReceiveReplyEmail     string                   `json:"-"`
+	AskerUserID           uint                     `json:"-"`
+	IsPrivate             bool                     `gorm:"default: FALSE; NOT NULL" json:"-"`
 }
 
 type CreateQuestionOptions struct {

diff --git a/internal/dbutil/type.go b/internal/dbutil/type.go
@@ -0,0 +1,34 @@
+// Copyright 2024 E99p1ant. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package dbutil
+
+import (
+	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
+)
+
+type ContentCensorPass bool
+
+func (ContentCensorPass) GormDBDataType(db *gorm.DB, field *schema.Field) string {
+	switch db.Dialector.Name() {
+	case "mysql":
+		return "boolean GENERATED ALWAYS AS (IFNULL(content_censor_metadata->'$.pass' = true, false)) STORED NOT NULL"
+	case "postgres":
+		return "BOOLEAN GENERATED ALWAYS AS (COALESCE(content_censor_metadata->>'$.pass' = 'true', false)) STORED"
+	}
+	return ""
+}
+
+type AnswerCensorPass bool
+
+func (AnswerCensorPass) GormDBDataType(db *gorm.DB, field *schema.Field) string {
+	switch db.Dialector.Name() {
+	case "mysql":
+		return "boolean GENERATED ALWAYS AS (IFNULL(answer_censor_metadata->'$.pass' = true, false)) STORED NOT NULL"
+	case "postgres":
+		return "BOOLEAN GENERATED ALWAYS AS (COALESCE(answer_censor_metadata->>'$.pass' = 'true', false)) STORED"
+	}
+	return ""
+}