多线程+队列

import sys
from sqltool.mysql_client import MySqlClient
from queue import Queue, Empty
import os
import threading

base_dir = os.path.dirname(os.path.abspath('__file__'))
# 系统分隔符
os_sep = os.sep
sys.path.append(base_dir)


MYSQL_HOST = os.getenv("MYSQL_HOST", "localhost")
MYSQL_PORT = int(os.getenv("MYSQL_PORT", 3306))
MYSQL_USER = os.getenv("MYSQL_USER", "root")
MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD", "123")

MYSQL_SPIDER_DB = os.getenv("MYSQL_SPIDER_DB", "fosseye")
MYSQL_LJ_ID_DB = os.getenv("MYSQL_LJ_ID_DB", "lj_id")
MYSQL_OPEN_SOURCE_DB = os.getenv("MYSQL_OPEN_SOURCE_DB", "open_source_projects")


class MavenJar:

    def __init__(self):
        self.maven_cli = MySqlClient(
            host=MYSQL_HOST,
            port=MYSQL_PORT,
            user=MYSQL_USER,
            password=MYSQL_PASSWORD,
            db=MYSQL_SPIDER_DB
        )

    def get_maven_mysql(self):
        sql = "SELECT * FROM `表名` WHERE jar_md5 IS NULL AND jar_name IS NOT NULL"
        return self.maven_cli.query(sql)

    def maven_jar(self, res):
        pass

    def run(self, que):
        while True:
            try:
                k = que.get_nowait()
            except Empty:
                return
            self.maven_jar(k)


if __name__ == '__main__':
    maven = MavenJar()
    que = Queue()
    datas = maven.get_maven_mysql()
    for data in datas:
        que.put(data)
    
    ths = [threading.Thread(target=maven.run, args=(que,)) for i in range(20)]
    for th in ths:
        th.start()
    for th in ths:
        th.join()

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值