gorm实现存入数组或json

package main

import (
	"database/sql/driver"
	"encoding/json"
	"errors"
	"fmt"
	"gorm.io/gorm"
	"strings"
)

type JSON struct {
	Name string `json:"name"`
}

// 实现 sql.Scanner 接口,Scan 将 value 扫描至 Jsonb
func (j *JSON) Scan(value interface{}) error {
	bytes, ok := value.([]byte)
	if !ok {
		return errors.New(fmt.Sprint("Failed to unmarshal JSONB value:", value))
	}
	return json.Unmarshal(bytes, &j)
}

// 实现 driver.Valuer 接口,Value 返回 json value
func (j JSON) Value() (driver.Value, error) {
	str, err := json.Marshal(j)
	if err != nil {
		return nil, err
	}
	return string(str), nil
}

type Array []string

// 实现 sql.Scanner 接口,Scan 将 value 扫描至 Jsonb
func (a *Array) Scan(value interface{}) error {
	bytes, ok := value.([]byte)
	if !ok {
		return errors.New(fmt.Sprint("Failed to scan Array value:", value))
	}
	*a = strings.Split(string(bytes), ",")
	return nil
}

// 实现 driver.Valuer 接口,Value 返回 json value
func (a Array) Value() (driver.Value, error) {
	if len(a) > 0 {
		var str string = a[0]
		for _, v := range a[1:] {
			str += "," + v
		}
		return str,nil
	} else {
		return "",nil
	}
}

var GDB *gorm.DB

func main() {
	GDB = getDB()
	//testJson()
	testArray()
}

func testJson() {
	type User struct {
		Name JSON
	}
	GDB.AutoMigrate(&User{})
	user := User{
		Name: JSON{
			Name: "lala",
		},
	}
	GDB.Create(&user)
	var fUser User
	GDB.Model(&user).Find(&fUser)
	fmt.Println(fUser)
}

func testArray() {
	type User struct {
		Name Array
	}
	GDB.AutoMigrate(&User{})
	user := User{
		Name: []string{
			"lala",
			"tata",
		},
	}
	GDB.Create(&user)
	var fUser User
	GDB.Model(&user).Find(&fUser)
	fmt.Println(fUser)
}

posted @   burndust  阅读(2101)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
点击右上角即可分享
微信分享提示