From 5cb8c977e4dcafc419b31493b78126fb96db687e Mon Sep 17 00:00:00 2001
From: LO Kam Tao Leo <leolo@leolo.org>
Date: Wed, 2 Nov 2022 20:20:36 +0000
Subject: [PATCH] Refactoring the code

---
 src/main/java/org/leolo/nrdatad/model/SmartEvent.java    | 2 ++
 src/main/java/org/leolo/nrdatad/model/SmartStepType.java | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/main/java/org/leolo/nrdatad/model/SmartEvent.java b/src/main/java/org/leolo/nrdatad/model/SmartEvent.java
index 54afeaa..b1e30ea 100644
--- a/src/main/java/org/leolo/nrdatad/model/SmartEvent.java
+++ b/src/main/java/org/leolo/nrdatad/model/SmartEvent.java
@@ -20,6 +20,8 @@ public enum SmartEvent {
     }
 
     public static SmartEvent parseCode(@NotNull String code){
+        if(code==null)
+            return null;
         for(SmartEvent se: SmartEvent.values()){
             if(code.equalsIgnoreCase(se.code)){
                 return se;
diff --git a/src/main/java/org/leolo/nrdatad/model/SmartStepType.java b/src/main/java/org/leolo/nrdatad/model/SmartStepType.java
index 0b1a7a9..5723885 100644
--- a/src/main/java/org/leolo/nrdatad/model/SmartStepType.java
+++ b/src/main/java/org/leolo/nrdatad/model/SmartStepType.java
@@ -54,6 +54,8 @@ public enum SmartStepType {
     }
 
     public static SmartStepType parseCode(@NotNull String code){
+        if(code==null)
+            return null;
         for(SmartStepType sst: SmartStepType.values()){
             if(code.equalsIgnoreCase(sst.code))
                 return sst;