sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve lock name to config_lock instead of search_index_lock

Browse source 
It is used to lock updates to all app config state, including processor
This commit is contained in:
Debanjum Singh Solanky 2023-07-15 12:07:07 -07:00
parent 58d86d7876
commit 7ad96036b0
2 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/khoj/configure.py
View file

@ -64,29 +64,29 @@ def configure_server(config: FullConfig, regenerate: bool, search_type: Optional
# Initialize Processor from Config
try:
state.search_index_lock.acquire()
state.config_lock.acquire()
state.processor_config = configure_processor(state.config.processor)
except Exception as e:
logger.error(f"🚨 Failed to configure processor")
raise e
finally:
state.search_index_lock.release()
state.config_lock.release()
# Initialize Search Models from Config
try:
state.search_index_lock.acquire()
state.config_lock.acquire()
state.SearchType = configure_search_types(state.config)
state.search_models = configure_search(state.search_models, state.config.search_type)
except Exception as e:
logger.error(f"🚨 Failed to configure search models")
raise e
finally:
state.search_index_lock.release()
state.config_lock.release()
# Initialize Content from Config
if state.search_models:
try:
state.search_index_lock.acquire()
state.config_lock.acquire()
state.content_index = configure_content(
state.content_index, state.config.content_type, state.search_models, regenerate, search_type
)
@ -94,7 +94,7 @@ def configure_server(config: FullConfig, regenerate: bool, search_type: Optional
logger.error(f"🚨 Failed to index content")
raise e
finally:
state.search_index_lock.release()
state.config_lock.release()
def configure_routes(app):
@ -114,7 +114,7 @@ if not state.demo:
@schedule.repeat(schedule.every(61).minutes)
def update_search_index():
try:
state.search_index_lock.acquire()
state.config_lock.acquire()
state.content_index = configure_content(
state.content_index, state.config.content_type, state.search_models, regenerate=False
)
@ -122,7 +122,7 @@ if not state.demo:
except Exception as e:
logger.error(f"🚨 Error updating content index via Scheduler: {e}")
finally:
state.search_index_lock.release()
state.config_lock.release()
def configure_search_types(config: FullConfig):

2
src/khoj/utils/state.py
View file

@ -24,7 +24,7 @@ host: str = None
port: int = None
cli_args: List[str] = None
query_cache = LRU()
search_index_lock = threading.Lock()
config_lock = threading.Lock()
SearchType = utils_config.SearchType
telemetry: List[Dict[str, str]] = []
previous_query: str = None