mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-24 04:26:00 -04:00
Yeuoly
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
from dify_plugin.core.server.__base.request_reader import RequestReader
|
|
from dify_plugin.core.server.__base.response_writer import ResponseWriter
|
|
|
|
|
|
class Session:
|
|
# class variable to store all sessions
|
|
_session_pool = set["Session"]()
|
|
_executor: ThreadPoolExecutor
|
|
# current session id
|
|
session_id: str
|
|
# reader and writer
|
|
reader: RequestReader
|
|
writer: ResponseWriter
|
|
|
|
def __init__(self, session_id: str, executor: ThreadPoolExecutor, reader: RequestReader, writer: ResponseWriter):
|
|
self.session_id = session_id
|
|
self._session_pool.add(self)
|
|
self._executor = executor
|
|
self.reader = reader
|
|
self.writer = writer
|
|
|
|
def __del__(self):
|
|
self._session_pool.remove(self)
|
|
|
|
def close(self):
|
|
self._session_pool.remove(self)
|
|
|
|
@classmethod
|
|
def get_session(cls, session_id: str):
|
|
for session in cls._session_pool:
|
|
if session.session_id == session_id:
|
|
return session
|
|
return None
|