kpush源码解析---服务端（alloc server）
继续上篇已经分析了android端代码，本篇将主要分析kpush服务端如何为连接的客户端分配用户信息，并为其分配tcp server。
上文中提到客户端的pushservice中会通过http post请求http://demo.kpush.cn/server/alloc，以获取用户及消息服务器ip/port信息。那么我们接下来看看kpush服务端关于该流程的相关代码。
如果想找到相关代码，但是我们目前对整个项目目录结构不太熟悉该怎么办呢？其实很容易的，我们先找到/server/alloc路由，搜索源代码可以发现在kpush/web/views/frontend.py文件中有对该路由做处理：alloc_server，该函数代码如下：
@bp.route('/server/alloc', methods=['POST'])
def alloc_server():
"""
# 方便测试
if not request.json_data:
return jsonify(
ret=proto.RET_INVALID_PARAMS
)
"""
appinfo = get_appinfo_by_appkey(request.json_data['appkey'])
web_logger.debug("appinfo: %s", appinfo)
if appinfo is None:
# 报错
jsonify(
ret=proto.RET_INVALID_PARAMS
)
return
user = create_or_update_user(dict(
appid=appinfo['appid'],
channel=request.json_data['channel'],
device_id=request.json_data['device_id'],
device_name=request.json_data.get('device_name'),
os=request.json_data.get('os'),
os_version=request.json_data.get('os_version'),
sdk_version=request.json_data.get('sdk_version'),
))
server_list = current_app.config['SERVER_LIST']
# 取模
server = server_list[user['uid'] % len(server_list)]
return current_app.response_class(pack_data(
dict(
ret=0,
user=dict(
uid=user['uid'],
key=user['key'],
),
server=dict(
host=server['outer_host'],
port=server['outer_port'],
)
)
), mimetype='application/json')仔细分下该段代码，我们可以发现该段代码主要完成了三个功能：
1. 根据post上来的相关数据创建或更新对应的用户信息，主要包含uid和key（还记得这两个数据是用来干嘛的吗？客户端会用这两个数据执行登录操作）
2. 根据uid对服务列表个数取模，为该客户端分配一个tcp server
3. 把用户信息和tcp服务器信息封装成json数据，并返回到客户端
现在我们回顾下上篇的内容，客户端会通过http post获取用户和tcp服务端ip，然后通过ferry连接上该tcp服务器，并使用uid和key进行登录。到此我们知道服务端是如何处理该请求，并返回相关的信息了。但是在上段函数中还有一个函数我们还没有去分析：create_or_update_user，根据函数名就可以知道该函数主要功能就是完成上述的第一个功能，该函数具体代码如下：
def create_or_update_user(user_info):
"""
返回或者创建user
"""
user_table = kit.mongo_client.get_default_database()[current_app.config['MONGO_TB_USER']]
user = user_table.find_one(dict(
appid=user_info['appid'],
device_id=user_info['device_id'],
))
if user is None:
new_user_info = dict(
uid=alloc_autoid("user"),
key=uuid.uuid4().hex,
)
new_user_info.update(user_info)
# 没有这个用户
user_table.insert(new_user_info)
else:
# 更新字段, 要用 $set 才会只更新指定的字段
user_table.update({
'_id': user['_id'],
}, {
"$set": user_info,
})
return user_table.find_one(dict(
appid=user_info['appid'],
device_id=user_info['device_id'],
))至此，为客户端创建用户信息，并分配tcp服务器的相关代码已经分析完毕，下一篇我们将分析消息推送流程相关代码。