yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #6351 from zhongwencool/prometheus-accept-header 

chore(HTTP): prometheus or json format by accept header
This commit is contained in:
zhongwencool 2021-12-06 16:12:54 +08:00 committed by GitHub
parent 1ae1460e44 f6c61189c4
commit fbe66cb9f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/emqx_prometheus/src/emqx_prometheus_api.erl
View File

@ -52,7 +52,13 @@ prometheus_data_api() ->
Metadata = #{
get => #{
description => <<"Get Prometheus Data">>,
responses => #{<<"200">> => schema(#{type => object})}
responses => #{<<"200">> =>
#{content =>
#{
'application/json' => #{schema => #{type => object}},
'text/plain' => #{schema => #{type => string}}
}}
}
}
},
{"/prometheus/stats", Metadata, stats}.
@ -72,8 +78,12 @@ prometheus(put, #{body := Body}) ->
end,
{200, emqx:get_raw_config([<<"prometheus">>], #{})}.
stats(get, #{query_string := Qs}) ->
Type = maps:get(<<"format_type">>, Qs, <<"json">>),
stats(get, #{headers := Headers}) ->
Type =
case maps:get(<<"accept">>, Headers, <<"text/plain">>) of
<<"application/json">> -> <<"json">>;
_ -> <<"prometheus">>
end,
Data = emqx_prometheus:collect(Type),
case Type of
<<"json">> -> {200, Data};