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 }
登錄后復制