support json field

Author: folospace

orm/create_table.go

@@ -342,7 +342,7 @@ func getMigrateColumns(table *queryTable) []dBColumn {
             }
         }
 
-        if column.Default == "" || SliceContain(definedDefault, strings.ToLower(column.Default)) < 0 {
+        if column.Default == "" || sliceContain(definedDefault, strings.ToLower(column.Default)) == false {
             column.Default = "'" + column.Default + "'"
         }
 

orm/json_field.go

@@ -0,0 +1,48 @@
+package orm
+
+import (
+    "database/sql/driver"
+    "encoding/json"
+    "unsafe"
+)
+
+//json ojbect/slice <=> go struct/slice <=> db json string
+type JsonField[T any] struct {
+    Data T
+}
+
+func NewJsonField[T any](data T) JsonField[T] {
+    return JsonField[T]{Data: data}
+}
+
+func (t JsonField[T]) MarshalJSON() ([]byte, error) {
+    return json.Marshal(t.Data)
+}
+
+func (t *JsonField[T]) UnmarshalJSON(data []byte) error {
+    return json.Unmarshal(data, &t.Data)
+}
+
+func (t JsonField[T]) Value() (driver.Value, error) {
+    data, err := json.Marshal(t.Data)
+    return *(*string)(unsafe.Pointer(&data)), err
+}
+
+func (t *JsonField[T]) Scan(raw any) error {
+    rawData, ok := raw.([]byte)
+    if !ok {
+        return nil
+    }
+
+    if len(rawData) == 0 {
+        return nil
+    }
+
+    var i T
+    err := json.Unmarshal(rawData, &i)
+    if err != nil {
+        return err
+    }
+    t.Data = i
+    return nil
+}

orm/json_fields.go

@@ -7,39 +7,40 @@ import (
     "strings"
 )
 
+//json int or long string int <=> go int64
 type JsonInt int64
 
-func (i JsonInt) MarshalJSON() ([]byte, error) {
-    return []byte("\"" + strconv.FormatInt(int64(i), 10) + "\""), nil
+func (t JsonInt) MarshalJSON() ([]byte, error) {
+    return []byte("\"" + strconv.FormatInt(int64(t), 10) + "\""), nil
 }
 
-func (i *JsonInt) UnmarshalJSON(data []byte) error {
+func (t *JsonInt) UnmarshalJSON(data []byte) error {
     val, err := strconv.ParseInt(strings.Trim(string(data), "\""), 10, 64)
     if err != nil {
         return err
     }
-    *i = JsonInt(val)
+    *t = JsonInt(val)
     return nil
 }
 
-func (i JsonInt) Value() (driver.Value, error) {
-    return int64(i), nil
+func (t JsonInt) Value() (driver.Value, error) {
+    return int64(t), nil
 }
 
-func (i JsonInt) ToString() string {
-    return strconv.FormatInt(int64(i), 10)
+func (t JsonInt) ToString() string {
+    return strconv.FormatInt(int64(t), 10)
 }
 
-func (i *JsonInt) Scan(v any) error {
+func (t *JsonInt) Scan(v any) error {
     val, ok := v.(int64)
     if ok {
-        *i = JsonInt(val)
+        *t = JsonInt(val)
         return nil
     } else {
         val, ok := v.([]uint8)
         if ok {
             v, _ := strconv.ParseInt(string(val), 10, 64)
-            *i = JsonInt(v)
+            *t = JsonInt(v)
             return nil
         }
     }

orm/json_int.go

@@ -7,6 +7,7 @@ import (
     "time"
 )
 
+//json int(13) <=> go time.time
 type JsonTime struct {
     time.Time
 }