新增：强制走索引支持

farseer-go · Jul 7, 2024 · 4dd5026 · 4dd5026
1 parent 4cb7a2a
commit 4dd5026
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tableSet.go b/tableSet.go
@@ -273,13 +273,17 @@ func (receiver *TableSet[Table]) Omit(columns ...string) *TableSet[Table] {
 }
 
 // ForceIndex 强制使用索引
-func (receiver *TableSet[Table]) ForceIndex(idxName string) {
-	receiver.forceIndexName = idxName
+func (receiver *TableSet[Table]) ForceIndex(idxName string) *TableSet[Table] {
+	session := receiver.getOrCreateSession()
+	session.forceIndexName = idxName
+	return session
 }
 
 // UseIndex 推荐使用索引
-func (receiver *TableSet[Table]) UseIndex(idxName string) {
-	receiver.useIndexName = idxName
+func (receiver *TableSet[Table]) UseIndex(idxName string) *TableSet[Table] {
+	session := receiver.getOrCreateSession()
+	session.useIndexName = idxName
+	return session
 }
 
 // Where 条件