feat(emqx_resource): read and save configs from and to file
This commit is contained in:
parent
e7ffa07a1a
commit
5d52ce044d
|
@ -0,0 +1,4 @@
|
|||
[{connectors,[#{<<"id">> => <<"mysql-abc">>,
|
||||
<<"type">> => <<"mysql_connenctor">>},
|
||||
#{<<"id">> => <<"pgsql-123">>,
|
||||
<<"type">> => <<"pgsql_connenctor">>}]}].
|
|
@ -2,18 +2,18 @@
|
|||
## EMQ X CONNECTOR Plugin
|
||||
##--------------------------------------------------------------------
|
||||
|
||||
## Base directory for emqx_connector indicating where to load configs from disk.
|
||||
##
|
||||
## Value: String
|
||||
## Default: "{{ etc_dir }}/connectors/"
|
||||
emqx_connectors: [
|
||||
{id: "mysql-abc",
|
||||
type: mysql_connenctor,
|
||||
config: {}
|
||||
},
|
||||
{id: "pgsql-123",
|
||||
type: pgsql_connenctor,
|
||||
config: {}
|
||||
},
|
||||
connectors: [
|
||||
{id: "mysql-abc"
|
||||
resource_type: emqx_connector_mysql
|
||||
config: {
|
||||
server: "127.0.0.1:3306"
|
||||
database: mqtt
|
||||
pool_size: 1
|
||||
user: root
|
||||
password: public
|
||||
auto_reconnect: true
|
||||
ssl: false
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
|
|
|
@ -1,2 +1,7 @@
|
|||
%%-*- mode: erlang -*-
|
||||
%% emqx_connector config mapping
|
||||
|
||||
{mapping, "connectors", "connectors", [
|
||||
{default, []},
|
||||
{datatype, string}
|
||||
]}.
|
|
@ -22,7 +22,7 @@ stop(_State) ->
|
|||
|
||||
load_config() ->
|
||||
case hocon:load("etc/plugins/emqx_connector.conf", #{format => map}) of
|
||||
{ok, #{<<"emqx_connectors">> := Connectors}} ->
|
||||
{ok, #{<<"connectors">> := Connectors}} ->
|
||||
lists:foreach(fun load_connector/1, Connectors);
|
||||
{error, Reason} ->
|
||||
error(Reason)
|
||||
|
|
|
@ -5,7 +5,9 @@
|
|||
|
||||
-emqx_resource_api_path("connectors/mysql").
|
||||
|
||||
-export([fields/1]).
|
||||
-export([ fields/1
|
||||
, on_config_to_file/1
|
||||
]).
|
||||
|
||||
%% callbacks of behaviour emqx_resource
|
||||
-export([ on_start/2
|
||||
|
@ -18,10 +20,14 @@
|
|||
|
||||
-export([do_health_check/1]).
|
||||
|
||||
%%=====================================================================
|
||||
fields("config") ->
|
||||
emqx_connector_schema_lib:relational_db_fields() ++
|
||||
emqx_connector_schema_lib:ssl_fields().
|
||||
|
||||
on_config_to_file(#{server := Server} = Config) ->
|
||||
Config#{server => emqx_connector_schema_lib:ip_port_to_string(Server)}.
|
||||
|
||||
%% ===================================================================
|
||||
|
||||
on_start(InstId, #{server := {Host, Port},
|
||||
|
@ -58,7 +64,7 @@ on_query(InstId, {sql, SQL}, AfterQuery, #{poolname := PoolName} = State) ->
|
|||
case Result = ecpool:pick_and_do(PoolName, {mysql, query, [SQL]}, no_handover) of
|
||||
{error, Reason} ->
|
||||
logger:debug("mysql connector ~p do sql query failed, sql: ~p, reason: ~p", [InstId, SQL, Reason]),
|
||||
emqx_resource:query_failure(AfterQuery);
|
||||
emqx_resource:query_failed(AfterQuery);
|
||||
_ ->
|
||||
emqx_resource:query_success(AfterQuery)
|
||||
end,
|
||||
|
|
|
@ -6,6 +6,7 @@
|
|||
]).
|
||||
|
||||
-export([ to_ip_port/1
|
||||
, ip_port_to_string/1
|
||||
]).
|
||||
|
||||
-typerefl_from_string({ip_port/0, emqx_connector_schema_lib, to_ip_port}).
|
||||
|
@ -99,4 +100,7 @@ to_ip_port(Str) ->
|
|||
_ -> {error, Str}
|
||||
end;
|
||||
_ -> {error, Str}
|
||||
end.
|
||||
end.
|
||||
|
||||
ip_port_to_string({Ip, Port}) ->
|
||||
inet:ntoa(Ip) ++ ":" ++ integer_to_list(Port).
|
||||
|
|
|
@ -15,7 +15,7 @@
|
|||
%%--------------------------------------------------------------------
|
||||
-type resource_type() :: module().
|
||||
-type instance_id() :: binary().
|
||||
-type resource_config() :: jsx:json_term().
|
||||
-type resource_config() :: term().
|
||||
-type resource_spec() :: map().
|
||||
-type resource_state() :: term().
|
||||
-type resource_data() :: #{
|
||||
|
|
|
@ -7,7 +7,8 @@
|
|||
[kernel,
|
||||
stdlib,
|
||||
gproc,
|
||||
hocon
|
||||
hocon,
|
||||
jsx
|
||||
]},
|
||||
{env,[]},
|
||||
{modules, []},
|
||||
|
|
|
@ -65,6 +65,7 @@
|
|||
, call_health_check/3 %% verify if the resource is working normally
|
||||
, call_stop/3 %% stop the instance
|
||||
, call_config_merge/4 %% merge the config when updating
|
||||
, call_config_to_file/2
|
||||
]).
|
||||
|
||||
-export([ list_instances/0 %% list all the instances, id only.
|
||||
|
@ -85,12 +86,15 @@
|
|||
, on_health_check/2
|
||||
, on_api_reply_format/1
|
||||
, on_config_merge/3
|
||||
, on_config_to_file/1
|
||||
]).
|
||||
|
||||
-callback on_api_reply_format(resource_data()) -> map().
|
||||
|
||||
-callback on_config_merge(resource_config(), resource_config(), term()) -> resource_config().
|
||||
|
||||
-callback on_config_to_file(resource_config()) -> jsx:json_term().
|
||||
|
||||
%% when calling emqx_resource:start/1
|
||||
-callback on_start(instance_id(), resource_config()) ->
|
||||
{ok, resource_state()} | {error, Reason :: term()}.
|
||||
|
@ -241,6 +245,10 @@ call_stop(InstId, Mod, ResourceState) ->
|
|||
call_config_merge(Mod, OldConfig, NewConfig, Params) ->
|
||||
?SAFE_CALL(Mod:on_config_merge(OldConfig, NewConfig, Params)).
|
||||
|
||||
-spec call_config_to_file(module(), resource_config()) -> jsx:json_term().
|
||||
call_config_to_file(Mod, Config) ->
|
||||
?SAFE_CALL(Mod:on_config_to_file(Config)).
|
||||
|
||||
-spec parse_config(resource_type(), binary() | term()) ->
|
||||
{ok, resource_config()} | {error, term()}.
|
||||
parse_config(ResourceType, RawConfig) when is_binary(RawConfig) ->
|
||||
|
@ -271,7 +279,7 @@ resource_type_from_str(ResourceType) ->
|
|||
false -> {error, {invalid_resource, Mod}}
|
||||
end
|
||||
catch error:badarg ->
|
||||
{error, {resourec_not_found, ResourceType}}
|
||||
{error, {resource_not_found, ResourceType}}
|
||||
end.
|
||||
|
||||
call_instance(InstId, Query) ->
|
||||
|
|
|
@ -34,7 +34,7 @@ get(Mod, #{id := Id}, _Params) ->
|
|||
|
||||
put(Mod, #{id := Id}, Params) ->
|
||||
ConfigParams = proplists:get_value(<<"config">>, Params),
|
||||
ResourceTypeStr = proplists:get_value(<<"resource_type">>, Params),
|
||||
ResourceTypeStr = proplists:get_value(<<"resource_type">>, Params, #{}),
|
||||
case emqx_resource:resource_type_from_str(ResourceTypeStr) of
|
||||
{ok, ResourceType} ->
|
||||
do_put(Mod, stringnify(Id), ConfigParams, ResourceType, Params);
|
||||
|
|
|
@ -110,8 +110,8 @@ load_config(RawConfig) when is_binary(RawConfig) ->
|
|||
Error -> Error
|
||||
end;
|
||||
|
||||
load_config(#{<<"id">> := Id, <<"resource_type">> := ResourceTypeStr,
|
||||
<<"config">> := MapConfig}) ->
|
||||
load_config(#{<<"id">> := Id, <<"resource_type">> := ResourceTypeStr} = Config) ->
|
||||
MapConfig = maps:get(<<"config">>, Config, #{}),
|
||||
case emqx_resource:resource_type_from_str(ResourceTypeStr) of
|
||||
{ok, ResourceType} -> parse_and_load_config(Id, ResourceType, MapConfig);
|
||||
Error -> Error
|
||||
|
@ -130,8 +130,11 @@ create_local(InstId, ResourceType, InstConf) ->
|
|||
end.
|
||||
|
||||
save_config_to_disk(InstId, ResourceType, Config) ->
|
||||
%% TODO: send an event to the config handler, and the hander (single process)
|
||||
%% will dump configs for all instances (from an ETS table) to a file.
|
||||
file:write_file(filename:join([emqx_data_dir(), binary_to_list(InstId) ++ ".conf"]),
|
||||
jsx:encode(#{id => InstId, resource_type => ResourceType, config => Config})).
|
||||
jsx:encode(#{id => InstId, resource_type => ResourceType,
|
||||
config => emqx_resource:call_config_to_file(Config)})).
|
||||
|
||||
emqx_data_dir() ->
|
||||
"data".
|
||||
|
|
Loading…
Reference in New Issue