mappingservice/southbound/src/main/java/org/opendaylight/lispflowmapping/southbound/LispSouthboundRPC.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.lispflowmapping.southbound;
  10
  11 import java.nio.ByteBuffer;
  12 import java.util.ArrayList;
  13 import java.util.List;
  14 import java.util.concurrent.Future;
  15
  16 import org.opendaylight.lispflowmapping.lisp.serializer.MapNotifySerializer;
  17 import org.opendaylight.lispflowmapping.lisp.serializer.MapRegisterSerializer;
  18 import org.opendaylight.lispflowmapping.lisp.serializer.MapReplySerializer;
  19 import org.opendaylight.lispflowmapping.lisp.serializer.MapRequestSerializer;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MessageType;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.GetStatsOutput;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.GetStatsOutputBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.OdlLispSbService;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapNotifyInput;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapRegisterInput;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapReplyInput;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.SendMapRequestInput;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.ctrl.msg.stats.ControlMessage;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.ctrl.msg.stats.ControlMessageBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.sb.rev150904.get.stats.output.ControlMessageStatsBuilder;
  31 import org.opendaylight.yangtools.yang.common.RpcError;
  32 import org.opendaylight.yangtools.yang.common.RpcResult;
  33 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36
  37 import com.google.common.util.concurrent.Futures;
  38
  39 /**
  40  * This class holds all RPCs methods for LispSouthbound Plugin.
  41  *
  42  * <p>
  43  * @author Florin Coras (fcoras@cisco.com)
  44  * @author Lorand Jakab (lojakab@cisco.com)
  45  */
  46
  47 public class LispSouthboundRPC implements OdlLispSbService {
  48
  49     protected static final Logger LOG = LoggerFactory.getLogger(LispSouthboundRPC.class);
  50
  51     private final LispSouthboundPlugin lispSbPlugin;
  52
  53     public LispSouthboundRPC(LispSouthboundPlugin lispSbPlugin) {
  54         this.lispSbPlugin = lispSbPlugin;
  55     }
  56
  57
  58     @Override
  59     public Future<RpcResult<Void>> sendMapNotify(SendMapNotifyInput mapNotifyInput) {
  60         LOG.trace("sendMapNotify called!!");
  61         if (mapNotifyInput != null) {
  62             ByteBuffer outBuffer = MapNotifySerializer.getInstance().serialize(mapNotifyInput.getMapNotify());
  63             lispSbPlugin.handleSerializedLispBuffer(mapNotifyInput.getTransportAddress(), outBuffer,
  64                     MessageType.MapNotify);
  65         } else {
  66             LOG.warn("MapNotify was null");
  67             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
  68         }
  69         return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
  70     }
  71
  72     @Override
  73     public Future<RpcResult<Void>> sendMapReply(SendMapReplyInput mapReplyInput) {
  74         LOG.trace("sendMapReply called!!");
  75         if (mapReplyInput != null) {
  76             ByteBuffer outBuffer = MapReplySerializer.getInstance().serialize(mapReplyInput.getMapReply());
  77             lispSbPlugin.handleSerializedLispBuffer(mapReplyInput.getTransportAddress(), outBuffer,
  78                     MessageType.MapReply);
  79         } else {
  80             LOG.warn("MapReply was null");
  81             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
  82         }
  83         return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
  84     }
  85
  86     @Override
  87     public Future<RpcResult<Void>> sendMapRequest(SendMapRequestInput mapRequestInput) {
  88         LOG.trace("sendMapRequest called!!");
  89         if (mapRequestInput != null) {
  90             ByteBuffer outBuffer = MapRequestSerializer.getInstance().serialize(mapRequestInput.getMapRequest());
  91             lispSbPlugin.handleSerializedLispBuffer(mapRequestInput.getTransportAddress(), outBuffer,
  92                     MessageType.MapRequest);
  93         } else {
  94             LOG.debug("MapRequest was null");
  95             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
  96         }
  97         return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
  98     }
  99
 100     @Override
 101     public Future<RpcResult<Void>> sendMapRegister(SendMapRegisterInput mapRegisterInput) {
 102         LOG.trace("sendMapRegister called!!");
 103         if (mapRegisterInput != null) {
 104             ByteBuffer outBuffer = MapRegisterSerializer.getInstance().serialize(mapRegisterInput.getMapRegister());
 105             lispSbPlugin.handleSerializedLispBuffer(mapRegisterInput.getTransportAddress(), outBuffer,
 106                     MessageType.MapRegister);
 107         } else {
 108             LOG.debug("MapRegister was null");
 109             return Futures.immediateFuture(RpcResultBuilder.<Void> failed().build());
 110         }
 111         return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
 112     }
 113
 114     @Override
 115     public Future<RpcResult<GetStatsOutput>> getStats() {
 116         LOG.trace("getStats called!!");
 117
 118         RpcResultBuilder<GetStatsOutput> rpcResultBuilder;
 119
 120         LispSouthboundStats stats = lispSbPlugin.getStats();
 121
 122         if (stats == null) {
 123             rpcResultBuilder = RpcResultBuilder.<GetStatsOutput>failed()
 124                     .withError(RpcError.ErrorType.APPLICATION, "data-missing", "No stats found");
 125         } else {
 126             rpcResultBuilder = RpcResultBuilder.success(createGetStatsOutput(stats));
 127         }
 128         return Futures.immediateFuture(rpcResultBuilder.build());
 129     }
 130
 131     @Override
 132     public Future<RpcResult<Void>> resetStats() {
 133         LOG.trace("resetStats called!!");
 134
 135         LispSouthboundStats stats = lispSbPlugin.getStats();
 136
 137         if (stats == null) {
 138             return Futures.immediateFuture(RpcResultBuilder.<Void> failed()
 139                     .withError(RpcError.ErrorType.APPLICATION, "data-missing", "No stats found")
 140                     .build());
 141         } else {
 142             stats.resetStats();
 143             return Futures.immediateFuture(RpcResultBuilder.<Void> success().build());
 144         }
 145     }
 146
 147     private static GetStatsOutput createGetStatsOutput(LispSouthboundStats stats) {
 148         long rxStats[] = stats.getRx();
 149         long txStats[] = stats.getTx();
 150
 151         ControlMessageStatsBuilder cmsb = new ControlMessageStatsBuilder();
 152         cmsb.setRxUnknown(stats.getRxUnknown());
 153         cmsb.setTxErrors(stats.getTxErrors());
 154
 155         List<ControlMessage> messages = new ArrayList<ControlMessage>();
 156         for (int i = 0; i <= LispSouthboundStats.MAX_LISP_TYPES; i++) {
 157             if (MessageType.forValue(i) == null) {
 158                 continue;
 159             }
 160             ControlMessageBuilder cmb = new ControlMessageBuilder();
 161             cmb.setMsgType(MessageType.forValue(i));
 162             cmb.setRxCount(rxStats[i]);
 163             cmb.setTxCount(txStats[i]);
 164             messages.add(cmb.build());
 165         }
 166
 167         cmsb.setControlMessage(messages);
 168         return new GetStatsOutputBuilder().setControlMessageStats(cmsb.build()).build();
 169     }
 170 }