snaps/thread_utils.py

   1 # Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
   2 #                    and others.  All rights reserved.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at:
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 from multiprocessing.pool import ThreadPool
  17
  18 _pool = None
  19
  20
  21 # Define a thread pool with a limit for how many simultaneous API requests
  22 # can be in progress at once.
  23 def worker_pool(size=5):
  24     global _pool
  25     if _pool is None:
  26         _pool = ThreadPool(processes=size)
  27     return _pool