partition

Author: folospace

orm/create_table.go

@@ -414,7 +414,7 @@ func getTypeAndDefault(val reflect.Value) (string, string) {
             types = "datetime"
         } else if _, ok := val.Interface().(time.Time); ok {
             types = "datetime"
-        } else {
+        } else if kind == reflect.Struct {
             realVal := reflect.New(typ)
             if _, ok := realVal.Elem().Field(0).Interface().(*time.Time); ok {
                 types = "datetime"
@@ -423,6 +423,8 @@ func getTypeAndDefault(val reflect.Value) (string, string) {
             } else {
                 types = "varchar(255)"
             }
+        } else {
+            types = "varchar(255)"
         }
         if types == "datetime" {
             defaults = "current_timestamp"

orm/query.go

@@ -324,6 +324,10 @@ func (q *Query[T]) OrHaving(column any, vals ...any) *Query[T] {
     }
     return newQuery
 }
+func (q *Query[T]) Partition(p string) *Query[T] {
+    q.tables[0].partition = p
+    return q
+}
 
 func (q *Query[T]) PartitionBy(column any) *Query[T] {
     val, err := q.parseColumn(column)

orm/query_insert.go

@@ -179,7 +179,7 @@ func (q *Query[T]) insert(data any) QueryResult {
         rawSql += " ignore"
     }
 
-    rawSql += " into " + q.tables[0].getTableName() + " " + insertSql
+    rawSql += " into " + q.tables[0].getTableNamePartition() + " " + insertSql
 
     if updateStr != "" {
         rawSql += " on duplicate key update " + updateStr

orm/query_table.go

@@ -14,6 +14,7 @@ type queryTable struct {
     joinCondition     where
     alias             string
     overrideTableName string //override table name
+    partition         string
     rawSql            string
     bindings          []any
 }
@@ -31,7 +32,7 @@ func (q queryTable) getAliasOrTableName() string {
 
 func (q queryTable) getTableNameAndAlias() string {
     var strs []string
-    temp := q.getTableName()
+    temp := q.getTableNamePartition()
     if temp != "" {
         strs = append(strs, temp)
     }
@@ -56,6 +57,27 @@ func (q queryTable) getTableName() string {
     return ""
 }
 
+func (q queryTable) getTableNamePartition() string {
+    if q.overrideTableName != "" {
+        return q.overrideTableName
+    }
+    if q.table.TableName() != "" {
+        if q.table.DatabaseName() != "" {
+            return q.table.DatabaseName() + "." + q.table.TableName() + q.getPartition()
+        } else {
+            return q.table.TableName() + q.getPartition()
+        }
+    }
+    return ""
+}
+
+func (q queryTable) getPartition() string {
+    if q.partition == "" {
+        return ""
+    }
+    return " partition(" + q.partition + ")"
+}
+
 func (q queryTable) getTags(index int, tagName string) []string {
     tags := strings.Split(q.tableStructType.Field(index).Tag.Get(tagName), ",")
     return tags