未知ai事件处理

apps/iot/src/iot_device.erl

@@ -22,6 +22,8 @@
 -export([init/1, handle_call/3, handle_cast/2, handle_info/2, terminate/2, code_change/3]).
 
 -record(state, {
+    %% 设备id
+    device_id :: integer(),
     device_uuid :: binary(),
     %% 设备是否授权
     auth_status :: integer(),
@@ -137,10 +139,10 @@ init([DeviceUUID]) when is_binary(DeviceUUID) ->
             lager:warning("[iot_device] device uuid: ~p, loaded from mysql failed", [DeviceUUID]),
             ignore
     end;
-init([#{<<"device_uuid">> := DeviceUUID, <<"authorize_status">> := AuthorizeStatus, <<"status">> := Status, <<"model_id">> := ModelId}]) ->
+init([#{<<"id">> := DeviceId, <<"device_uuid">> := DeviceUUID, <<"authorize_status">> := AuthorizeStatus, <<"status">> := Status, <<"model_id">> := ModelId}]) ->
     {ok, AiEventThrottle} = application:get_env(iot, ai_event_throttle),
 
-    {ok, #state{device_uuid = DeviceUUID, ai_event_throttle = AiEventThrottle,
+    {ok, #state{device_id = DeviceId, device_uuid = DeviceUUID, ai_event_throttle = AiEventThrottle,
         status = as_integer(Status), auth_status = as_integer(AuthorizeStatus), model_id = as_integer(ModelId)}}.
 
 %% @private
@@ -204,9 +206,10 @@ handle_cast({auth, false}, State = #state{device_uuid = DeviceUUID}) ->
     {noreply, State#state{auth_status = 0}};
 
 %% ai事件的延迟整流逻辑, 保证在间隔事件内只发送一次
-handle_cast({ai_event, EventType, Params0}, State = #state{device_uuid = DeviceUUID, ai_event_throttle = EventThrottle, ai_event_ttl = EventTTL}) ->
+handle_cast({ai_event, EventType, Params0}, State = #state{device_id = DeviceId, device_uuid = DeviceUUID, ai_event_throttle = EventThrottle, ai_event_ttl = EventTTL}) ->
     %% 查询数据库，覆盖params的部分信息
-    Params = rewrite_ai_event(EventType, Params0),
+    case rewrite_ai_event(EventType, Params0) of
+        {ok, Params} ->
             case maps:find(EventType, EventThrottle) of
                 {ok, Interval} ->
                     LastTimestamp = maps:get(EventType, EventTTL, 0),
@@ -227,6 +230,14 @@ handle_cast({ai_event, EventType, Params0}, State = #state{device_uuid = DeviceU
                     iot_ai_router:route_uuid(DeviceUUID, EventType, Params),
                     {noreply, State}
             end;
+        error ->
+            %% 未知事件类型，报警
+            {ok, EventCode} = maps:find(<<"event_code">>, Params0),
+            Warn = iolist_to_binary([<<"设备ID:">>, integer_to_binary(DeviceId), <<", 未知ai事件:">>, EventCode]),
+            iot_watchdog:warn(Warn),
+
+            {noreply, State}
+    end;
 
 %% 设备数据转发
 handle_cast({handle_data, Fields, Timestamp}, State = #state{device_uuid = DeviceUUID, model_id = ModelId}) ->
@@ -324,12 +335,12 @@ extract_build_location_code(<<BuildLocationCode:9/binary, Suffix/binary>>) ->
 extract_build_location_code(Code) when is_binary(Code) ->
     Code.
 
--spec rewrite_ai_event(EventType :: integer(), Params :: map()) -> NParams :: map().
+-spec rewrite_ai_event(EventType :: integer(), Params :: map()) -> {ok, NParams :: map()} | error.
 rewrite_ai_event(EventType, Params) when is_integer(EventType), is_map(Params) ->
     %% 查询数据库，覆盖params的部分信息
     case ai_event_bo:get_event_by_event_type(EventType) of
         {ok, #{<<"event_code">> := EventCode, <<"event_name">> := EventName}} ->
-            Params#{<<"event_code">> => EventCode, <<"description">> => EventName};
+            {ok, Params#{<<"event_code">> => EventCode, <<"description">> => EventName}};
         undefined ->
-            Params
+            error
     end.

apps/iot/src/iot_watchdog.erl

@@ -49,12 +49,14 @@
 %%% API
 %%%===================================================================
 
+%% 检测主机的硬盘 内存 CPU
 -spec detection(HostUUID :: binary(), Name :: binary(), Metric :: map()) -> no_return().
 detection(HostUUID, Name, Metric) when is_binary(HostUUID), is_binary(Name), is_map(Metric) ->
     gen_server:cast(?SERVER, {detection, HostUUID, Name, Metric});
 detection(HostUUID, Name, _) when is_binary(HostUUID), is_binary(Name) ->
     ok.
 
+%% 直接报警
 -spec warn(Warn :: binary()) -> no_return().
 warn(Warn) when is_binary(Warn) ->
     gen_server:cast(?SERVER, {warn, Warn}).