def test_redis_write():
data = request.get_json()
key = data.get('key')
val = data.get('val')
RedisUtil.set_redis_data(key, val)
return "set: %s" % val
def user_job_handle():
data = request.get_json()
# 记录入参
serviceId = data.get('serviceID')
user_ip = data.get('ip')
user_port = data.get('port')
req_data = data.get('req_data')
# 生成userToken，分配id
id = Token.gen_service_token()
user_token = UserToken(service_id=serviceId, ip=user_ip, port=user_port, user_id=id)
# token存redis
RedisUtil.set_redis_data(user_token.user_id, user_token)
# 根据serviceId获取redis中chainInfo并注入
# 这里的chain_data应该对应内容为一个Chain_Info类
chain_data = RedisUtil.get_redis_data("serviceId_%d" % serviceId)
# chain_info = ChainInfo(chain_data["num"], chain_data["mini_service"])
# 初始化调用链状态
business_data = UserBusiness(is_migration=False, offset=0, data=req_data)
# 封装UserService
user_service = UserService(user_token=user_token, service_bus=business_data, service_chain=chain_data)
# 这部分直接调用相关的map内置函数去处理对应的业务逻辑
if not service_map.set_user_service(user_service):
return "该业务已经存在，业务异常"
# compute_handler.compute_us_func(user_service) 该业务已经直接启动
return jsonify({"user_id": id, "redirect_result": "your target ip is: {}:{} now".format(user_ip, user_port)})
def add_server_address(port: int):
peer_data = RedisUtil.get_redis_data("peers")
ip = get_host_ip()
if not peer_data:
init_data = {ip: port}
RedisUtil.set_redis_data("peers", init_data)
return
peer_data[ip] = port
RedisUtil.set_redis_data("peers", peer_data)
def admin_add_service():
data = request.get_json()
serviceId = data.get('serviceId')
num = data.get('num')
mini_service = data.get('mini_service')
# 数据服务存入redis
new_chain_info = ChainInfo(num, mini_service)
RedisUtil.set_redis_data("serviceId_%d" % serviceId, new_chain_info)
return "service: %d is added now" % serviceId
def api_upload():
if request.method == 'POST':
if 'file' not in request.files:
return "No file field in the request!"
f = request.files['file'] # 从表单的file字段获取文件，file为该表单的name值
if f:
fileName = f.filename
if fileName == '' or not allowed_file(fileName):
print("no such named files\n")
return json.dumps({'success': False}), 404
else:
fileHandle = {
"process_file": f.read().decode('utf-8', 'ignore')
}
# 记录入参
serviceId = int(request.form['serviceID'])
user_ip = request.form['ip']
user_port = request.form['port']
# 生成userToken，分配id
id = Token.gen_service_token()
user_token = UserToken(service_id=serviceId,
ip=user_ip,
port=user_port,
user_id=id)
# token存redis
RedisUtil.set_redis_data(user_token.user_id, user_token)
# 根据serviceId获取redis中chainInfo并注入
# 这里的chain_data应该对应内容为一个Chain_Info类
chain_data = RedisUtil.get_redis_data("serviceId_%d" %
serviceId)
# chain_info = ChainInfo(chain_data["num"], chain_data["mini_service"])
# 初始化调用链状态
business_data = UserBusiness(is_migration=False,
offset=0,
data=fileHandle)
# 封装UserService
user_service = UserService(user_token=user_token,
service_bus=business_data,
service_chain=chain_data)
# 这部分直接调用相关的map内置函数去处理对应的业务逻辑
if not service_map.set_user_service(user_service):
return "该业务已经存在，业务异常"
print("read success!\n")
return json.dumps({'success': True, "user_id": id}), 200
return "Please use Post Method to upload."
def test_redis_read():
data = request.get_json()
key = data.get('key')
chain = RedisUtil.get_redis_data(key)
return "get the type of the return object: {} and its msg: {}".format(
type(chain), chain.__str__())
def user_migration_handle():
data = request.get_json()
userId = data.get('userId')
serviceId = data.get('serviceId')
ip = data.get('ip')
port = data.get('port')
# 修改redis中用户的ip及port信息
token = RedisUtil.get_redis_data(userId)
RedisUtil.set_redis_data(userId, UserToken(user_id=userId, service_id=token["serviceID"], ip=ip, port=port))
# 后续迁移处理信息部分UserBusiness的flag都变成True
# 并且交给迁移转发模块进行处理，不在本地进行处理
if not migration_sender(userId=userId, serviceId=serviceId, flag=0, ip=ip):
return "Operate migration process fail!"
return "Migration is started, new ip: %s, new port: %d" % (ip, port)
def get_target_peer(ip: str) -> int:
peers = RedisUtil.get_redis_data("peers")
for peer_ip, peer_port in peers.items():
if peer_ip == ip:
return peer_port
return -1