define paramter for reactor thread pool size

Precommit-Verified: dd6c120b4089d83a8364f85693568a5cac01db1b0cd4c0aa2b66e42090052663
pryorda · Jul 20, 2022 · 0624e51 · 0624e51
1 parent f32eaa9
commit 0624e51
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vmware_exporter/vmware_exporter.py b/vmware_exporter/vmware_exporter.py
@@ -2042,6 +2042,8 @@ def main(argv=None):
     parser.add_argument('-v', '--version', action="version",
                         version='vmware_exporter {version}'.format(version=__version__),
                         help='Print version and exit')
+    parser.add_argument('-ps', '--pool-size', dest="pool_size", type=int,
+                        default=25, help="Set thread pool size")
 
     args = parser.parse_args(argv or sys.argv[1:])
 
@@ -2050,7 +2052,7 @@ def main(argv=None):
         raise ValueError("Invalid log level: {level}".format(level=args.loglevel))
     logging.basicConfig(level=numeric_level, format='%(asctime)s %(levelname)s:%(message)s')
 
-    reactor.suggestThreadPoolSize(25)
+    reactor.suggestThreadPoolSize(args.pool_size)
 
     factory = Site(registerEndpoints(args))
     logging.info("Starting web server on port {address}:{port}".format(address=args.address, port=args.port))