Merge "Bug 809: Enhancements to the toaster example"
[controller.git] / opendaylight / md-sal / forwardingrules-manager / src / main / java / org / opendaylight / controller / frm / meter / MeterProvider.java
1 /**
2  * Copyright (c) 2014 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.controller.frm.meter;
9
10 import org.opendaylight.controller.sal.binding.api.data.DataChangeListener;
11 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
12 import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService;
18 import org.opendaylight.yangtools.concepts.ListenerRegistration;
19 import org.opendaylight.yangtools.yang.binding.DataObject;
20 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
21 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
22 import org.slf4j.Logger;
23 import org.slf4j.LoggerFactory;
24
25 public class MeterProvider implements AutoCloseable {
26
27     private final static Logger LOG = LoggerFactory.getLogger(MeterProvider.class);
28
29     private DataProviderService dataService;
30     private SalMeterService salMeterService;
31
32     /* DataChangeListener */
33     private MeterChangeListener meterDataChangeListener;
34     ListenerRegistration<DataChangeListener> meterDataChangeListenerRegistration;
35
36     public void start() {
37         /* Build Path */
38         InstanceIdentifierBuilder<Nodes> nodesBuilder = InstanceIdentifier.<Nodes> builder(Nodes.class);
39         InstanceIdentifierBuilder<Node> nodeChild = nodesBuilder.<Node> child(Node.class);
40         InstanceIdentifierBuilder<FlowCapableNode> augmentFlowCapNode = nodeChild.<FlowCapableNode> augmentation(FlowCapableNode.class);
41         InstanceIdentifierBuilder<Meter> meterChild = augmentFlowCapNode.<Meter> child(Meter.class);
42         final InstanceIdentifier<? extends DataObject> meterDataObjectPath = meterChild.toInstance();
43
44         /* DataChangeListener registration */
45         this.meterDataChangeListener = new MeterChangeListener(this.salMeterService);
46         this.meterDataChangeListenerRegistration = this.dataService.registerDataChangeListener(meterDataObjectPath, meterDataChangeListener);
47         LOG.info("Meter Config Provider started.");
48     }
49     
50     protected DataModificationTransaction startChange() {
51         return this.dataService.beginTransaction();
52     }
53
54     public void close() throws Exception {
55         if(meterDataChangeListenerRegistration != null){
56             meterDataChangeListenerRegistration.close();
57         }
58     }
59
60     public void setDataService(final DataProviderService dataService) {
61         this.dataService = dataService;
62     }
63
64     public void setSalMeterService(final SalMeterService salMeterService) {
65         this.salMeterService = salMeterService;
66     }
67 }