From 0dec0c9fcfeffee635886086200b12c78d9be754 Mon Sep 17 00:00:00 2001
From: anlicheng <244108715@qq.com>
Date: Mon, 4 Aug 2025 17:59:18 +0800
Subject: [PATCH] fix

---
 apps/dimension_apn/src/api_handler.erl | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/apps/dimension_apn/src/api_handler.erl b/apps/dimension_apn/src/api_handler.erl
index dd33c45..06566c7 100644
--- a/apps/dimension_apn/src/api_handler.erl
+++ b/apps/dimension_apn/src/api_handler.erl
@@ -86,16 +86,15 @@ read_body(Req, AccData) ->
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 
 %% 向用户推送数据
-handle_request("POST", "/api/push", #{<<"env">> := <<"prod">>}, Notifications) ->
-    {ok, Pid} = dimension_apn_worker:start_link(apns_prod),
+handle_request("POST", "/api/push", GetParams, Notifications) ->
+    Profile = case maps:get(<<"env">>, GetParams, <<"prod">>) of
+                  <<"prod">> ->
+                      apns_prod;
+                  <<"dev">> ->
+                      apns_dev
+              end,
+    {ok, Pid} = dimension_apn_worker:start_link(Profile),
     dimension_apn_worker:push(Pid, Notifications),
-
-    {ok, 200, dimension_utils:json_data(<<"success">>)};
-
-handle_request("POST", "/api/push", _, Notifications) ->
-    {ok, Pid} = dimension_apn_worker:start_link(apns_dev),
-    dimension_apn_worker:push(Pid, Notifications),
-
     {ok, 200, dimension_utils:json_data(<<"success">>)};
 
 handle_request(_, Path, _, _) ->