add Scan

NekoWheel · Jan 12, 2024 · 9141394 · 9141394
1 parent 9b7df54
commit 9141394
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/internal/dbutil/type.go b/internal/dbutil/type.go
@@ -5,6 +5,8 @@
 package dbutil
 
 import (
+	"database/sql"
+
 	"gorm.io/gorm"
 	"gorm.io/gorm/schema"
 )
@@ -21,6 +23,15 @@ func (ContentCensorPass) GormDBDataType(db *gorm.DB, field *schema.Field) string
 	return ""
 }
 
+func (c *ContentCensorPass) Scan(value interface{}) error {
+	var i sql.NullBool
+	if err := i.Scan(value); err != nil {
+		return err
+	}
+	*c = ContentCensorPass(i.Bool)
+	return nil
+}
+
 type AnswerCensorPass bool
 
 func (AnswerCensorPass) GormDBDataType(db *gorm.DB, field *schema.Field) string {
@@ -32,3 +43,12 @@ func (AnswerCensorPass) GormDBDataType(db *gorm.DB, field *schema.Field) string
 	}
 	return ""
 }
+
+func (c *AnswerCensorPass) Scan(value interface{}) error {
+	var i sql.NullBool
+	if err := i.Scan(value); err != nil {
+		return err
+	}
+	*c = AnswerCensorPass(i.Bool)
+	return nil
+}