Go語言中GORM存取數(shù)組/自定義類型數(shù)據(jù)

更新時間：2023年01月17日 09:28:09 作者：Go~Go~Go~

在使用gorm時往往默認的數(shù)據(jù)類型不滿足我們的要求,需要使用一些自定義數(shù)據(jù)類型作為字段類型,下面這篇文章主要給大家介紹了關(guān)于Go語言中GORM存取數(shù)組/自定義類型數(shù)據(jù)的相關(guān)資料,需要的朋友可以參考下

GORM存取數(shù)組類型數(shù)據(jù)

在GORM不支持數(shù)組類型變量數(shù)據(jù)的存取，可以通過GROM的自定義數(shù)據(jù)類型實現(xiàn)。

一、GORM的自定義類型

1. Scanner/Valuer接口

? GORM的自定義的數(shù)據(jù)類型必須實現(xiàn)Scanner/Valuer接口。

(1) Scanner接口的Scan方法，是從數(shù)據(jù)庫讀取數(shù)據(jù)到Go變量時需要進行的解析處理，與解碼的過程類型。

(2) Valuer接口的Value方法，是將Go變量存到數(shù)據(jù)庫時進行編碼處理。

2. 數(shù)組變量

? 雖然GORM不支持數(shù)組變量，但是可以通過實現(xiàn)Scanner/Valuer接口，在數(shù)據(jù)庫取/存時進行類似解碼/編碼的處理，使數(shù)組變量成為數(shù)據(jù)庫可支持的變量類型。

(1) 實現(xiàn)Sacnner接口

Scan函數(shù)，從數(shù)據(jù)庫讀取數(shù)據(jù)后，對其進行處理，然后獲得Go類型的變量。

type Strs []string

func (m *Strs) Scan(val interface{}) error {
	s := val.([]uint8)
	ss := strings.Split(string(s), "|")
	*m = ss
	return nil
}

如上自定義Strs類型，底層類型未字符串?dāng)?shù)組。Scan函數(shù)將讀取的val值，按照字符串處理，以"|"為分隔符進行切割，獲得字符串?dāng)?shù)組類型，然后使用指針賦值。

(2) 實現(xiàn)Valuer接口

Value函數(shù)，將數(shù)據(jù)存到數(shù)據(jù)庫時，對數(shù)據(jù)進行處理，獲得數(shù)據(jù)庫支持的類型。

func (m Strs) Value() (driver.Value, error) {
	str := strings.Join(m, "|")
	return str, nil
}

如上將Strs存入數(shù)據(jù)庫前，將數(shù)組內(nèi)的字符串用"|"拼接，獲得數(shù)據(jù)庫支持的string類型后再存入數(shù)據(jù)庫。

3. 測試

定義如下函數(shù)進行測試

(1) 模型變量

type User struct {
	ID   uint `gorm:"primary_key"`
	Name string
	Pics Strs `gorm:"type:longtext"`  // 自定義的數(shù)組類型，在數(shù)據(jù)庫中存為長文本類型
}

? (2) Service函數(shù)

func SaveUser(user User) error {
	var err error
	err = db.Save(&user).Error
	return err
}

func GetUser(name string) (User, error) {
	var user User
	err := db.First(&user, "name = ?", name).Error
	return user, err
}

? (3) testService函數(shù)

func TestSaveUser() {
   user := User{
      Name: "Jason",
      Pics: Strs{
         "123124",
         "gtsrbxrzsfcv",
      },
   }
   err := SaveUser(user)
   if err != nil {
      log.Panicln("保存失??！", err)
   } else {
      log.Println("保存成功！")
   }
}

func TestGetUser() {
   user, err := GetUser("Jason")
   if err != nil {
      log.Panicln("獲取失??！", err)
   } else {
      log.Println("獲取成功！", user)
   }
}

? (4) 運行結(jié)果

2022/07/30 18:18:41 保存成功！
2022/07/30 18:18:41 獲取成功！ {3 Jason [123124 gtsrbxrzsfcv]}

在數(shù)據(jù)庫存儲中體現(xiàn)為：

mysql> select *from users;
+----+-------+---------------------+
| id | name  | pics                |
+----+-------+---------------------+
|  3 | Jason | 123124|gtsrbxrzsfcv |
+----+-------+---------------------+
1 row in set (0.00 sec)

二、實際生產(chǎn)

上面的例子中展示了Scanner/Valuer接口的簡單使用，實際生產(chǎn)中將數(shù)組類型轉(zhuǎn)換為json字符串進行保存，下面展示一個完整的例子，以一個User擁有一張包含信息的銀行卡為例（當(dāng)然可以通過一對一關(guān)聯(lián)4444實現(xiàn)）。

(1) 實現(xiàn)自定義類型

type User struct {
   ID    uint `gorm:"primary_key"`
   Name  string
   Cards Card `gorm:"json"`
}

type Card struct {  // 指定json的Tag。
   Type     int    `json:"type"`
   Account  string `json:"account"`
   Password string `json:"password"`
}

// Scan 解碼json字符串
func (card *Card) Scan(val interface{}) error {
   b, _ := val.([]byte)
   return json.Unmarshal(b, card)
}

// Value 編碼json
func (card Card) Value() (value driver.Value, err error) {
	return json.Marshal(card)
}

? (2) 數(shù)據(jù)表模型

type User struct {
   ID    uint `gorm:"primary_key"`
   Name  string
   Cards Card `gorm:"json"`  // 指定為json類型
}

?總結(jié)

到此這篇關(guān)于Go語言中GORM存取數(shù)組/自定義類型數(shù)據(jù)的文章就介紹到這了,更多相關(guān)GORM存取數(shù)組自定義類型數(shù)據(jù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: