python 多线程库 vthread

多线程库vthread

安装

pip install vthread

线程池

如果不加装饰器就是普通的单线程，只用加一行就能在不破坏代码原来的结构直接实现线程池操作，能支持参数传递，支持分组。

• 不加装饰器

import time
start = time.time()
def foolfunc(num):
	print(f"test foolnum:{num}")
for i in range(10):
	# 使用sleep(1)即每次循环等待1秒用于检测线程池操作
	time.sleep(1)
	foolfunc(i)
end = time.time()
print(f"执行循环总用时{end - start}s")

代码运行结果：

• 加装饰器

import time
import vthread
start = time.time()
@vthread.pool(6)
def foolfunc(num):
	# 使用sleep(1)即每次等待1秒 用于检测线程池操作
	time.sleep(1)
	print(f"test foolnum:{num}")
for i in range(10):
	foolfunc(i)
end = time.time()
print(f"执行循环总用时{end - start}s")

代码运行结果：

分组线程池

• 可以实现代码中其中几个函数有m个线程，另外几个函数有n个线程。

import time
import vthread
# 额外开启线程池组的话，最好不要用gqueue=0
# gqueue=0是默认参数
pool_1 = vthread.pool(5, gqueue=1)  # 定义5个线程，组名为1
pool_2 = vthread.pool(2, gqueue=2)  # 定义2个线程，组名为2
@pool_1
def foolfunc1(num):
	time.sleep(1)
	print(f"test foolnum1:{num}")
@pool_2
def foolfunc2(num):
	time.sleep(1)
	print(f"test foolnum2:{num}")
@pool_2
def foolfunc3(num):
	time.sleep(1)
	print(f"test foolnum3:{num}")
for i in range(5):
	foolfunc1(i)
for i in range(5):
	foolfunc2(i)
for i in range(5):
	foolfunc3(i)

代码运行结果：

原子封装

• 如果需要考虑对函数内某些操作进行锁的操作，需要把某些操作进行原子化，可以把原子化的操作写成函数，用vthread.atom装饰。

import time
import vthread
@vthread.thread(5)
def foolfunc():
	@vthread.atom  # 对函数加锁封装
	def do_some_fool_thing1():
		pass
	@vthread.atom  # 对函数加锁封装
	def do_some_fool_thing2():
		pass
	# 执行时就会实现原子操作
	do_some_fool_thing1()
	do_some_fool_thing2()

线程池分组

• 假如在使用过程中装饰了多个函数会怎么计算线程池的线程数量呢？ 这里给出了说明，在 vthread.pool 函数库中，是以 gqueue 这个参数来确定线程池分组的。 而相同的分组，则会默认使用最后一个 “人为定义” 的伺服线程数量。如果不给线程数量，这样就意味着线程池数量为默认的cpu核心数。

#==============================================#
#                                              #
#  注意！关于多个函数装饰器，线程池数量的定义    #
#                                              #
#==============================================#
# 例.1
@vthread.pool(10)
def foolfunc1():
    pass
@vthread.pool(18)
def foolfunc1():
    pass
# 这样就意味着gqueue=0的线程池数量为18
# 例.2
@vthread.pool(10)
def foolfunc1():
    pass
@vthread.pool()
def foolfunc1():
    pass
# 这样就意味着gqueue=0的线程池数量为10
# 例.3
@vthread.pool()
def foolfunc1():
    pass
@vthread.pool()
def foolfunc1():
    pass
# 这样就意味着gqueue=0的线程池数量为默认的cpu核心数
# 例.4
pool1 = vthread.pool(gqueue=1)
pool2 = vthread.pool(6,gqueue=2)
pool2 = vthread.pool(8,gqueue=2)
# 这样就意味着gqueue=1的线程池数量为默认的cpu核心数，gqueue=2的线程池数量为8

普通多线程

import time
import vthread
# 只要这一行就能让函数变成开5个线程执行同个函数
@vthread.thread(5)
def foolfunc(num):
    time.sleep(1)
    print(f"test1 foolnumb: {num}")
foolfunc(123) # 加入装饰器后，这个函数就变成了开5个线程执行的函数了

执行代码结果：

注意：
vthread.thread 不带参数的方式只能装饰一个函数，装饰多个函数会出现问题，仅用于测试单个函数。
vthread.thread(1) 带参数的可以装饰多个函数，但是已经有了分组线程池的强大功能。