Add basic southbound statistics

[lispflowmapping.git] / mappingservice / southbound / src / main / java / org / opendaylight / lispflowmapping / southbound / LispSouthboundRPC.java

diff --git a/mappingservice/southbound/src/main/java/org/opendaylight/lispflowmapping/southbound/LispSouthboundRPC.java b/mappingservice/southbound/src/main/java/org/opendaylight/lispflowmapping/southbound/LispSouthboundRPC.java
index ac3fec3db8b1e280d7e79d6e697761f709032f4e..d6068407241366e8dc62f119f6b07e70f03983e8 100644 (file)
--- a/mappingservice/southbound/src/main/java/org/opendaylight/lispflowmapping/southbound/LispSouthboundRPC.java
+++ b/mappingservice/southbound/src/main/java/org/opendaylight/lispflowmapping/southbound/LispSouthboundRPC.java
@@ -9,17 +9,26 @@
 package org.opendaylight.lispflowmapping.southbound;
 
 import java.nio.ByteBuffer;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.concurrent.Future;
 
 import org.opendaylight.lispflowmapping.lisp.serializer.MapNotifySerializer;
 import org.opendaylight.lispflowmapping.lisp.serializer.MapRegisterSerializer;
 import org.opendaylight.lispflowmapping.lisp.serializer.MapReplySerializer;
 import org.opendaylight.lispflowmapping.lisp.serializer.MapRequestSerializer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.MessageType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.GetStatsOutput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.GetStatsOutputBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.LispSbService;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapNotifyInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapRegisterInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapReplyInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapRequestInput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.ctrl.msg.stats.ControlMessage;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.ctrl.msg.stats.ControlMessageBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.get.stats.output.ControlMessageStatsBuilder;
+import org.opendaylight.yangtools.yang.common.RpcError;
 import org.opendaylight.yangtools.yang.common.RpcResult;
 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
 import org.slf4j.Logger;
@@ -39,10 +48,6 @@ public class LispSouthboundRPC implements LispSbService {
 
     protected static final Logger LOG = LoggerFactory.getLogger(LispSouthboundRPC.class);
 
-    private final String MAP_NOTIFY = "MapNotify";
-    private final String MAP_REPlY = "MapReply";
-    private final String MAP_REQUEST = "MapRequest";
-    private final String MAP_REGISTER = "MapRegister";
     private final LispSouthboundPlugin lispSbPlugin;
 
     public LispSouthboundRPC(LispSouthboundPlugin lispSbPlugin) {
@@ -55,7 +60,7 @@ public class LispSouthboundRPC implements LispSbService {
         LOG.trace("sendMapNotify called!!");
         if (mapNotifyInput != null) {
             ByteBuffer outBuffer = MapNotifySerializer.getInstance().serialize(mapNotifyInput.getMapNotify());
-            lispSbPlugin.handleSerializedLispBuffer(mapNotifyInput.getTransportAddress(), outBuffer, MAP_NOTIFY);
+            lispSbPlugin.handleSerializedLispBuffer(mapNotifyInput.getTransportAddress(), outBuffer, MessageType.MapNotify);
         } else {
             LOG.warn("MapNotify was null");
             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
@@ -68,7 +73,7 @@ public class LispSouthboundRPC implements LispSbService {
         LOG.trace("sendMapReply called!!");
         if (mapReplyInput != null) {
             ByteBuffer outBuffer = MapReplySerializer.getInstance().serialize(mapReplyInput.getMapReply());
-            lispSbPlugin.handleSerializedLispBuffer(mapReplyInput.getTransportAddress(), outBuffer, MAP_REPlY);
+            lispSbPlugin.handleSerializedLispBuffer(mapReplyInput.getTransportAddress(), outBuffer, MessageType.MapReply);
         } else {
             LOG.warn("MapReply was null");
             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
@@ -81,7 +86,7 @@ public class LispSouthboundRPC implements LispSbService {
         LOG.trace("sendMapRequest called!!");
         if (mapRequestInput != null) {
             ByteBuffer outBuffer = MapRequestSerializer.getInstance().serialize(mapRequestInput.getMapRequest());
-            lispSbPlugin.handleSerializedLispBuffer(mapRequestInput.getTransportAddress(), outBuffer, MAP_REQUEST);
+            lispSbPlugin.handleSerializedLispBuffer(mapRequestInput.getTransportAddress(), outBuffer, MessageType.MapRequest);
         } else {
             LOG.debug("MapRequest was null");
             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
@@ -94,11 +99,68 @@ public class LispSouthboundRPC implements LispSbService {
         LOG.trace("sendMapRegister called!!");
         if (mapRegisterInput != null) {
             ByteBuffer outBuffer = MapRegisterSerializer.getInstance().serialize(mapRegisterInput.getMapRegister());
-            lispSbPlugin.handleSerializedLispBuffer(mapRegisterInput.getTransportAddress(), outBuffer, MAP_REGISTER);
+            lispSbPlugin.handleSerializedLispBuffer(mapRegisterInput.getTransportAddress(), outBuffer, MessageType.MapRegister);
         } else {
             LOG.debug("MapRegister was null");
             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
         }
         return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
     }
+
+    @Override
+    public Future<RpcResult<GetStatsOutput>> getStats() {
+        LOG.trace("getStats called!!");
+
+        RpcResultBuilder<GetStatsOutput> rpcResultBuilder;
+
+        LispSouthboundStats stats = lispSbPlugin.getStats();
+
+        if (stats == null) {
+            rpcResultBuilder = RpcResultBuilder.<GetStatsOutput>failed()
+                    .withError(RpcError.ErrorType.APPLICATION, "data-missing", "No stats found");
+        } else {
+            rpcResultBuilder = RpcResultBuilder.success(createGetStatsOutput(stats));
+        }
+        return Futures.immediateFuture(rpcResultBuilder.build());
+    }
+
+    @Override
+    public Future<RpcResult<Void>> resetStats() {
+        LOG.trace("resetStats called!!");
+
+        LispSouthboundStats stats = lispSbPlugin.getStats();
+
+        if (stats == null) {
+            return Futures.immediateFuture(RpcResultBuilder.<Void> failed()
+                    .withError(RpcError.ErrorType.APPLICATION, "data-missing", "No stats found")
+                    .build());
+        } else {
+            stats.resetStats();
+            return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
+        }
+    }
+
+    private static GetStatsOutput createGetStatsOutput(LispSouthboundStats stats) {
+        long rxStats[] = stats.getRx();
+        long txStats[] = stats.getTx();
+
+        ControlMessageStatsBuilder cmsb = new ControlMessageStatsBuilder();
+        cmsb.setRxUnknown(stats.getRxUnknown());
+        cmsb.setTxErrors(stats.getTxErrors());
+
+        List<ControlMessage> messages = new ArrayList<ControlMessage>();
+        for (int i = 0; i <= LispSouthboundStats.MAX_LISP_TYPES; i++) {
+            if (MessageType.forValue(i) == null) {
+                continue;
+            }
+            ControlMessageBuilder cmb = new ControlMessageBuilder();
+            cmb.setMsgType(MessageType.forValue(i));
+            cmb.setRxCount(rxStats[i]);
+            cmb.setTxCount(txStats[i]);
+            messages.add(cmb.build());
+        }
+
+        cmsb.setControlMessage(messages);
+        return new GetStatsOutputBuilder().setControlMessageStats(cmsb.build()).build();
+    }
 }