日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后，達者為師

網站首頁 編程語言 正文

Go語言中GORM存取數組/自定義類型數據_Golang

作者：Go~Go~Go~ ? 更新時間： 2023-03-13 編程語言

GORM存取數組類型數據

在GORM不支持數組類型變量數據的存取，可以通過GROM的自定義數據類型實現。

一、GORM的自定義類型

1. Scanner/Valuer接口

? GORM的自定義的數據類型必須實現Scanner/Valuer接口。

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

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

2. 數組變量

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

(1) 實現Sacnner接口

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

type Strs []string

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

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

(2) 實現Valuer接口

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

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

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

3. 測試

定義如下函數進行測試

(1) 模型變量

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

? (2) Service函數

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函數

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) 運行結果

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

在數據庫存儲中體現為：

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

二、實際生產

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

(1) 實現自定義類型

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) 數據表模型

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

?總結

原文鏈接：https://blog.csdn.net/js010111/article/details/126076320

欄目分類
最近更新