zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-12 03:33:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: add context, part5 (#17392)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-07-27 10:20:16 +08:00
committed by GitHub
parent 4b39106cd4
commit f53518c110
182 changed files with 1249 additions and 1066 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/engine/infinity/metadata.go
View File

@@ -29,6 +29,7 @@ import (
"ragflow/internal/utility"
infinity "github.com/infiniflow/infinity-go-sdk"
"gorm.io/gorm"
"go.uber.org/zap"
)
@@ -794,7 +795,7 @@ const metaPushdownMaxSize = 10000
// nil -> push-down was not viable / errored / result overflowed the
// push-down cap (caller should fall back to in-memory)
// []string{} -> push-down succeeded but found 0 matching docs (empty result is definitive)
func (e *infinityEngine) FilterDocIdsByMetaPushdown(ctx context.Context, kbIDs []string, conditions []map[string]interface{}, logic string) []string {
func (e *infinityEngine) FilterDocIdsByMetaPushdown(ctx context.Context, sqlDB *gorm.DB, kbIDs []string, conditions []map[string]interface{}, logic string) []string {
if len(conditions) == 0 || len(kbIDs) == 0 {
return nil
}
@@ -806,7 +807,7 @@ func (e *infinityEngine) FilterDocIdsByMetaPushdown(ctx context.Context, kbIDs [
}
// Get tenant ID from first KB
tenantID, err := dao.GetTenantIDByKBID(kbIDs[0])
tenantID, err := dao.GetTenantIDByKBID(ctx, sqlDB, kbIDs[0])
if err != nil {
common.Warn("FilterDocIdsByMetaPushdown: failed to get tenant for KB", zap.String("kbID", kbIDs[0]), zap.Error(err))
return nil
@@ -836,13 +837,13 @@ func (e *infinityEngine) FilterDocIdsByMetaPushdown(ctx context.Context, kbIDs [
whereClause = kbFilter + " AND (" + whereClause + ")"
// Use Infinity connection to execute query
db, release, err := e.client.checkoutDatabase(ctx, "metadata.go")
if err != nil || db == nil {
infinityDB, release, err := e.client.checkoutDatabase(ctx, "metadata.go")
if err != nil || infinityDB == nil {
return nil
}
defer release()
table, err := db.GetTable(tableName)
table, err := infinityDB.GetTable(tableName)
if err != nil || table == nil {
return nil
}