コード例 #1
0
ファイル: scan_es2redis.py プロジェクト: SwoJa/ruman
def scan_es2redis_week():
    count = 0
    s_re = scan(es_user_portrait, query={'query':{'match_all': {}}, 'size':1000}, index=portrait_index_name, doc_type=portrait_index_type)
    user_info = {}
    start_ts = time.time()
    while True:
        try:
            scan_re = s_re.next()['_source']
            count += 1
            uid = scan_re['uid']
            user_info[uid] = {'fansnum':scan_re['fansnum'], 'domain':scan_re['domain']}
            update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY, json.dumps(user_info))
            user_info = {}
            if count % 1000 == 0 and count != 0:
                end_ts = time.time()
                print '%s sec count 1000' % (end_ts - start_ts)
                start_ts = end_ts
        except StopIteration:
            print 'all done'
            if user_info:
                update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY, json.dumps(user_info))
                user_info = {}
            break
        except Exception, r:
            raise r
            break
コード例 #2
0
def scan_es2redis_week():
    count = 0
    s_re = scan(es_user_portrait,
                query={
                    'query': {
                        'match_all': {}
                    },
                    'size': 1000
                },
                index=portrait_index_name,
                doc_type=portrait_index_type)
    user_info = {}
    start_ts = time.time()
    while True:
        try:
            scan_re = s_re.next()['_source']
            count += 1
            uid = scan_re['uid']
            user_info[uid] = {
                'fansnum': scan_re['fansnum'],
                'domain': scan_re['domain']
            }
            update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY,
                                    json.dumps(user_info))
            user_info = {}
            #log_should_delete
            if count % 1000 == 0 and count != 0:
                end_ts = time.time()
                print '%s sec count 1000' % (end_ts - start_ts)
                start_ts = end_ts
            #log_should_delete
        except StopIteration:
            if user_info:
                update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY,
                                        json.dumps(user_info))
                user_info = {}
            break
        except Exception, r:
            raise r
            break
コード例 #3
0
ファイル: scan_es2redis.py プロジェクト: SwoJa/ruman
            if count % 1000 == 0 and count != 0:
                end_ts = time.time()
                print '%s sec count 1000' % (end_ts - start_ts)
                start_ts = end_ts
        except StopIteration:
            print 'all done'
            if user_info:
                update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY, json.dumps(user_info))
                user_info = {}
            break
        except Exception, r:
            raise r
            break

    if user_info:
        update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY, json.dumps(user_info))
    print 'count:', count

#scan es to redis as queue for update month
#write in version: 16-02-27
#order time task for every month
def scan_es2redis_month():
    count = 0
    s_re = scan(es_user_portrait, query={'query':{'match_all': {}}, 'size':1000}, index=portrait_index_name, doc_type=portrait_index_type)
    start_ts = time.time()
    user_info = {}
    while True:
        try:
            scan_re = s_re.next()['_source']
            count += 1
            uid = scan_re['uid']
コード例 #4
0
                end_ts = time.time()
                print '%s sec count 1000' % (end_ts - start_ts)
                start_ts = end_ts
            #log_should_delete
        except StopIteration:
            if user_info:
                update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY,
                                        json.dumps(user_info))
                user_info = {}
            break
        except Exception, r:
            raise r
            break

    if user_info:
        update_week_redis.lpush(UPDATE_WEEK_REDIS_KEY, json.dumps(user_info))


if __name__ == '__main__':
    log_time_ts = time.time()
    log_time_date = ts2datetime(log_time_ts)
    print 'cron/text_attribute/scan_es2redis_week.py&start&' + log_time_date

    try:
        scan_es2redis_week()
    except Exception, e:
        print e, '&error&', ts2date(time.time())

    log_time_ts = time.time()
    log_time_date = ts2datetime(log_time_ts)
    print 'cron/text_attribute/scan_es2redis_week.py&end&' + log_time_date