pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/message/PCEPNotificationMessageParser.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  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 package org.opendaylight.protocol.pcep.impl.message;
   9
  10 import com.google.common.collect.Lists;
  11
  12 import io.netty.buffer.ByteBuf;
  13 import io.netty.buffer.Unpooled;
  14
  15 import java.util.List;
  16
  17 import org.opendaylight.protocol.pcep.spi.AbstractMessageParser;
  18 import org.opendaylight.protocol.pcep.spi.MessageUtil;
  19 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
  20 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
  21 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcntfBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcntfMessage;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.CNotification;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.PcntfMessageBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.notifications.Notifications;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.notifications.NotificationsBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.notifications.Rps;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.notifications.RpsBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
  33
  34 /**
  35  * Parser for {@link PcntfMessage}
  36  */
  37 public class PCEPNotificationMessageParser extends AbstractMessageParser {
  38
  39     public static final int TYPE = 5;
  40
  41     public PCEPNotificationMessageParser(final ObjectRegistry registry) {
  42         super(registry);
  43     }
  44
  45     @Override
  46     public void serializeMessage(final Message message, final ByteBuf out) {
  47         if (!(message instanceof PcntfMessage)) {
  48             throw new IllegalArgumentException("Wrong instance of Message. Passed instance of " + message.getClass()
  49                     + ". Needed PcntfMessage.");
  50         }
  51         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.PcntfMessage msg = ((PcntfMessage) message).getPcntfMessage();
  52
  53         ByteBuf buffer = Unpooled.buffer();
  54         for (final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.Notifications n : msg.getNotifications()) {
  55             if (n.getRps() != null && !n.getRps().isEmpty()) {
  56                 for (final Rps rps : n.getRps()) {
  57                     buffer.writeBytes(serializeObject(rps.getRp()));
  58                 }
  59             }
  60             if (n.getNotifications() == null || n.getNotifications().isEmpty()) {
  61                 throw new IllegalArgumentException("Message must contain at least one notification object");
  62             } else {
  63                 for (final Notifications not : n.getNotifications()) {
  64                     buffer.writeBytes(serializeObject(not.getCNotification()));
  65                 }
  66             }
  67         }
  68         MessageUtil.formatMessage(TYPE, buffer, out);
  69     }
  70
  71     @Override
  72     protected Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
  73         if (objects == null) {
  74             throw new IllegalArgumentException("Passed list can't be null.");
  75         }
  76         if (objects.isEmpty()) {
  77             throw new PCEPDeserializerException("Notification message cannot be empty.");
  78         }
  79
  80         final List<org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.Notifications> compositeNotifications = Lists.newArrayList();
  81
  82         while (!objects.isEmpty()) {
  83             org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.Notifications comObj;
  84             comObj = getValidNotificationComposite(objects, errors);
  85
  86             if (comObj == null) {
  87                 break;
  88             }
  89             compositeNotifications.add(comObj);
  90         }
  91         if (compositeNotifications.isEmpty()) {
  92             throw new PCEPDeserializerException("Atleast one Notifications is mandatory.");
  93         }
  94         if (!objects.isEmpty()) {
  95             throw new PCEPDeserializerException("Unprocessed Objects: " + objects);
  96         }
  97         return new PcntfBuilder().setPcntfMessage(new PcntfMessageBuilder().setNotifications(compositeNotifications).build()).build();
  98     }
  99
 100     private static org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.Notifications getValidNotificationComposite(
 101             final List<Object> objects, final List<Message> errors) {
 102         final List<Rps> requestParameters = Lists.newArrayList();
 103         final List<org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.notifications.Notifications> notifications = Lists.newArrayList();
 104         Object obj;
 105
 106         State state = State.Init;
 107         while (!objects.isEmpty() && !state.equals(State.End)) {
 108             obj = objects.get(0);
 109             switch (state) {
 110             case Init:
 111                 state = State.RpIn;
 112                 if (obj instanceof Rp) {
 113                     final Rp rp = (Rp) obj;
 114                     if (rp.isProcessingRule()) {
 115                         errors.add(createErrorMsg(PCEPErrors.P_FLAG_NOT_SET));
 116                         return null;
 117                     }
 118                     requestParameters.add(new RpsBuilder().setRp(rp).build());
 119                     state = State.Init;
 120                     break;
 121                 }
 122             case RpIn:
 123                 state = State.NotificationIn;
 124                 if (obj instanceof CNotification) {
 125                     final CNotification n = (CNotification) obj;
 126                     notifications.add(new NotificationsBuilder().setCNotification(n).build());
 127                     state = State.RpIn;
 128                     break;
 129                 }
 130             case NotificationIn:
 131                 state = State.End;
 132                 break;
 133             case End:
 134                 break;
 135             }
 136             if (!state.equals(State.End)) {
 137                 objects.remove(0);
 138             }
 139         }
 140
 141         if (notifications.isEmpty()) {
 142             return null;
 143         }
 144
 145         return new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.NotificationsBuilder().setNotifications(
 146                 notifications).setRps(requestParameters).build();
 147     }
 148
 149     private enum State {
 150         Init, RpIn, NotificationIn, End
 151     }
 152 }