commit da1e19b72bceb1c5a094613c286c1331ff1a6c1b
author Yiyang Zhou <yiyangzhou1@xiaohongshu.com> Fri Aug 13 16:30:58 2021 +0800
committer Jens Geyer <Jens-G@users.noreply.github.com> Sun Nov 14 00:54:12 2021 +0100
tree f4aa8be8afcddfde605160e8dda7296e759a5d84
parent ea39a14a1fbc8a7683e91ab4bdd23f8b207c8d5d

THRIFT-5454: add __setState__ function to TProcessPoolServer to enable correct multiprocessing behavior related to pickling

Client: Python

lib/py/src/server/TProcessPoolServer.py

diff --git a/lib/py/src/server/TProcessPoolServer.py b/lib/py/src/server/TProcessPoolServer.py
index fe6dc81..d2e97c5 100644
--- a/lib/py/src/server/TProcessPoolServer.py
+++ b/lib/py/src/server/TProcessPoolServer.py
@@ -42,6 +42,11 @@
         self.stopCondition = Condition()
         self.postForkCallback = None
 
+    def __getstate__(self):
+        state=self.__dict__.copy()
+        state['workers'] = None
+        return state
+
     def setPostForkCallback(self, callback):
         if not callable(callback):
             raise TypeError("This is not a callback!")