过滤查询时间戳相等
This commit is contained in:
parent
81343908e4
commit
95855fb148
@ -1,5 +1,6 @@
|
||||
from pymongo import UpdateOne
|
||||
from pydantic import Field
|
||||
import pandas as pd
|
||||
|
||||
from .task import Task
|
||||
from utils import *
|
||||
@ -16,11 +17,14 @@ class SummaryFunc(Task):
|
||||
need: list = Field(None, title='消耗')
|
||||
ftype: str = Field(..., title='功能')
|
||||
data: dict = Field(None, title='功能数据')
|
||||
cdate: int = Field(..., title='当天0点')
|
||||
|
||||
def cleaning(self, cursor_list):
|
||||
for cursor in cursor_list: # type:dict
|
||||
for event_coll, ts in cursor.items(): # type:str,dict
|
||||
logger.info(f'开始处理{self.game_name} 处理 {event_coll} ...')
|
||||
if ts['cursor_st'] == ts['cursor_et']:
|
||||
continue
|
||||
logger.info(f'开始处理{self.game_name} 处理 {event_coll} 游标 {ts}')
|
||||
where = {
|
||||
'_event_name': 'Func',
|
||||
'_event_time': {
|
||||
@ -33,10 +37,13 @@ class SummaryFunc(Task):
|
||||
bulk_data = []
|
||||
for item in self.local_db[event_coll].find(where, projection):
|
||||
try:
|
||||
item['cdate'] = int(pd.Timestamp(item['_event_time'], unit='s', tz=self.timezone) \
|
||||
.normalize().timestamp())
|
||||
model = self.Model(**item)
|
||||
data = model.dict()
|
||||
bulk_data.append(UpdateOne({'_id': data['_id']}, {'$set': data}, upsert=True))
|
||||
except Exception as e:
|
||||
logger.error(e)
|
||||
# logger.error(f'ftype {item["ftype"]} msg:{e}')
|
||||
pass
|
||||
self.remote_db[self.task_name].bulk_write(bulk_data, ordered=False)
|
||||
self.set_cursor(cursor_st=ts['cursor_st'], cursor_et=ts['cursor_et'])
|
||||
|
@ -46,11 +46,11 @@ class Task(metaclass=abc.ABCMeta):
|
||||
elif int(time.time()) - last_ts > time_out:
|
||||
# 任务超时
|
||||
# todo 钉钉通知
|
||||
logger.info('钉钉通知')
|
||||
logger.info(f'{self.game_name} 钉钉通知')
|
||||
return False
|
||||
else:
|
||||
# 正在运行没超时
|
||||
logger.info('正在运行没超时')
|
||||
logger.info(f'{self.game_name} 正在运行没超时')
|
||||
return False
|
||||
|
||||
def set_run_ts(self):
|
||||
|
Loading…
Reference in New Issue
Block a user