get slice

Author: folospace

orm/query_delete.go

@@ -5,10 +5,11 @@ func (q *Query[T]) Delete(primaryIds ...interface{}) QueryResult {
         q.setErr(ErrTableNotSelected)
         return q.result
     }
-    if len(primaryIds) > 0 {
-        return q.WherePrimary(primaryIds).delete()
+
+    if len(primaryIds) == 1 {
+        return q.WherePrimary(primaryIds[0]).delete()
     } else {
-        return q.delete()
+        return q.WherePrimary(primaryIds).delete()
     }
 }
 

orm/query_get.go

@@ -9,7 +9,12 @@ import (
 //get first T
 func (q *Query[T]) Get(primaryIds ...interface{}) (T, QueryResult) {
     var ret T
-    res := q.WherePrimary(primaryIds).Limit(1).GetTo(&ret)
+    var res QueryResult
+    if len(primaryIds) == 1 {
+        res = q.WherePrimary(primaryIds[0]).Limit(1).GetTo(&ret)
+    } else {
+        res = q.WherePrimary(primaryIds).Limit(1).GetTo(&ret)
+    }
     return ret, res
 }
 
@@ -37,7 +42,12 @@ func (q *Query[T]) GetString() (string, QueryResult) {
 //get slice T
 func (q *Query[T]) Gets(primaryIds ...interface{}) ([]T, QueryResult) {
     var ret []T
-    res := q.WherePrimary(primaryIds).GetTo(&ret)
+    var res QueryResult
+    if len(primaryIds) == 1 {
+        res = q.WherePrimary(primaryIds[0]).GetTo(&ret)
+    } else {
+        res = q.WherePrimary(primaryIds).GetTo(&ret)
+    }
     return ret, res
 }