You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
149 lines
3.3 KiB
149 lines
3.3 KiB
package datatypes
|
|
|
|
import (
|
|
"context"
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
// JSONType give a generic data type for json encoded data.
|
|
type JSONType[T any] struct {
|
|
data T
|
|
}
|
|
|
|
func NewJSONType[T any](data T) JSONType[T] {
|
|
return JSONType[T]{
|
|
data: data,
|
|
}
|
|
}
|
|
|
|
// Data return data with generic Type T
|
|
func (j JSONType[T]) Data() T {
|
|
return j.data
|
|
}
|
|
|
|
// Value return json value, implement driver.Valuer interface
|
|
func (j JSONType[T]) Value() (driver.Value, error) {
|
|
return json.Marshal(j.data)
|
|
}
|
|
|
|
// Scan scan value into JSONType[T], implements sql.Scanner interface
|
|
func (j *JSONType[T]) Scan(value interface{}) error {
|
|
var bytes []byte
|
|
switch v := value.(type) {
|
|
case []byte:
|
|
bytes = v
|
|
case string:
|
|
bytes = []byte(v)
|
|
default:
|
|
return errors.New(fmt.Sprint("Failed to unmarshal JSONB value:", value))
|
|
}
|
|
return json.Unmarshal(bytes, &j.data)
|
|
}
|
|
|
|
// MarshalJSON to output non base64 encoded []byte
|
|
func (j JSONType[T]) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(j.data)
|
|
}
|
|
|
|
// UnmarshalJSON to deserialize []byte
|
|
func (j *JSONType[T]) UnmarshalJSON(b []byte) error {
|
|
return json.Unmarshal(b, &j.data)
|
|
}
|
|
|
|
// GormDataType gorm common data type
|
|
func (JSONType[T]) GormDataType() string {
|
|
return "json"
|
|
}
|
|
|
|
// GormDBDataType gorm db data type
|
|
func (JSONType[T]) GormDBDataType(db *gorm.DB, field *schema.Field) string {
|
|
switch db.Dialector.Name() {
|
|
case "sqlite":
|
|
return "JSON"
|
|
case "mysql":
|
|
return "JSON"
|
|
case "postgres":
|
|
return "JSONB"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (js JSONType[T]) GormValue(ctx context.Context, db *gorm.DB) clause.Expr {
|
|
data, _ := js.MarshalJSON()
|
|
|
|
switch db.Dialector.Name() {
|
|
case "mysql":
|
|
if v, ok := db.Dialector.(*mysql.Dialector); ok && !strings.Contains(v.ServerVersion, "MariaDB") {
|
|
return gorm.Expr("CAST(? AS JSON)", string(data))
|
|
}
|
|
}
|
|
|
|
return gorm.Expr("?", string(data))
|
|
}
|
|
|
|
// JSONSlice give a generic data type for json encoded slice data.
|
|
type JSONSlice[T any] []T
|
|
|
|
func NewJSONSlice[T any](s []T) JSONSlice[T] {
|
|
return JSONSlice[T](s)
|
|
}
|
|
|
|
// Value return json value, implement driver.Valuer interface
|
|
func (j JSONSlice[T]) Value() (driver.Value, error) {
|
|
return json.Marshal(j)
|
|
}
|
|
|
|
// Scan scan value into JSONType[T], implements sql.Scanner interface
|
|
func (j *JSONSlice[T]) Scan(value interface{}) error {
|
|
var bytes []byte
|
|
switch v := value.(type) {
|
|
case []byte:
|
|
bytes = v
|
|
case string:
|
|
bytes = []byte(v)
|
|
default:
|
|
return errors.New(fmt.Sprint("Failed to unmarshal JSONB value:", value))
|
|
}
|
|
return json.Unmarshal(bytes, &j)
|
|
}
|
|
|
|
// GormDataType gorm common data type
|
|
func (JSONSlice[T]) GormDataType() string {
|
|
return "json"
|
|
}
|
|
|
|
// GormDBDataType gorm db data type
|
|
func (JSONSlice[T]) GormDBDataType(db *gorm.DB, field *schema.Field) string {
|
|
switch db.Dialector.Name() {
|
|
case "sqlite":
|
|
return "JSON"
|
|
case "mysql":
|
|
return "JSON"
|
|
case "postgres":
|
|
return "JSONB"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (j JSONSlice[T]) GormValue(ctx context.Context, db *gorm.DB) clause.Expr {
|
|
data, _ := json.Marshal(j)
|
|
|
|
switch db.Dialector.Name() {
|
|
case "mysql":
|
|
if v, ok := db.Dialector.(*mysql.Dialector); ok && !strings.Contains(v.ServerVersion, "MariaDB") {
|
|
return gorm.Expr("CAST(? AS JSON)", string(data))
|
|
}
|
|
}
|
|
|
|
return gorm.Expr("?", string(data))
|
|
}
|