chore: import upstream snapshot with attribution

2026-07-13 12:47:05 +08:00
commit 4f3b7da785
7394 changed files with 2005594 additions and 0 deletions
@@ -0,0 +1,74 @@
+/* ******************************************************************************
+ *
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Apache License, Version 2.0 which is available at
+ * https://www.apache.org/licenses/LICENSE-2.0.
+ *
+ *  See the NOTICE file distributed with this work for additional
+ *  information regarding copyright ownership.
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ ******************************************************************************/
+
+//
+// @author raver119@gmail.com
+//
+
+#ifndef SAMEDIFF_THREADPOOL_H
+#define SAMEDIFF_THREADPOOL_H
+#include <execution/BlockingQueue.h>
+#include <execution/CallableInterface.h>
+#include <execution/CallableWithArguments.h>
+#include <execution/Ticket.h>
+#include <system/common.h>
+
+#include <atomic>
+#include <list>
+#include <mutex>
+#include <queue>
+#include <thread>
+#include <vector>
+
+namespace samediff {
+class SD_LIB_EXPORT ThreadPool {
+ private:
+  std::vector<std::thread> _threads;
+  std::vector<BlockingQueue<CallableWithArguments*>*> _queues;
+  std::vector<CallableInterface*> _interfaces;
+
+  std::mutex _lock;
+  std::atomic<int> _available;
+  std::queue<Ticket*> _tickets;
+
+ protected:
+  ThreadPool();
+  ~ThreadPool();
+
+ public:
+  static ThreadPool& getInstance();
+
+  /**
+   * This method returns list of pointers to threads ONLY if num_threads of threads were available upon request,
+   * returning empty list otherwise
+   * @param num_threads
+   * @return
+   */
+  Ticket* tryAcquire(int num_threads);
+
+  /**
+   * This method marks specified number of threads as released, and available for use
+   * @param num_threads
+   */
+  void release(int num_threads = 1);
+
+  void release(Ticket* ticket);
+};
+}  // namespace samediff
+
+#endif  // DEV_TESTS_THREADPOOL_H