pcep/auto-bandwidth-extension/src/main/java/org/opendaylight/protocol/pcep/auto/bandwidth/extension/Activator.java

   1 /*
   2  * Copyright (c) 2016 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
   9 package org.opendaylight.protocol.pcep.auto.bandwidth.extension;
  10
  11 import com.google.common.annotations.VisibleForTesting;
  12 import java.util.ArrayList;
  13 import java.util.List;
  14 import org.opendaylight.protocol.pcep.spi.PCEPExtensionProviderContext;
  15 import org.opendaylight.protocol.pcep.spi.pojo.AbstractPCEPExtensionProviderActivator;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.auto.bandwidth.rev171025.bandwidth.usage.object.BandwidthUsage;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev171025.Pcrpt;
  18
  19 public class Activator extends AbstractPCEPExtensionProviderActivator {
  20
  21     private final int bandwidthUsageObjectType;
  22
  23     @VisibleForTesting
  24     public Activator() {
  25         this(1);
  26     }
  27
  28     public Activator(final int bandwidthUsageObjectType) {
  29         this.bandwidthUsageObjectType = bandwidthUsageObjectType;
  30     }
  31
  32     @Override
  33     protected List<AutoCloseable> startImpl(final PCEPExtensionProviderContext context) {
  34         final List<AutoCloseable> regs = new ArrayList<>();
  35
  36         final BandwidthUsageObjectCodec bandwidthUsageObjectCodec = new BandwidthUsageObjectCodec(this.bandwidthUsageObjectType);
  37         regs.add(context.registerObjectParser(BandwidthUsageObjectCodec.CLASS, bandwidthUsageObjectCodec.getType(), bandwidthUsageObjectCodec));
  38         regs.add(context.registerObjectSerializer(BandwidthUsage.class, bandwidthUsageObjectCodec));
  39
  40         final PcRptMessageCodec pcRptMessageCodec = new PcRptMessageCodec(context.getObjectHandlerRegistry());
  41         regs.add(context.registerMessageParser(PcRptMessageCodec.TYPE, pcRptMessageCodec));
  42         regs.add(context.registerMessageSerializer(Pcrpt.class, pcRptMessageCodec));
  43
  44         return regs;
  45     }
  46
  47 }