From 70fb429c98a91e418b2e890979c4caea79cb55b5 Mon Sep 17 00:00:00 2001
From: William Cheng <wing328hk@gmail.com>
Date: Sat, 19 Jun 2021 10:50:52 +0800
Subject: [PATCH] update erlang server samples

---
 .../petstore/erlang-server/src/openapi_logic_handler.erl       | 3 +--
 .../server/petstore/erlang-server/src/openapi_pet_handler.erl  | 2 ++
 samples/server/petstore/erlang-server/src/openapi_server.erl   | 2 +-
 .../petstore/erlang-server/src/openapi_store_handler.erl       | 2 ++
 .../server/petstore/erlang-server/src/openapi_user_handler.erl | 2 ++
 5 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/samples/server/petstore/erlang-server/src/openapi_logic_handler.erl b/samples/server/petstore/erlang-server/src/openapi_logic_handler.erl
index 2c05ce24836..1a36f0a40f4 100644
--- a/samples/server/petstore/erlang-server/src/openapi_logic_handler.erl
+++ b/samples/server/petstore/erlang-server/src/openapi_logic_handler.erl
@@ -6,8 +6,7 @@
 -type handler_response() ::{
     Status :: cowboy:http_status(),
     Headers :: cowboy:http_headers(),
-    Body :: #{}
-}.
+    Body :: jsx:json_term()}.
 
 -export_type([handler_response/0]).
 
diff --git a/samples/server/petstore/erlang-server/src/openapi_pet_handler.erl b/samples/server/petstore/erlang-server/src/openapi_pet_handler.erl
index 203cc07b28a..a6cbaab177c 100644
--- a/samples/server/petstore/erlang-server/src/openapi_pet_handler.erl
+++ b/samples/server/petstore/erlang-server/src/openapi_pet_handler.erl
@@ -475,5 +475,7 @@ validate_headers(_, Req) -> {true, Req}.
 
 prepare_body(204, Body) when map_size(Body) == 0; length(Body) == 0 ->
     <<>>;
+prepare_body(304, Body) when map_size(Body) == 0; length(Body) == 0 ->
+    <<>>;
 prepare_body(_Code, Body) ->
     jsx:encode(Body).
diff --git a/samples/server/petstore/erlang-server/src/openapi_server.erl b/samples/server/petstore/erlang-server/src/openapi_server.erl
index 45f6d1a39d5..02500173821 100644
--- a/samples/server/petstore/erlang-server/src/openapi_server.erl
+++ b/samples/server/petstore/erlang-server/src/openapi_server.erl
@@ -64,4 +64,4 @@ get_default_opts(LogicHandler) ->
     #{env => get_default_dispatch(LogicHandler)}.
 
 store_key(Key, Value, Opts) ->
-    lists:keystore(Key, 1, Opts, {Key, Value}).
+    maps:put(Key, Value, Opts).
diff --git a/samples/server/petstore/erlang-server/src/openapi_store_handler.erl b/samples/server/petstore/erlang-server/src/openapi_store_handler.erl
index b7969837147..98341c066c2 100644
--- a/samples/server/petstore/erlang-server/src/openapi_store_handler.erl
+++ b/samples/server/petstore/erlang-server/src/openapi_store_handler.erl
@@ -270,5 +270,7 @@ validate_headers(_, Req) -> {true, Req}.
 
 prepare_body(204, Body) when map_size(Body) == 0; length(Body) == 0 ->
     <<>>;
+prepare_body(304, Body) when map_size(Body) == 0; length(Body) == 0 ->
+    <<>>;
 prepare_body(_Code, Body) ->
     jsx:encode(Body).
diff --git a/samples/server/petstore/erlang-server/src/openapi_user_handler.erl b/samples/server/petstore/erlang-server/src/openapi_user_handler.erl
index f0d9179392c..ad9f11d7300 100644
--- a/samples/server/petstore/erlang-server/src/openapi_user_handler.erl
+++ b/samples/server/petstore/erlang-server/src/openapi_user_handler.erl
@@ -437,5 +437,7 @@ validate_headers(_, Req) -> {true, Req}.
 
 prepare_body(204, Body) when map_size(Body) == 0; length(Body) == 0 ->
     <<>>;
+prepare_body(304, Body) when map_size(Body) == 0; length(Body) == 0 ->
+    <<>>;
 prepare_body(_Code, Body) ->
     jsx:encode(Body).