added the ability to expand the models

logic/utils/utils.go

@@ -189,6 +189,7 @@ type JustId struct { Id string }
 type Generic struct{ reflect.Type }
 
 var NotFoundError = errors.New("Not found")
+var CouldNotInsert = errors.New("Could not insert")
 
 func generateQuery(t reflect.Type) (query string, nargs int) {
 	nargs = t.NumField()
@@ -200,6 +201,10 @@ func generateQuery(t reflect.Type) (query string, nargs int) {
         if !ok {
             name = field.Name;
         }
+
+        if name == "__nil__" {
+            continue
+        }
         query += strings.ToLower(name) + ","
 	}
     
@@ -214,7 +219,13 @@ func GetDbMultitple[T interface{}](c *Context, tablename string, args ...any) ([
 
     query, nargs := generateQuery(t)
 
-	rows, err := c.Db.Query(fmt.Sprintf("select %s from %s", query, tablename), args...)
+    db_query, err := c.Prepare(fmt.Sprintf("select %s from %s", query, tablename))
+    if err != nil {
+        return nil, err
+    }
+    defer db_query.Close()
+
+	rows, err := db_query.Query(args...)
 	if err != nil {
 		return nil, err
 	}
@@ -251,6 +262,43 @@ func mapRow(store interface{}, rows *sql.Rows, nargs int) (err error) {
     return nil
 }
 
+func InsertReturnId(c *Context, store interface{}, tablename string, returnName string) (id string, err error) {
+    t := reflect.TypeOf(store).Elem()
+
+    query, nargs := generateQuery(t)
+
+    query2 := ""
+    for i := 0; i < nargs; i += 1 {
+        query2 += fmt.Sprintf("$%d,", i)
+    }
+    // Remove last quotation
+	query2 = query2[0 : len(query2)-1]
+
+    val := reflect.ValueOf(store).Elem()
+	scan_args := make([]interface{}, nargs);
+    for i := 0; i < nargs; i++ {
+        valueField := val.Field(i)
+        scan_args[i] = valueField.Interface()
+    }
+
+	rows, err := c.Db.Query(fmt.Sprintf("insert into %s (%s) values (%s) returning %s", tablename, query, query2, returnName), scan_args...)
+	if err != nil {
+		return
+	}
+	defer rows.Close()
+
+	if !rows.Next() {
+		return "", CouldNotInsert
+	}
+
+	err = rows.Scan(&id)
+	if err != nil {
+		return
+	}
+
+    return
+}
+
 func GetDBOnce(c *Context, store interface{}, tablename string, args ...any) error {
     t := reflect.TypeOf(store).Elem()