多进程和多线程请求实例

采用多线程和多进程请求网页

#coding=utf-8
import requests
import time
import threading
import multiprocessing

def attack_work(name,num):
    count = 1
    while True:
        if count > num:
            break
        rip = rand_ip()
        url = "http://www.baidu.com/"
        headers = {
                    "Referer":"http://www.baidu.com"
        }
        response = requests.get(url, headers=headers)
        print("{}，第 {} 次请求----\n".format(name,count))
        # if response.status_code == 200:
        #     print('{} {}'.format(response.status_code, "success!\n"))
        #     time.sleep(1)
        # else:
        #     print('{} {}'.format(response.status_code, "failed!\n"))
        count += 1
        # time.sleep(0.5)
        
        
def timeLogger(func):
    def wrapper(*args,**kwargs):
        start_time = time.time()
        func(*args,**kwargs)
        duration = time.time()-start_time
        print("function: {} took {}s.".format(func.__name__,round(duration,3)))
    return wrapper
    

@timeLogger
def multi_thread(m,n):  # m为开启线程数，n为每个线程攻击的次数
    print("######启动{}个线程进行请求，每个线程请求{}次，共进行{}次请求######\n".format(m,n,m*n))
    lst = []
    for i in range(m):
        tname='线程：threadid-{}'.format(i+1)
        t = threading.Thread(target=attack_work,name=tname,args=(tname,n))
        t.start()
        lst.append(t)
    
    for t in lst:
        t.join()
        
@timeLogger
def multi_process(m,n): # m为开启进程数，n为每个进程攻击的次数
    print("######启动{}个进程进行请求，每个进程请求{}次，共进行{}次请求######\n".format(m,n,m*n))
    lst = []
    for i in range(m):
        pname="进程：processid-{}".format(i+1)
        p = multiprocessing.Process(target=attack_work,name=pname,args=(pname,n))
        p.start()
        lst.append(p)

    for p in lst:
        p.join()

if __name__=="__main__":
    multi_thread(20,2000) 
    # multi_process(20,2000)