From cc79802d6c0d1e0dbf9a0a8e49d563f24c52afd5 Mon Sep 17 00:00:00 2001 From: Feng Lee Date: Wed, 18 Sep 2019 19:46:18 +0800 Subject: [PATCH] Add function 'get_caps/3' --- src/emqx_mqtt_caps.erl | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/emqx_mqtt_caps.erl b/src/emqx_mqtt_caps.erl index ec7f55330..25d2ee5e4 100644 --- a/src/emqx_mqtt_caps.erl +++ b/src/emqx_mqtt_caps.erl @@ -26,6 +26,7 @@ -export([ get_caps/1 , get_caps/2 + , get_caps/3 ]). -export([default/0]). @@ -114,10 +115,13 @@ get_caps(Zone) -> -spec(get_caps(emqx_zone:zone(), publish|subscribe) -> caps()). get_caps(Zone, publish) -> with_env(Zone, '$mqtt_pub_caps', fun pub_caps/1); - get_caps(Zone, subscribe) -> with_env(Zone, '$mqtt_sub_caps', fun sub_caps/1). +-spec(get_caps(emqx_zone:zone(), atom(), term()) -> term()). +get_caps(Zone, Cap, Def) -> + emqx_zone:get_env(Zone, Cap, Def). + pub_caps(Zone) -> filter_caps(?PUBCAP_KEYS, get_caps(Zone)).