From c2d084a788b2d565e0b22d8456cb4233953a43e3 Mon Sep 17 00:00:00 2001
From: kvii2202 <13721952202@163.com>
Date: Wed, 16 Mar 2022 20:44:43 +0800
Subject: [PATCH] fix args length error.

---
 chainable_api.go | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/chainable_api.go b/chainable_api.go
index aff735c6..f02db47d 100644
--- a/chainable_api.go
+++ b/chainable_api.go
@@ -124,13 +124,23 @@ func (db *DB) with(name string, recursive bool, args ...interface{}) (tx *DB) {
 			}
 		}
 	case []string:
-		for _, col := range arg {
-			if strings.Contains(col, ",") {
-				tx.AddError(fmt.Errorf("unsupported mixed string type"))
-				return
+		switch len(args) {
+		case 1:
+			tx.AddError(fmt.Errorf("unsupported with subquery type %T", arg))
+			return
+		case 2:
+			// arg[0] is columns ans arg[1] is subquery
+			for _, col := range arg {
+				if strings.Contains(col, ",") {
+					tx.AddError(fmt.Errorf("unsupported mixed string type"))
+					return
+				}
 			}
+			columns = arg
+		default:
+			tx.AddError(fmt.Errorf("unsupported multiple columns %v", args[1:len(args)-1]))
+			return
 		}
-		columns = arg
 	default:
 		// specify optional columns field with a wrong type
 		if len(args) > 1 {