From b70a4a9ec833fa564fa25dc4b9acb857e1f9266b Mon Sep 17 00:00:00 2001
From: John Wang <takatost@gmail.com>
Date: Thu, 25 May 2023 20:23:17 +0800
Subject: [PATCH] fix: shared lock

---
 .../index_tool_callback_handler.py              | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/api/core/callback_handler/index_tool_callback_handler.py b/api/core/callback_handler/index_tool_callback_handler.py
index f0c9379413..669e0829a5 100644
--- a/api/core/callback_handler/index_tool_callback_handler.py
+++ b/api/core/callback_handler/index_tool_callback_handler.py
@@ -31,8 +31,17 @@ class DatasetIndexToolCallbackHandler(IndexToolCallbackHandler):
             index_node_id = node.node.doc_id
 
             # add hit count to document segment
-            db.session.query(DocumentSegment).filter(
-                DocumentSegment.dataset_id == self.dataset_id,
-                DocumentSegment.index_node_id == index_node_id
-            ).update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
+            update_stmt = (
+                db.session.query(DocumentSegment)
+                .filter(
+                    DocumentSegment.dataset_id == self.dataset_id,
+                    DocumentSegment.index_node_id == index_node_id
+                )
+                .update(
+                    {DocumentSegment.hit_count: DocumentSegment.hit_count + 1},
+                    synchronize_session=False
+                )
+            )
 
+            db.session.execute(update_stmt)
+            db.session.commit()