Commit 69e76b94 authored by 张彦钊's avatar 张彦钊

change test file

parent e2c96f26
...@@ -96,9 +96,9 @@ def write_redis(device_id,cid_list): ...@@ -96,9 +96,9 @@ def write_redis(device_id,cid_list):
print("insert redis fail") print("insert redis fail")
print(e) print(e)
def group_redis(x): def group_redis(device_id,cid_list):
device_id = list(x)[0] # device_id = x[0]
cid_list = list(x)[1] # cid_list = x[1]
db = pymysql.connect(host='172.16.40.158', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='eagle') db = pymysql.connect(host='172.16.40.158', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='eagle')
sql = "select b.id from src_mimas_prod_api_diary_tags a left join src_zhengxing_api_tag b " \ sql = "select b.id from src_mimas_prod_api_diary_tags a left join src_zhengxing_api_tag b " \
"on a.tag_id = b.id where b.tag_type = '3' and a.diary_id in {}".format(tuple(cid_list)) "on a.tag_id = b.id where b.tag_type = '3' and a.diary_id in {}".format(tuple(cid_list))
...@@ -128,7 +128,7 @@ def group_redis(x): ...@@ -128,7 +128,7 @@ def group_redis(x):
def group_write(rdd): def group_write(rdd):
rdd.groupByKey().foreachPartition(lambda x:group_redis(x)) rdd.groupByKey().foreachPartition(lambda x:x.map())
return "good" return "good"
...@@ -145,7 +145,8 @@ kafkaParams = {"metadata.broker.list": "172.16.44.25:9092,172.16.44.31:9092,172. ...@@ -145,7 +145,8 @@ kafkaParams = {"metadata.broker.list": "172.16.44.25:9092,172.16.44.31:9092,172.
stream = KafkaUtils.createDirectStream(ssc, ["gm-maidian-data"], kafkaParams) stream = KafkaUtils.createDirectStream(ssc, ["gm-maidian-data"], kafkaParams)
transformstream = stream.transform(lambda x:model(x)).foreachRDD(group_write) transformstream = stream.transform(lambda x:model(x)).foreachRDD(lambda x:x.groupByKey().
foreachPartition(group_redis))
# transformstream.pprint() # transformstream.pprint()
# print(transformstream) # print(transformstream)
ssc.start() ssc.start()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment