182 lines
7.9 KiB
Erlang
182 lines
7.9 KiB
Erlang
%%%-------------------------------------------------------------------
|
|
%%% @author licheng5
|
|
%%% @copyright (C) 2020, <COMPANY>
|
|
%%% @doc
|
|
%%%
|
|
%%% @end
|
|
%%% Created : 26. 4月 2020 3:36 下午
|
|
%%%-------------------------------------------------------------------
|
|
-module(container_handler).
|
|
-author("licheng5").
|
|
-include("iot.hrl").
|
|
|
|
-define(REQ_TIMEOUT, 10000).
|
|
|
|
%% API
|
|
-export([handle_request/4]).
|
|
|
|
handle_request("GET", "/container/get_all", #{<<"uuid">> := UUID}, _) when is_binary(UUID) ->
|
|
%% 检查ConfigJson是否是合法的json字符串
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(-1, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
case iot_host:get_containers(Pid) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(-1, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(-1, Reason)}
|
|
end
|
|
end;
|
|
|
|
%% 下发config.json, 微服务接受后，保存服务配置
|
|
handle_request("POST", "/container/push_config", _,
|
|
#{<<"uuid">> := UUID, <<"container_name">> := ContainerName, <<"config">> := Config, <<"timeout">> := Timeout0})
|
|
when is_binary(UUID), is_binary(ContainerName), is_binary(Config), is_integer(Timeout0) ->
|
|
|
|
%% 检查ConfigJson是否是合法的json字符串
|
|
true = iot_util:is_json(Config),
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(-1, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
Timeout = Timeout0 * 1000,
|
|
case iot_host:config_container(Pid, ContainerName, Config) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, Timeout) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(-1, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(-1, Reason)}
|
|
end
|
|
end;
|
|
|
|
%% 部署微服务
|
|
handle_request("POST", "/container/deploy", _, #{<<"uuid">> := UUID, <<"task_id">> := TaskId, <<"config">> := Config})
|
|
when is_binary(UUID), is_integer(TaskId), is_map(Config) ->
|
|
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(404, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
case iot_host:deploy_container(Pid, TaskId, Config) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end
|
|
end;
|
|
|
|
%% 启动服务
|
|
handle_request("POST", "/container/start", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(404, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
case iot_host:start_container(Pid, ContainerName) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end
|
|
end;
|
|
|
|
%% 停止服务
|
|
handle_request("POST", "/container/stop", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(404, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
case iot_host:stop_container(Pid, ContainerName) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end
|
|
end;
|
|
|
|
handle_request("POST", "/container/kill", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(404, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
case iot_host:kill_container(Pid, ContainerName) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end
|
|
end;
|
|
|
|
%% 删除容器
|
|
handle_request("POST", "/container/remove", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
|
|
case iot_host:get_pid(UUID) of
|
|
undefined ->
|
|
{ok, 200, iot_util:json_error(404, <<"host not found">>)};
|
|
Pid when is_pid(Pid) ->
|
|
case iot_host:remove_container(Pid, ContainerName) of
|
|
{ok, Ref} ->
|
|
case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
|
|
{ok, Result} ->
|
|
{ok, 200, iot_util:json_data(Result)};
|
|
{error, Reason} ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end;
|
|
{error, Reason} when is_binary(Reason) ->
|
|
{ok, 200, iot_util:json_error(400, Reason)}
|
|
end
|
|
end;
|
|
|
|
%handle_request("POST", "/container/task_log", _, #{<<"uuid">> := UUID, <<"task_id">> := TaskId}) when is_binary(UUID), is_integer(TaskId) ->
|
|
% case iot_host:get_pid(UUID) of
|
|
% undefined ->
|
|
% {ok, 200, iot_util:json_error(404, <<"host not found">>)};
|
|
% Pid when is_pid(Pid) ->
|
|
% case iot_host:task_log(Pid, TaskId) of
|
|
% {ok, Ref} ->
|
|
% case iot_host:await_reply(Ref, 5000) of
|
|
% {ok, Result} ->
|
|
% {ok, 200, iot_util:json_data(Result)};
|
|
% {error, Reason} ->
|
|
% {ok, 200, iot_util:json_error(400, Reason)}
|
|
% end;
|
|
% {error, Reason} when is_binary(Reason) ->
|
|
% {ok, 200, iot_util:json_error(400, Reason)}
|
|
% end
|
|
% end;
|
|
|
|
handle_request(_, Path, _, _) ->
|
|
Path1 = list_to_binary(Path),
|
|
{ok, 200, iot_util:json_error(-1, <<"url: ", Path1/binary, " not found">>)}.
|
|
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
%% helper methods
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% |