122 lines
4.3 KiB
Erlang
122 lines
4.3 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2020-2023 EMQ Technologies Co., Ltd. All Rights Reserved.
|
|
%%
|
|
%% Licensed under the Apache License, Version 2.0 (the "License");
|
|
%% you may not use this file except in compliance with the License.
|
|
%% You may obtain a copy of the License at
|
|
%%
|
|
%% http://www.apache.org/licenses/LICENSE-2.0
|
|
%%
|
|
%% Unless required by applicable law or agreed to in writing, software
|
|
%% distributed under the License is distributed on an "AS IS" BASIS,
|
|
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
%% See the License for the specific language governing permissions and
|
|
%% limitations under the License.
|
|
%%--------------------------------------------------------------------
|
|
-type resource_type() :: module().
|
|
-type resource_id() :: binary().
|
|
-type raw_resource_config() :: binary() | raw_term_resource_config().
|
|
-type raw_term_resource_config() :: #{binary() => term()} | [raw_term_resource_config()].
|
|
-type resource_config() :: term().
|
|
-type resource_spec() :: map().
|
|
-type resource_state() :: term().
|
|
-type resource_status() :: connected | disconnected | connecting | stopped.
|
|
-type callback_mode() :: always_sync | async_if_possible.
|
|
-type query_mode() :: simple_sync | simple_async | sync | async | no_queries.
|
|
-type result() :: term().
|
|
-type reply_fun() :: {fun((result(), Args :: term()) -> any()), Args :: term()} | undefined.
|
|
-type query_opts() :: #{
|
|
%% The key used for picking a resource worker
|
|
pick_key => term(),
|
|
timeout => timeout(),
|
|
expire_at => infinity | integer(),
|
|
async_reply_fun => reply_fun(),
|
|
simple_query => boolean(),
|
|
is_buffer_supported => boolean(),
|
|
reply_to => reply_fun()
|
|
}.
|
|
-type resource_data() :: #{
|
|
id := resource_id(),
|
|
mod := module(),
|
|
callback_mode := callback_mode(),
|
|
query_mode := query_mode(),
|
|
config := resource_config(),
|
|
error := term(),
|
|
state := resource_state(),
|
|
status := resource_status()
|
|
}.
|
|
-type resource_group() :: binary().
|
|
-type creation_opts() :: #{
|
|
%%======================================= Deprecated Opts BEGIN
|
|
%% use health_check_interval instead
|
|
health_check_timeout => integer(),
|
|
%% use start_timeout instead
|
|
wait_for_resource_ready => integer(),
|
|
%% use health_check_interval instead
|
|
auto_retry_interval => integer(),
|
|
%% use health_check_interval instead
|
|
auto_restart_interval => pos_integer() | infinity,
|
|
%%======================================= Deprecated Opts END
|
|
worker_pool_size => non_neg_integer(),
|
|
%% use `integer()` compatibility to release 5.0.0 bpapi
|
|
health_check_interval => integer(),
|
|
%% We can choose to block the return of emqx_resource:start until
|
|
%% the resource connected, wait max to `start_timeout` ms.
|
|
start_timeout => pos_integer(),
|
|
%% If `start_after_created` is set to true, the resource is started right
|
|
%% after it is created. But note that a `started` resource is not guaranteed
|
|
%% to be `connected`.
|
|
start_after_created => boolean(),
|
|
batch_size => pos_integer(),
|
|
batch_time => pos_integer(),
|
|
max_buffer_bytes => pos_integer(),
|
|
query_mode => query_mode(),
|
|
resume_interval => pos_integer(),
|
|
inflight_window => pos_integer()
|
|
}.
|
|
-type query_result() ::
|
|
ok
|
|
| {ok, term()}
|
|
| {ok, term(), term()}
|
|
| {ok, term(), term(), term()}
|
|
| {error, {recoverable_error, term()}}
|
|
| {error, term()}.
|
|
|
|
-define(WORKER_POOL_SIZE, 16).
|
|
|
|
-define(DEFAULT_BUFFER_BYTES, 256 * 1024 * 1024).
|
|
-define(DEFAULT_BUFFER_BYTES_RAW, <<"256MB">>).
|
|
|
|
-define(DEFAULT_REQUEST_TTL, timer:seconds(45)).
|
|
-define(DEFAULT_REQUEST_TTL_RAW, <<"45s">>).
|
|
|
|
%% count
|
|
-define(DEFAULT_BATCH_SIZE, 1).
|
|
|
|
%% milliseconds
|
|
-define(DEFAULT_BATCH_TIME, 0).
|
|
-define(DEFAULT_BATCH_TIME_RAW, <<"0ms">>).
|
|
|
|
%% count
|
|
-define(DEFAULT_INFLIGHT, 100).
|
|
|
|
%% milliseconds
|
|
-define(HEALTHCHECK_INTERVAL, 15000).
|
|
-define(HEALTHCHECK_INTERVAL_RAW, <<"15s">>).
|
|
|
|
%% milliseconds
|
|
-define(DEFAULT_METRICS_FLUSH_INTERVAL, 5_000).
|
|
-define(DEFAULT_METRICS_FLUSH_INTERVAL_RAW, <<"5s">>).
|
|
|
|
%% milliseconds
|
|
-define(START_TIMEOUT, 5000).
|
|
-define(START_TIMEOUT_RAW, <<"5s">>).
|
|
|
|
%% boolean
|
|
-define(START_AFTER_CREATED, true).
|
|
|
|
-define(TEST_ID_PREFIX, "_probe_:").
|
|
-define(RES_METRICS, resource_metrics).
|
|
|
|
-define(RESOURCE_ALLOCATION_TAB, emqx_resource_allocations).
|