From 1fab120c67cbd743f00ec4b6b4125e4fcfdfa3b8 Mon Sep 17 00:00:00 2001
From: Song Gao <disxiaofei@163.com>
Date: Tue, 16 Apr 2024 18:25:43 +0800
Subject: [PATCH] fix: fix index store panic error (#2776)

Signed-off-by: yisaer <disxiaofei@163.com>
---
 extensions/sqldatabase/sqlgen/sqlServerDialect_test.go   | 8 ++++++++
 extensions/sqldatabase/sqlgen/sqlgen.go                  | 4 +---
 extensions/sqldatabase/sqlgen/templateSqlDialect.go      | 4 +---
 extensions/sqldatabase/sqlgen/templateSqlDialect_test.go | 7 +++++++
 4 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/extensions/sqldatabase/sqlgen/sqlServerDialect_test.go b/extensions/sqldatabase/sqlgen/sqlServerDialect_test.go
index e9a47c0c31..c4c3f4a6e7 100644
--- a/extensions/sqldatabase/sqlgen/sqlServerDialect_test.go
+++ b/extensions/sqldatabase/sqlgen/sqlServerDialect_test.go
@@ -202,6 +202,14 @@ func TestGenerateSQLWithMultiIndex(t *testing.T) {
 			},
 			sql: `select * from t where col2 > '2' AND col1 > '1' order by 'col2' ASC, 'col1' ASC limit 3`,
 		},
+		{
+			cfg: &InternalSqlQueryCfg{
+				Table: "t",
+				Limit: 3,
+				store: store.NewIndexFieldWrap(),
+			},
+			sql: `select * from t  limit 3`,
+		},
 	}
 
 	for _, tc := range testcases {
diff --git a/extensions/sqldatabase/sqlgen/sqlgen.go b/extensions/sqldatabase/sqlgen/sqlgen.go
index 41744ebb52..0d5e9b1fd9 100644
--- a/extensions/sqldatabase/sqlgen/sqlgen.go
+++ b/extensions/sqldatabase/sqlgen/sqlgen.go
@@ -58,9 +58,7 @@ func (i *InternalSqlQueryCfg) InitIndexFieldStore() {
 		i.store.Init(f)
 		return
 	}
-	if len(i.IndexFields) > 0 {
-		i.store.Init(i.IndexFields...)
-	}
+	i.store.Init(i.IndexFields...)
 }
 
 func (i *InternalSqlQueryCfg) SetIndexValue(v interface{}) {
diff --git a/extensions/sqldatabase/sqlgen/templateSqlDialect.go b/extensions/sqldatabase/sqlgen/templateSqlDialect.go
index 61a2e9d86e..2c51e75757 100644
--- a/extensions/sqldatabase/sqlgen/templateSqlDialect.go
+++ b/extensions/sqldatabase/sqlgen/templateSqlDialect.go
@@ -113,9 +113,7 @@ func (t *TemplateSqlQueryCfg) InitIndexFieldStore() {
 		t.store.Init(f)
 		return
 	}
-	if len(t.IndexFields) > 0 {
-		t.store.Init(t.IndexFields...)
-	}
+	t.store.Init(t.IndexFields...)
 }
 
 func (t *TemplateSqlQueryCfg) SetIndexValue(v interface{}) {
diff --git a/extensions/sqldatabase/sqlgen/templateSqlDialect_test.go b/extensions/sqldatabase/sqlgen/templateSqlDialect_test.go
index 2c35dba6bc..f0382a1db1 100644
--- a/extensions/sqldatabase/sqlgen/templateSqlDialect_test.go
+++ b/extensions/sqldatabase/sqlgen/templateSqlDialect_test.go
@@ -220,6 +220,13 @@ func TestGenerateTemplateWithMultiIndex(t *testing.T) {
 			},
 			sql: "select * from table where col1 > `1` AND col2 > `2` order by co1 ASC, co2 ASC",
 		},
+		{
+			cfg: &TemplateSqlQueryCfg{
+				TemplateSql: "select * from table",
+				store:       store.NewIndexFieldWrap(),
+			},
+			sql: "select * from table",
+		},
 	}
 
 	for _, tc := range testcases {