From 42d2f0be1032d874a1c276f83a8999ceaf08cd6e Mon Sep 17 00:00:00 2001
From: Andrew Shvayka <ashvayka@thingsboard.io>
Date: Fri, 29 Mar 2019 18:23:58 +0200
Subject: [PATCH] Geofencing rule nodes

---
 .../engine/geo/AbstractGeofencingNode.java    | 13 +++++++++++-
 .../engine/geo/EntityGeofencingState.java     |  2 +-
 .../engine/geo/TbGpsGeofencingActionNode.java | 20 ++++++++-----------
 ...bGpsGeofencingActionNodeConfiguration.java |  1 -
 .../engine/geo/TbGpsGeofencingFilterNode.java |  8 +-------
 ...bGpsGeofencingFilterNodeConfiguration.java | 13 ++++++++++--
 6 files changed, 33 insertions(+), 24 deletions(-)

diff --git a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/AbstractGeofencingNode.java b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/AbstractGeofencingNode.java
index f876967917..2aec28f08b 100644
--- a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/AbstractGeofencingNode.java
+++ b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/AbstractGeofencingNode.java
@@ -97,7 +97,14 @@ public abstract class AbstractGeofencingNode<T extends TbGpsGeofencingFilterNode
                 throw new TbNodeException("Missing perimeter definition!");
             }
         } else {
-            return config.getPerimeters();
+            Perimeter perimeter = new Perimeter();
+            perimeter.setPerimeterType(config.getPerimeterType());
+            perimeter.setCenterLatitude(config.getCenterLatitude());
+            perimeter.setCenterLongitude(config.getCenterLongitude());
+            perimeter.setRange(config.getRange());
+            perimeter.setRangeUnit(config.getRangeUnit());
+            perimeter.setPolygonsDefinition(config.getPolygonsDefinition());
+            return Collections.singletonList(perimeter);
         }
     }
 
@@ -116,5 +123,9 @@ public abstract class AbstractGeofencingNode<T extends TbGpsGeofencingFilterNode
         return value;
     }
 
+    @Override
+    public void destroy() {
+
+    }
 
 }
diff --git a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/EntityGeofencingState.java b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/EntityGeofencingState.java
index 4024cbf22e..16be08d73b 100644
--- a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/EntityGeofencingState.java
+++ b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/EntityGeofencingState.java
@@ -24,6 +24,6 @@ public class EntityGeofencingState {
 
     private boolean inside;
     private long stateSwitchTime;
-    private boolean staid;
+    private boolean stayed;
 
 }
diff --git a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNode.java b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNode.java
index 952109286f..26a5d55a5f 100644
--- a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNode.java
+++ b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNode.java
@@ -46,8 +46,8 @@ import java.util.concurrent.TimeoutException;
 @RuleNode(
         type = ComponentType.ACTION,
         name = "gps geofencing events",
-        configClazz = TbGpsGeofencingFilterNodeConfiguration.class,
-        relationTypes = {"Entered", "Left", "Stays Inside", "Stays Outside"},
+        configClazz = TbGpsGeofencingActionNodeConfiguration.class,
+        relationTypes = {"Entered", "Left", "Inside", "Outside"},
         nodeDescription = "Produces incoming messages using GPS based geofencing",
         nodeDetails = "Extracts latitude and longitude parameters from incoming message and returns different events based on configuration parameters")
 public class TbGpsGeofencingActionNode extends AbstractGeofencingNode<TbGpsGeofencingActionNodeConfiguration> {
@@ -68,7 +68,7 @@ public class TbGpsGeofencingActionNode extends AbstractGeofencingNode<TbGpsGeofe
                         .get(1, TimeUnit.MINUTES);
                 if (entry.isPresent()) {
                     JsonObject element = parser.parse(entry.get().getValueAsString()).getAsJsonObject();
-                    return new EntityGeofencingState(element.get("inside").getAsBoolean(), element.get("stateSwitchTime").getAsLong(), element.get("staid").getAsBoolean());
+                    return new EntityGeofencingState(element.get("inside").getAsBoolean(), element.get("stateSwitchTime").getAsLong(), element.get("stayed").getAsBoolean());
                 } else {
                     return new EntityGeofencingState(false, 0L, false);
                 }
@@ -79,10 +79,11 @@ public class TbGpsGeofencingActionNode extends AbstractGeofencingNode<TbGpsGeofe
         if (entityState.getStateSwitchTime() == 0L || entityState.isInside() != matches) {
             switchState(ctx, msg.getOriginator(), entityState, matches, ts);
             ctx.tellNext(msg, matches ? "Entered" : "Left");
-        } else if (!entityState.isStaid()) {
+        } else if (!entityState.isStayed()) {
             long stayTime = ts - entityState.getStateSwitchTime();
             if (stayTime > (entityState.isInside() ? config.getMinInsideDuration() : config.getMinOutsideDuration())) {
                 setStaid(ctx, msg.getOriginator(), entityState);
+                ctx.tellNext(msg, entityState.isInside() ? "Inside" : "Outside");
             }
         }
     }
@@ -90,12 +91,12 @@ public class TbGpsGeofencingActionNode extends AbstractGeofencingNode<TbGpsGeofe
     private void switchState(TbContext ctx, EntityId entityId, EntityGeofencingState entityState, boolean matches, long ts) {
         entityState.setInside(matches);
         entityState.setStateSwitchTime(ts);
-        entityState.setStaid(false);
+        entityState.setStayed(false);
         persist(ctx, entityId, entityState);
     }
 
     private void setStaid(TbContext ctx, EntityId entityId, EntityGeofencingState entityState) {
-        entityState.setStaid(true);
+        entityState.setStayed(true);
         persist(ctx, entityId, entityState);
     }
 
@@ -103,17 +104,12 @@ public class TbGpsGeofencingActionNode extends AbstractGeofencingNode<TbGpsGeofe
         JsonObject object = new JsonObject();
         object.addProperty("inside", entityState.isInside());
         object.addProperty("stateSwitchTime", entityState.getStateSwitchTime());
-        object.addProperty("staid", entityState.isStaid());
+        object.addProperty("stayed", entityState.isStayed());
         AttributeKvEntry entry = new BaseAttributeKvEntry(new StringDataEntry(ctx.getNodeId(), gson.toJson(object)), System.currentTimeMillis());
         List<AttributeKvEntry> attributeKvEntryList = Collections.singletonList(entry);
         ctx.getAttributesService().save(ctx.getTenantId(), entityId, DataConstants.SERVER_SCOPE, attributeKvEntryList);
     }
 
-    @Override
-    public void destroy() {
-
-    }
-
     @Override
     protected Class<TbGpsGeofencingActionNodeConfiguration> getConfigClazz() {
         return TbGpsGeofencingActionNodeConfiguration.class;
diff --git a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNodeConfiguration.java b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNodeConfiguration.java
index dac9e7d23d..f6b275e008 100644
--- a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNodeConfiguration.java
+++ b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingActionNodeConfiguration.java
@@ -37,7 +37,6 @@ public class TbGpsGeofencingActionNodeConfiguration extends TbGpsGeofencingFilte
         configuration.setLatitudeKeyName("latitude");
         configuration.setLongitudeKeyName("longitude");
         configuration.setFetchPerimeterInfoFromMessage(true);
-        configuration.setPerimeters(Collections.emptyList());
         configuration.setMinInsideDuration(TimeUnit.MINUTES.toMillis(1));
         configuration.setMinOutsideDuration(TimeUnit.MINUTES.toMillis(1));
         return configuration;
diff --git a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNode.java b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNode.java
index 66a23ca2a8..e816cb6cdc 100644
--- a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNode.java
+++ b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNode.java
@@ -55,13 +55,7 @@ public class TbGpsGeofencingFilterNode extends AbstractGeofencingNode<TbGpsGeofe
 
     @Override
     public void onMsg(TbContext ctx, TbMsg msg) throws TbNodeException {
-        boolean matches = checkMatches(msg);
-        ctx.tellNext(msg, matches ? "True" : "False");
-    }
-
-    @Override
-    public void destroy() {
-
+        ctx.tellNext(msg, checkMatches(msg) ? "True" : "False");
     }
 
     @Override
diff --git a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNodeConfiguration.java b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNodeConfiguration.java
index 7d54699741..ca2ac4c8d3 100644
--- a/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNodeConfiguration.java
+++ b/rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/geo/TbGpsGeofencingFilterNodeConfiguration.java
@@ -33,7 +33,17 @@ public class TbGpsGeofencingFilterNodeConfiguration implements NodeConfiguration
     private String latitudeKeyName;
     private String longitudeKeyName;
     private boolean fetchPerimeterInfoFromMessage;
-    private List<Perimeter> perimeters;
+
+    private PerimeterType perimeterType;
+
+    //For Polygons
+    private String polygonsDefinition;
+
+    //For Circles
+    private Double centerLatitude;
+    private Double centerLongitude;
+    private Double range;
+    private RangeUnit rangeUnit;
 
     @Override
     public TbGpsGeofencingFilterNodeConfiguration defaultConfiguration() {
@@ -41,7 +51,6 @@ public class TbGpsGeofencingFilterNodeConfiguration implements NodeConfiguration
         configuration.setLatitudeKeyName("latitude");
         configuration.setLongitudeKeyName("longitude");
         configuration.setFetchPerimeterInfoFromMessage(true);
-        configuration.setPerimeters(Collections.emptyList());
         return configuration;
     }
 }