fix(kafka_producer): make status `connecting` while the client fails to connect
Fixes https://emqx.atlassian.net/browse/EMQX-11408 To make it consistent with the previous bridge behavior. Also, introduces macros for resource status to avoid problems with typos.
This commit is contained in:
parent
36b5d58957
commit
b92821188b
|
@ -202,33 +202,36 @@ lookup(Type, Name) ->
|
||||||
%% The connector should always exist
|
%% The connector should always exist
|
||||||
%% ... but, in theory, there might be no channels associated to it when we try
|
%% ... but, in theory, there might be no channels associated to it when we try
|
||||||
%% to delete the connector, and then this reference will become dangling...
|
%% to delete the connector, and then this reference will become dangling...
|
||||||
InstanceData =
|
ConnectorData =
|
||||||
case emqx_resource:get_instance(ConnectorId) of
|
case emqx_resource:get_instance(ConnectorId) of
|
||||||
{ok, _, Data} ->
|
{ok, _, Data} ->
|
||||||
Data;
|
Data;
|
||||||
{error, not_found} ->
|
{error, not_found} ->
|
||||||
#{}
|
#{}
|
||||||
end,
|
end,
|
||||||
%% Find the Bridge V2 status from the InstanceData
|
%% Find the Bridge V2 status from the ConnectorData
|
||||||
Channels = maps:get(added_channels, InstanceData, #{}),
|
ConnectorStatus = maps:get(status, ConnectorData, undefined),
|
||||||
|
Channels = maps:get(added_channels, ConnectorData, #{}),
|
||||||
BridgeV2Id = id(Type, Name, BridgeConnector),
|
BridgeV2Id = id(Type, Name, BridgeConnector),
|
||||||
ChannelStatus = maps:get(BridgeV2Id, Channels, undefined),
|
ChannelStatus = maps:get(BridgeV2Id, Channels, undefined),
|
||||||
{DisplayBridgeV2Status, ErrorMsg} =
|
{DisplayBridgeV2Status, ErrorMsg} =
|
||||||
case ChannelStatus of
|
case {ChannelStatus, ConnectorStatus} of
|
||||||
#{status := connected} ->
|
{#{status := ?status_connected}, _} ->
|
||||||
{connected, <<"">>};
|
{?status_connected, <<"">>};
|
||||||
#{status := Status, error := undefined} ->
|
{#{error := resource_not_operational}, ?status_connecting} ->
|
||||||
|
{?status_connecting, <<"Not installed">>};
|
||||||
|
{#{status := Status, error := undefined}, _} ->
|
||||||
{Status, <<"Unknown reason">>};
|
{Status, <<"Unknown reason">>};
|
||||||
#{status := Status, error := Error} ->
|
{#{status := Status, error := Error}, _} ->
|
||||||
{Status, emqx_utils:readable_error_msg(Error)};
|
{Status, emqx_utils:readable_error_msg(Error)};
|
||||||
undefined ->
|
{undefined, _} ->
|
||||||
{disconnected, <<"Pending installation">>}
|
{?status_disconnected, <<"Not installed">>}
|
||||||
end,
|
end,
|
||||||
{ok, #{
|
{ok, #{
|
||||||
type => bin(Type),
|
type => bin(Type),
|
||||||
name => bin(Name),
|
name => bin(Name),
|
||||||
raw_config => RawConf,
|
raw_config => RawConf,
|
||||||
resource_data => InstanceData,
|
resource_data => ConnectorData,
|
||||||
status => DisplayBridgeV2Status,
|
status => DisplayBridgeV2Status,
|
||||||
error => ErrorMsg
|
error => ErrorMsg
|
||||||
}}
|
}}
|
||||||
|
|
|
@ -20,6 +20,7 @@
|
||||||
|
|
||||||
-include_lib("eunit/include/eunit.hrl").
|
-include_lib("eunit/include/eunit.hrl").
|
||||||
-include_lib("common_test/include/ct.hrl").
|
-include_lib("common_test/include/ct.hrl").
|
||||||
|
-include_lib("emqx_resource/include/emqx_resource.hrl").
|
||||||
|
|
||||||
-import(emqx_common_test_helpers, [on_exit/1]).
|
-import(emqx_common_test_helpers, [on_exit/1]).
|
||||||
|
|
||||||
|
@ -43,7 +44,7 @@ con_schema() ->
|
||||||
{
|
{
|
||||||
con_type(),
|
con_type(),
|
||||||
hoconsc:mk(
|
hoconsc:mk(
|
||||||
hoconsc:map(name, typerefl:map()),
|
hoconsc:map(name, hoconsc:ref(?MODULE, connector_config)),
|
||||||
#{
|
#{
|
||||||
desc => <<"Test Connector Config">>,
|
desc => <<"Test Connector Config">>,
|
||||||
required => false
|
required => false
|
||||||
|
@ -52,6 +53,15 @@ con_schema() ->
|
||||||
}
|
}
|
||||||
].
|
].
|
||||||
|
|
||||||
|
fields(connector_config) ->
|
||||||
|
[
|
||||||
|
{enable, hoconsc:mk(typerefl:boolean(), #{})},
|
||||||
|
{resource_opts, hoconsc:mk(typerefl:map(), #{})},
|
||||||
|
{on_start_fun, hoconsc:mk(typerefl:binary(), #{})},
|
||||||
|
{on_get_status_fun, hoconsc:mk(typerefl:binary(), #{})},
|
||||||
|
{on_add_channel_fun, hoconsc:mk(typerefl:binary(), #{})}
|
||||||
|
].
|
||||||
|
|
||||||
con_config() ->
|
con_config() ->
|
||||||
#{
|
#{
|
||||||
<<"enable">> => true,
|
<<"enable">> => true,
|
||||||
|
@ -112,6 +122,7 @@ setup_mocks() ->
|
||||||
|
|
||||||
catch meck:new(emqx_connector_schema, MeckOpts),
|
catch meck:new(emqx_connector_schema, MeckOpts),
|
||||||
meck:expect(emqx_connector_schema, fields, 1, con_schema()),
|
meck:expect(emqx_connector_schema, fields, 1, con_schema()),
|
||||||
|
meck:expect(emqx_connector_schema, connector_type_to_bridge_types, 1, [con_type()]),
|
||||||
|
|
||||||
catch meck:new(emqx_connector_resource, MeckOpts),
|
catch meck:new(emqx_connector_resource, MeckOpts),
|
||||||
meck:expect(emqx_connector_resource, connector_to_resource_type, 1, con_mod()),
|
meck:expect(emqx_connector_resource, connector_to_resource_type, 1, con_mod()),
|
||||||
|
@ -159,15 +170,7 @@ init_per_testcase(_TestCase, Config) ->
|
||||||
ets:new(fun_table_name(), [named_table, public]),
|
ets:new(fun_table_name(), [named_table, public]),
|
||||||
%% Create a fake connector
|
%% Create a fake connector
|
||||||
{ok, _} = emqx_connector:create(con_type(), con_name(), con_config()),
|
{ok, _} = emqx_connector:create(con_type(), con_name(), con_config()),
|
||||||
[
|
Config.
|
||||||
{mocked_mods, [
|
|
||||||
emqx_connector_schema,
|
|
||||||
emqx_connector_resource,
|
|
||||||
|
|
||||||
emqx_bridge_v2
|
|
||||||
]}
|
|
||||||
| Config
|
|
||||||
].
|
|
||||||
|
|
||||||
end_per_testcase(_TestCase, _Config) ->
|
end_per_testcase(_TestCase, _Config) ->
|
||||||
ets:delete(fun_table_name()),
|
ets:delete(fun_table_name()),
|
||||||
|
@ -846,6 +849,51 @@ t_start_operation_when_on_add_channel_gives_error(_Config) ->
|
||||||
),
|
),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
t_lookup_status_when_connecting(_Config) ->
|
||||||
|
ResponseETS = ets:new(response_ets, [public]),
|
||||||
|
ets:insert(ResponseETS, {on_get_status_value, ?status_connecting}),
|
||||||
|
OnGetStatusFun = wrap_fun(fun() ->
|
||||||
|
ets:lookup_element(ResponseETS, on_get_status_value, 2)
|
||||||
|
end),
|
||||||
|
|
||||||
|
ConnectorConfig = emqx_utils_maps:deep_merge(con_config(), #{
|
||||||
|
<<"on_get_status_fun">> => OnGetStatusFun,
|
||||||
|
<<"resource_opts">> => #{<<"start_timeout">> => 100}
|
||||||
|
}),
|
||||||
|
ConnectorName = ?FUNCTION_NAME,
|
||||||
|
ct:pal("connector config:\n ~p", [ConnectorConfig]),
|
||||||
|
{ok, _} = emqx_connector:create(con_type(), ConnectorName, ConnectorConfig),
|
||||||
|
|
||||||
|
ActionName = my_test_action,
|
||||||
|
ChanStatusFun = wrap_fun(fun() -> ?status_disconnected end),
|
||||||
|
ActionConfig = (bridge_config())#{
|
||||||
|
<<"on_get_channel_status_fun">> => ChanStatusFun,
|
||||||
|
<<"connector">> => atom_to_binary(ConnectorName)
|
||||||
|
},
|
||||||
|
ct:pal("action config:\n ~p", [ActionConfig]),
|
||||||
|
{ok, _} = emqx_bridge_v2:create(bridge_type(), ActionName, ActionConfig),
|
||||||
|
|
||||||
|
%% Top-level status is connecting if the connector status is connecting, but the
|
||||||
|
%% channel is not yet installed. `resource_data.added_channels.$channel_id.status'
|
||||||
|
%% contains true internal status.
|
||||||
|
{ok, Res} = emqx_bridge_v2:lookup(bridge_type(), ActionName),
|
||||||
|
?assertMatch(
|
||||||
|
#{
|
||||||
|
%% This is the action's public status
|
||||||
|
status := ?status_connecting,
|
||||||
|
resource_data :=
|
||||||
|
#{
|
||||||
|
%% This is the connector's status
|
||||||
|
status := ?status_connecting
|
||||||
|
}
|
||||||
|
},
|
||||||
|
Res
|
||||||
|
),
|
||||||
|
#{resource_data := #{added_channels := Channels}} = Res,
|
||||||
|
[{_Id, ChannelData}] = maps:to_list(Channels),
|
||||||
|
?assertMatch(#{status := ?status_disconnected}, ChannelData),
|
||||||
|
ok.
|
||||||
|
|
||||||
%% Helper Functions
|
%% Helper Functions
|
||||||
|
|
||||||
wait_until(Fun) ->
|
wait_until(Fun) ->
|
||||||
|
|
|
@ -587,7 +587,7 @@ t_broken_bridge_config(Config) ->
|
||||||
<<"type">> := ?BRIDGE_TYPE,
|
<<"type">> := ?BRIDGE_TYPE,
|
||||||
<<"connector">> := <<"does_not_exist">>,
|
<<"connector">> := <<"does_not_exist">>,
|
||||||
<<"status">> := <<"disconnected">>,
|
<<"status">> := <<"disconnected">>,
|
||||||
<<"error">> := <<"Pending installation">>
|
<<"error">> := <<"Not installed">>
|
||||||
}
|
}
|
||||||
]},
|
]},
|
||||||
request_json(get, uri([?ROOT]), Config)
|
request_json(get, uri([?ROOT]), Config)
|
||||||
|
@ -640,7 +640,7 @@ t_fix_broken_bridge_config(Config) ->
|
||||||
<<"type">> := ?BRIDGE_TYPE,
|
<<"type">> := ?BRIDGE_TYPE,
|
||||||
<<"connector">> := <<"does_not_exist">>,
|
<<"connector">> := <<"does_not_exist">>,
|
||||||
<<"status">> := <<"disconnected">>,
|
<<"status">> := <<"disconnected">>,
|
||||||
<<"error">> := <<"Pending installation">>
|
<<"error">> := <<"Not installed">>
|
||||||
}
|
}
|
||||||
]},
|
]},
|
||||||
request_json(get, uri([?ROOT]), Config)
|
request_json(get, uri([?ROOT]), Config)
|
||||||
|
|
|
@ -43,8 +43,8 @@ on_start(
|
||||||
) ->
|
) ->
|
||||||
Fun = emqx_bridge_v2_SUITE:unwrap_fun(FunRef),
|
Fun = emqx_bridge_v2_SUITE:unwrap_fun(FunRef),
|
||||||
Fun(Conf);
|
Fun(Conf);
|
||||||
on_start(_InstId, _Config) ->
|
on_start(_InstId, Config) ->
|
||||||
{ok, #{}}.
|
{ok, Config}.
|
||||||
|
|
||||||
on_add_channel(
|
on_add_channel(
|
||||||
_InstId,
|
_InstId,
|
||||||
|
|
|
@ -481,11 +481,11 @@ on_get_status(
|
||||||
case wolff_client_sup:find_client(ClientId) of
|
case wolff_client_sup:find_client(ClientId) of
|
||||||
{ok, Pid} ->
|
{ok, Pid} ->
|
||||||
case wolff_client:check_connectivity(Pid) of
|
case wolff_client:check_connectivity(Pid) of
|
||||||
ok -> connected;
|
ok -> ?status_connected;
|
||||||
{error, Error} -> {connecting, State, Error}
|
{error, Error} -> {?status_connecting, State, Error}
|
||||||
end;
|
end;
|
||||||
{error, _Reason} ->
|
{error, _Reason} ->
|
||||||
connecting
|
?status_connecting
|
||||||
end.
|
end.
|
||||||
|
|
||||||
on_get_channel_status(
|
on_get_channel_status(
|
||||||
|
@ -499,10 +499,10 @@ on_get_channel_status(
|
||||||
#{kafka_topic := KafkaTopic} = maps:get(ChannelId, Channels),
|
#{kafka_topic := KafkaTopic} = maps:get(ChannelId, Channels),
|
||||||
try
|
try
|
||||||
ok = check_topic_and_leader_connections(ClientId, KafkaTopic),
|
ok = check_topic_and_leader_connections(ClientId, KafkaTopic),
|
||||||
connected
|
?status_connected
|
||||||
catch
|
catch
|
||||||
throw:#{reason := restarting} ->
|
throw:#{reason := restarting} ->
|
||||||
conneting
|
?status_connecting
|
||||||
end.
|
end.
|
||||||
|
|
||||||
check_topic_and_leader_connections(ClientId, KafkaTopic) ->
|
check_topic_and_leader_connections(ClientId, KafkaTopic) ->
|
||||||
|
|
|
@ -23,6 +23,8 @@
|
||||||
-include_lib("snabbkaffe/include/snabbkaffe.hrl").
|
-include_lib("snabbkaffe/include/snabbkaffe.hrl").
|
||||||
-include_lib("brod/include/brod.hrl").
|
-include_lib("brod/include/brod.hrl").
|
||||||
|
|
||||||
|
-import(emqx_common_test_helpers, [on_exit/1]).
|
||||||
|
|
||||||
-define(TYPE, kafka_producer).
|
-define(TYPE, kafka_producer).
|
||||||
|
|
||||||
%%------------------------------------------------------------------------------
|
%%------------------------------------------------------------------------------
|
||||||
|
@ -55,6 +57,13 @@ end_per_suite(Config) ->
|
||||||
emqx_cth_suite:stop(Apps),
|
emqx_cth_suite:stop(Apps),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
init_per_testcase(_TestCase, Config) ->
|
||||||
|
Config.
|
||||||
|
|
||||||
|
end_per_testcase(_TestCase, _Config) ->
|
||||||
|
emqx_common_test_helpers:call_janitor(60_000),
|
||||||
|
ok.
|
||||||
|
|
||||||
%%-------------------------------------------------------------------------------------
|
%%-------------------------------------------------------------------------------------
|
||||||
%% Helper fns
|
%% Helper fns
|
||||||
%%-------------------------------------------------------------------------------------
|
%%-------------------------------------------------------------------------------------
|
||||||
|
@ -163,6 +172,16 @@ kafka_hosts_string() ->
|
||||||
KafkaPort = os:getenv("KAFKA_PLAIN_PORT", "9092"),
|
KafkaPort = os:getenv("KAFKA_PLAIN_PORT", "9092"),
|
||||||
KafkaHost ++ ":" ++ KafkaPort.
|
KafkaHost ++ ":" ++ KafkaPort.
|
||||||
|
|
||||||
|
create_connector(Name, Config) ->
|
||||||
|
Res = emqx_connector:create(?TYPE, Name, Config),
|
||||||
|
on_exit(fun() -> emqx_connector:remove(?TYPE, Name) end),
|
||||||
|
Res.
|
||||||
|
|
||||||
|
create_action(Name, Config) ->
|
||||||
|
Res = emqx_bridge_v2:create(?TYPE, Name, Config),
|
||||||
|
on_exit(fun() -> emqx_bridge_v2:remove(?TYPE, Name) end),
|
||||||
|
Res.
|
||||||
|
|
||||||
%%------------------------------------------------------------------------------
|
%%------------------------------------------------------------------------------
|
||||||
%% Testcases
|
%% Testcases
|
||||||
%%------------------------------------------------------------------------------
|
%%------------------------------------------------------------------------------
|
||||||
|
@ -302,3 +321,24 @@ t_unknown_topic(_Config) ->
|
||||||
emqx_bridge_v2_testlib:get_bridge_api(?TYPE, BridgeName)
|
emqx_bridge_v2_testlib:get_bridge_api(?TYPE, BridgeName)
|
||||||
),
|
),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
t_bad_url(_Config) ->
|
||||||
|
ConnectorName = <<"test_connector">>,
|
||||||
|
ActionName = <<"test_action">>,
|
||||||
|
ActionConfig = bridge_v2_config(<<"test_connector">>),
|
||||||
|
ConnectorConfig0 = connector_config(),
|
||||||
|
ConnectorConfig = ConnectorConfig0#{<<"bootstrap_hosts">> := <<"bad_host:9092">>},
|
||||||
|
?assertMatch({ok, _}, create_connector(ConnectorName, ConnectorConfig)),
|
||||||
|
?assertMatch({ok, _}, create_action(ActionName, ActionConfig)),
|
||||||
|
?assertMatch(
|
||||||
|
{ok, #{
|
||||||
|
resource_data :=
|
||||||
|
#{
|
||||||
|
status := connecting,
|
||||||
|
error := [#{reason := unresolvable_hostname}]
|
||||||
|
}
|
||||||
|
}},
|
||||||
|
emqx_connector:lookup(?TYPE, ConnectorName)
|
||||||
|
),
|
||||||
|
?assertMatch({ok, #{status := connecting}}, emqx_bridge_v2:lookup(?TYPE, ActionName)),
|
||||||
|
ok.
|
||||||
|
|
|
@ -13,6 +13,16 @@
|
||||||
%% See the License for the specific language governing permissions and
|
%% See the License for the specific language governing permissions and
|
||||||
%% limitations under the License.
|
%% limitations under the License.
|
||||||
%%--------------------------------------------------------------------
|
%%--------------------------------------------------------------------
|
||||||
|
|
||||||
|
%% bridge/connector/action status
|
||||||
|
-define(status_connected, connected).
|
||||||
|
-define(status_connecting, connecting).
|
||||||
|
-define(status_disconnected, disconnected).
|
||||||
|
%% Note: the `stopped' status can only be emitted by `emqx_resource_manager'... Modules
|
||||||
|
%% implementing `emqx_resource' behavior should not return it. The `rm_' prefix is to
|
||||||
|
%% remind us of that.
|
||||||
|
-define(rm_status_stopped, stopped).
|
||||||
|
|
||||||
-type resource_type() :: module().
|
-type resource_type() :: module().
|
||||||
-type resource_id() :: binary().
|
-type resource_id() :: binary().
|
||||||
-type channel_id() :: binary().
|
-type channel_id() :: binary().
|
||||||
|
@ -21,8 +31,12 @@
|
||||||
-type resource_config() :: term().
|
-type resource_config() :: term().
|
||||||
-type resource_spec() :: map().
|
-type resource_spec() :: map().
|
||||||
-type resource_state() :: term().
|
-type resource_state() :: term().
|
||||||
-type resource_status() :: connected | disconnected | connecting | stopped.
|
%% Note: the `stopped' status can only be emitted by `emqx_resource_manager'... Modules
|
||||||
-type channel_status() :: connected | connecting | disconnected.
|
%% implementing `emqx_resource' behavior should not return it.
|
||||||
|
-type resource_status() ::
|
||||||
|
?status_connected | ?status_disconnected | ?status_connecting | ?rm_status_stopped.
|
||||||
|
-type health_check_status() :: ?status_connected | ?status_disconnected | ?status_connecting.
|
||||||
|
-type channel_status() :: ?status_connected | ?status_connecting | ?status_disconnected.
|
||||||
-type callback_mode() :: always_sync | async_if_possible.
|
-type callback_mode() :: always_sync | async_if_possible.
|
||||||
-type query_mode() ::
|
-type query_mode() ::
|
||||||
simple_sync
|
simple_sync
|
||||||
|
|
Loading…
Reference in New Issue