update

eda/smart_rank/dist_update_user_portrait.py

@@ -110,6 +110,7 @@ def get_user_tag_score(cl_id, all_log_df, gm_kv_cli, size=10):
             finally_score_lst = finally_score[["tag_id","tag_score"]].to_dict('record')
             tag_id_list = tag_list2dict(finally_score_lst, size)
             # 写gmkv
+            gm_kv_cli = redis.Redis(host="172.16.40.135", port=5379, db=2, socket_timeout=2000)
             cl_id_portrait_key = "user:portrait_tags:cl_id:" + str(cl_id)
             tag_id_list_json = json.dumps(tag_id_list)
             gm_kv_cli.set(cl_id_portrait_key, tag_id_list_json)
@@ -154,7 +155,6 @@ if __name__ == '__main__':
 
         spark = SparkSession.builder.config(conf=sparkConf).enableHiveSupport().getOrCreate()
         spark.sparkContext.setLogLevel("WARN")
-        gm_kv_cli = redis.Redis(host="172.16.40.135", port=5379, db=2, socket_timeout=2000)
         device_ids_lst_rdd = spark.sparkContext.parallelize(device_ids_lst)
         result = device_ids_lst_rdd.repartition(100).map(lambda x: get_user_tag_score(x, all_log_df, gm_kv_cli))
         result.collect()