131 lines
3.8 KiB
Erlang
131 lines
3.8 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2022-2023 EMQ Technologies Co., Ltd. All Rights Reserved.
|
|
%%--------------------------------------------------------------------
|
|
-module(emqx_ee_bridge_mysql).
|
|
|
|
-include_lib("typerefl/include/types.hrl").
|
|
-include_lib("hocon/include/hoconsc.hrl").
|
|
-include_lib("emqx_bridge/include/emqx_bridge.hrl").
|
|
-include_lib("emqx_resource/include/emqx_resource.hrl").
|
|
|
|
-import(hoconsc, [mk/2, enum/1, ref/2]).
|
|
|
|
-export([
|
|
conn_bridge_examples/1
|
|
]).
|
|
|
|
-export([
|
|
namespace/0,
|
|
roots/0,
|
|
fields/1,
|
|
desc/1
|
|
]).
|
|
|
|
-define(DEFAULT_SQL, <<
|
|
"insert into t_mqtt_msg(msgid, topic, qos, payload, arrived) "
|
|
"values (${id}, ${topic}, ${qos}, ${payload}, FROM_UNIXTIME(${timestamp}/1000))"
|
|
>>).
|
|
|
|
%% -------------------------------------------------------------------------------------------------
|
|
%% api
|
|
|
|
conn_bridge_examples(Method) ->
|
|
[
|
|
#{
|
|
<<"mysql">> => #{
|
|
summary => <<"MySQL Bridge">>,
|
|
value => values(Method)
|
|
}
|
|
}
|
|
].
|
|
|
|
values(get) ->
|
|
maps:merge(values(post), ?METRICS_EXAMPLE);
|
|
values(post) ->
|
|
#{
|
|
enable => true,
|
|
type => mysql,
|
|
name => <<"foo">>,
|
|
server => <<"127.0.0.1:3306">>,
|
|
database => <<"test">>,
|
|
pool_size => 8,
|
|
username => <<"root">>,
|
|
password => <<"">>,
|
|
sql => ?DEFAULT_SQL,
|
|
local_topic => <<"local/topic/#">>,
|
|
resource_opts => #{
|
|
worker_pool_size => 1,
|
|
health_check_interval => ?HEALTHCHECK_INTERVAL_RAW,
|
|
auto_restart_interval => ?AUTO_RESTART_INTERVAL_RAW,
|
|
batch_size => ?DEFAULT_BATCH_SIZE,
|
|
batch_time => ?DEFAULT_BATCH_TIME,
|
|
query_mode => async,
|
|
max_queue_bytes => ?DEFAULT_QUEUE_SIZE
|
|
}
|
|
};
|
|
values(put) ->
|
|
values(post).
|
|
|
|
%% -------------------------------------------------------------------------------------------------
|
|
%% Hocon Schema Definitions
|
|
namespace() -> "bridge_mysql".
|
|
|
|
roots() -> [].
|
|
|
|
fields("config") ->
|
|
[
|
|
{enable, mk(boolean(), #{desc => ?DESC("config_enable"), default => true})},
|
|
{sql,
|
|
mk(
|
|
binary(),
|
|
#{desc => ?DESC("sql_template"), default => ?DEFAULT_SQL, format => <<"sql">>}
|
|
)},
|
|
{local_topic,
|
|
mk(
|
|
binary(),
|
|
#{desc => ?DESC("local_topic"), default => undefined}
|
|
)},
|
|
{resource_opts,
|
|
mk(
|
|
ref(?MODULE, "creation_opts"),
|
|
#{
|
|
required => false,
|
|
default => #{},
|
|
desc => ?DESC(emqx_resource_schema, <<"resource_opts">>)
|
|
}
|
|
)}
|
|
] ++
|
|
(emqx_connector_mysql:fields(config) --
|
|
emqx_connector_schema_lib:prepare_statement_fields());
|
|
fields("creation_opts") ->
|
|
Opts = emqx_resource_schema:fields("creation_opts"),
|
|
[O || {Field, _} = O <- Opts, not is_hidden_opts(Field)];
|
|
fields("post") ->
|
|
[type_field(), name_field() | fields("config")];
|
|
fields("put") ->
|
|
fields("config");
|
|
fields("get") ->
|
|
emqx_bridge_schema:status_fields() ++ fields("post").
|
|
|
|
desc("config") ->
|
|
?DESC("desc_config");
|
|
desc(Method) when Method =:= "get"; Method =:= "put"; Method =:= "post" ->
|
|
["Configuration for MySQL using `", string:to_upper(Method), "` method."];
|
|
desc("creation_opts" = Name) ->
|
|
emqx_resource_schema:desc(Name);
|
|
desc(_) ->
|
|
undefined.
|
|
|
|
%% -------------------------------------------------------------------------------------------------
|
|
%% internal
|
|
is_hidden_opts(Field) ->
|
|
lists:member(Field, [
|
|
async_inflight_window
|
|
]).
|
|
|
|
type_field() ->
|
|
{type, mk(enum([mysql]), #{required => true, desc => ?DESC("desc_type")})}.
|
|
|
|
name_field() ->
|
|
{name, mk(binary(), #{required => true, desc => ?DESC("desc_name")})}.
|