def get_offline_user_resource_visitinfo(uid, resource_type_list): cache_errorCode, cache_res = cache_models.get_cache_offline_user_resource_visitinfo(uid, resource_type_list) return_none_keys = get_return_none_keys(cache_res, resource_type_list) db_errorCode = True db_res = {} if len(return_none_keys) > 0: logger.warning('get_offline_user_resource_visitinfo, cache has return none key: %s' % return_none_keys) db_errorCode, db_res = db_models.get_db_offline_user_resource_visitinfo(uid, return_none_keys) #logger.debug('db_Code: %s, db_res:%s' % (db_errorCode, db_res)) for resource_type , visitinfo_data in db_res.items(): cache_models.set_cache_offline_user_resource_visitinfo(uid, resource_type, visitinfo_data) logger.debug('get_offline_user_resource_visitinfo, set cache uid[%s], resource_type[%s]' % (uid, resource_type)) if True == db_errorCode: for resource_type in [x for x in return_none_keys if x not in db_res.keys()]: cache_models.set_cache_offline_user_resource_visitinfo(uid, resource_type, DataDef.CACHE_DEFAULT_NONE_VALUE, DataDef.CACHE_NONE_EXPIRE_SECONDS) logger.debug('get_offline_user_resource_visitinfo, set cache uid[%s] resource_type[%s] None Value' % (uid, resource_type)) return (cache_errorCode and db_errorCode, merge_db_cache_result(cache_res, db_res))
def test_db_user_resource_visitinfo(self): uid = '1234567' resource_type_list = ['music', 'news', 'movie'] visit_info = VisitInfo() visit_info.pv_count = 5 visit_info.click_count = 1 visit_info.weight = 0.2 visit_info_data = visit_info.SerializeToString() for resource_type in resource_type_list: self.assertEqual( True, db_models_update.set_db_offline_user_resource_visitinfo(uid, resource_type, visit_info_data)) prefer_res = {} for resource_type in resource_type_list: prefer_res[resource_type] = visit_info res = db_models.get_db_offline_user_resource_visitinfo(uid, resource_type_list) real_res = {} self.assertEqual(True, res[0]) for k, v in res[1].items(): res_visit_info = VisitInfo() res_visit_info.ParseFromString(v) real_res[k] = res_visit_info self.assertEqual(prefer_res, real_res)