php小編香蕉介紹:在Gorm中,當你嘗試在自定義的字符串類型上進行掃描操作時,可能會遇到錯誤。這個問題可能會導致掃描儀無法正確解析字符串,從而導致程序出錯。這是因為Gorm默認使用`Scan`方法來掃描字符串類型的字段,但對于自定義的字符串類型,`Scan`方法可能無法正確處理。解決這個問題的方法是使用`Value`方法來手動解析字符串,確保程序能夠正確執行。通過這種方式,你可以避免在使用Gorm時遇到關于掃描儀錯誤的問題。
問題內容
我編寫了以下實體:
type datacategory string
const (
datacategory1 datacategory = "category1"
datacategory2 datacategory = "category2"
)
type data struct {
name string `json:"name"`
categories []datacategory `json:"category" gorm:"type:text[]"`
}
登錄后復制
我在數據庫中手動創建了一行,并用 category1 和 category2 填充了類別數組類型。
但是讀取數據時會出現以下錯誤:
sql: scan error on column index 19, name "category": unsupported scan, storing driver.value type string into type *[]datacategory
登錄后復制
示例值方法:
func (s DataCategorySlice) Value() (driver.Value, error) {
if s == nil {
return nil, nil
}
if len(s) == 0 {
return "{}", nil
}
v := []string{}
for _, dc := range s {
v = append(v, string(dc))
}
result := fmt.Sprintf("{%s}", strings.Join(v, ","))
return result, nil
}
登錄后復制
解決方法
以下示例假設您使用 postgresql 作為 rdbms,并且 datacategory 值不包含逗號或未轉義的單引號。
// declare the custom type type datacategoryslice []datacategory
登錄后復制
// implement driver.valuer to encode the value into the
// correct format that is accepted by the target rdbms
func (s datacategoryslice) value() (driver.value, error) {
if s == nil {
return nil, nil
}
if len(s) == 0 {
return []byte(`{}`), nil
}
v := []byte(`{`)
for i := range s {
v = append(v, s[i]...)
v = append(v, ',')
}
v[len(v)-1] = '}' // replace last comma with closing brace
return v, nil
}
登錄后復制
// implement scanner to decode the raw source
// value as retrieved from the database
func (s *datacategoryslice) scan(src any) error {
var data []byte
switch v := src.(type) {
case []byte:
data = v
case string:
data = []byte(v)
case nil:
return nil
default:
return fmt.errorf("unsupported type: %t", src)
}
if len(data) == 0 {
return nil
}
data = data[1:len(data)-1] // remove surrounding braces
for _, v := range bytes.split(data, []byte{','}) {
*s = append(*s, datacategory(v))
}
return nil
}
登錄后復制
或者,如果從 value() (driver.value, error) 返回 []byte 不起作用,例如它會導致 “錯誤:格式錯誤的數組文字: (sqlstate 22p02)”,那么您可以嘗試使用 string 作為返回類型。
來自@kozhioyrin的示例
// implement driver.Valuer to encode the value into the
// correct format that is accepted by the target RDBMS
func (s DataCategorySlice) Value() (driver.Value, error) {
if s == nil {
return nil, nil
}
if len(s) == 0 {
return "{}", nil
}
v := []string{}
for _, dc := range s {
v = append(v, string(dc))
}
result := fmt.Sprintf("{%s}", strings.Join(v, ","))
return result, nil
}
登錄后復制






