174 lines
6.5 KiB
Python
174 lines
6.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
from myredis.myredis import my_redis
|
||
import datetime
|
||
from mysql.mmysql import MysqlBase
|
||
from prod_config import mysql_promotion_config, log_path, ad_list_interface_port, BEGIN, END
|
||
from log.mylog import define_logger
|
||
import logging
|
||
import pdb
|
||
import json
|
||
|
||
define_logger(f"{log_path}/ad_tasks.log")
|
||
log = logging.getLogger(__name__)
|
||
|
||
limit = 100
|
||
|
||
BEGIN = '1999-01-01'
|
||
END = '3000-01-01'
|
||
mydb = MysqlBase(**mysql_promotion_config)
|
||
remove_list = ('ad_num', 'gameid')
|
||
|
||
|
||
def get_area_by_locationid(ldid):
|
||
area = []
|
||
sql = f"select area from location where id={ldid}"
|
||
data = mydb.query(sql)
|
||
if data:
|
||
for line in data:
|
||
pdb.set_trace()
|
||
area.append(line[0].replace('[','').replace(']','').replace('"',""))
|
||
else:
|
||
log.error(f"get area from location failed!,sql={sql}")
|
||
return area
|
||
|
||
|
||
def split_ad_info(localtionid, line):
|
||
# 从locationid获取位置相关属性，与ad信息打包，产生提供给接口用的数据
|
||
location_sql = f"select area,x,y,x_offset,y_offset,type,mode,ld_property from location WHERE in_used=1 and id={localtionid}"
|
||
data = mydb.query(location_sql)
|
||
try:
|
||
line['area'], line['x'], line['y'], line['x_offset'], line['y_offset'], line['type'], line['mode'], line[
|
||
'ld_property'] = data[0]
|
||
except Exception:
|
||
log.error(f"get localtion info by sql={location_sql} error!", exc_info=True)
|
||
return None
|
||
for item in remove_list:
|
||
line.pop(item)
|
||
return line
|
||
|
||
|
||
def sadd_adkey(key, line, item):
|
||
my_redis.sadd(key, line['id'])
|
||
my_redis.expire(key, 120)
|
||
info_key = f"adinfo::{line['id']}::info"
|
||
new_line = split_ad_info(item, line)
|
||
if new_line:
|
||
my_redis.hmset(info_key, new_line)
|
||
my_redis.expire(info_key, 60 * 60)
|
||
else:
|
||
log.error(f"split adinfo about locationid failed! localtionid={item} ")
|
||
|
||
|
||
|
||
def send_cache_data():
|
||
log.info("start update cache !")
|
||
now = datetime.datetime.today().strftime("%Y-%m-%d %H:%M:%S")
|
||
all = []
|
||
get_data_sql = f"""select
|
||
id,
|
||
name,
|
||
ad_num,
|
||
ad_title,
|
||
ad_body,
|
||
ad_image,
|
||
jump_param,
|
||
ad_sort,
|
||
companyid,
|
||
gameid,
|
||
channelid,
|
||
jump_status,
|
||
locationid ,
|
||
ad_property,
|
||
createtime
|
||
from
|
||
ad
|
||
where
|
||
status=1
|
||
and in_used=1
|
||
and '{now}'> begin_time
|
||
and '{now}'<end_time ;
|
||
"""
|
||
|
||
data = mydb.query(get_data_sql)
|
||
if data:
|
||
for line in data:
|
||
if line:
|
||
item = {}
|
||
try:
|
||
item['id'], item['name'], item['ad_num'], item['ad_title'], item['ad_body'], item['ad_image'], item[
|
||
'jump_param'], item['ad_sort'], item['companyid'], item['gameid'], item['channelid'], item[
|
||
'jump_status'], item['locationid'], item['ad_property'],item['createtime'] = line
|
||
if item.get('jump_param', "") and (item['ad_property'].find("jump_param") == -1):
|
||
try:
|
||
temp = {}
|
||
temp['jump_param'] = item.get('jump_param', "")
|
||
if not item['ad_property']:
|
||
item['ad_property'] = {}
|
||
if not isinstance(item.get('ad_property', {}), dict):
|
||
if not item.get('ad_property', {}):
|
||
item['ad_property'] = {}.update(temp)
|
||
else:
|
||
item['ad_property'] = json.loads(
|
||
item.get('ad_property', {}).replace("‘", '"').replace("’", '"'))
|
||
item['ad_property'].update(temp)
|
||
item['ad_property'] = json.dumps(item['ad_property'])
|
||
except Exception:
|
||
log.error(f"write {item}", exc_info=True)
|
||
all.append(item)
|
||
except Exception:
|
||
log.error("split data failed", exc_info=True)
|
||
# 检查ID是否存在播放列表中，以及播放次数是否完毕
|
||
if all:
|
||
# log.info(f"get data was {all}!\n")
|
||
for line in all:
|
||
locationid = json.loads(line.get('locationid'))
|
||
for item in locationid:
|
||
pdb.set_trace()
|
||
area = get_area_by_locationid(item)
|
||
for one in area:
|
||
key = f"ad::{line.get('gameid', 0)}::{line.get('channelid', 0)}::{one}::{item}"
|
||
if int(line['ad_num']) == 0:
|
||
sadd_adkey(key, line, item)
|
||
elif int(line['ad_num']) > 0:
|
||
num = my_redis.get(f"adnum::{line['id']}::num")
|
||
if not num:
|
||
num = 0
|
||
if int(line['ad_num']) > int(num):
|
||
sadd_adkey(key, line, item)
|
||
else:
|
||
if my_redis.sismember(key, line['id']):
|
||
my_redis.srem(key, line['id'])
|
||
log.info(f"remove {line['id']} from {key}!")
|
||
else:
|
||
log.error(f"get ad_num from mysql failed! ad_num={line['ad_num']}")
|
||
|
||
# 删除过期的数据
|
||
log.info("remove expire data from cache!")
|
||
expire_sql = f"""select
|
||
id,gameid,channelid,locationid,status
|
||
from
|
||
ad
|
||
where
|
||
'{now}'> end_time
|
||
or status in (3,4) """
|
||
remove_data = mydb.query(expire_sql)
|
||
if remove_data:
|
||
for line in remove_data:
|
||
try:
|
||
id, gameid, channelid, locationid = line
|
||
for item in locationid:
|
||
area = get_area_by_locationid(item)
|
||
for one in area:
|
||
key = f"ad::{line.get('gameid', 0)}::{line.get('channelid', 0)}::{one}::{item}"
|
||
if my_redis.sismember(key, id):
|
||
my_redis.srem(key, id)
|
||
my_redis.expire(f"ad::{id}::info", 1)
|
||
log.info(f"remove {id} from {key} success!")
|
||
except Exception:
|
||
log.error("拆解过期数据出错!", exc_info=True)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
log.info('start!')
|
||
send_cache_data()
|