Merge "Fix for Bug 3"
[controller.git] / opendaylight / md-sal / forwardingrules-manager / src / main / java / org / opendaylight / controller / frm / meter / MeterTransaction.xtend
1 package org.opendaylight.controller.frm.meter
2
3 import org.opendaylight.controller.frm.AbstractTransaction
4 import org.opendaylight.controller.md.sal.common.api.data.DataModification
5 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef
6 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node
7 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInputBuilder
8 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInputBuilder
9 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService
10 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInputBuilder
11 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.OriginalMeterBuilder
12 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.UpdatedMeterBuilder
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.Meter
14 import org.opendaylight.yangtools.yang.binding.DataObject
15 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterRef
17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Uri
18
19 class MeterTransaction extends AbstractTransaction {
20     
21     @Property
22     val SalMeterService salMeterService;
23         
24     new(DataModification<InstanceIdentifier<? extends DataObject>, DataObject> modification,SalMeterService salMeterService) {
25         super(modification)    
26         _salMeterService = salMeterService;
27     }
28     
29     override remove(InstanceIdentifier<?> instanceId, DataObject obj) {
30         if(obj instanceof Meter) {
31             val meter = (obj as Meter)
32             val nodeInstanceId = instanceId.firstIdentifierOf(Node);
33             val builder = new RemoveMeterInputBuilder(meter);
34             builder.setNode(new NodeRef(nodeInstanceId));
35             builder.setMeterRef(new MeterRef(instanceId));
36             builder.setTransactionUri(new Uri(modification.getIdentifier() as String));
37             _salMeterService.removeMeter(builder.build());            
38         }
39     }
40     
41     override update(InstanceIdentifier<?> instanceId, DataObject originalObj, DataObject updatedObj) {
42         if(originalObj instanceof Meter && updatedObj instanceof Meter) {
43             val originalMeter = (originalObj as Meter)
44             val updatedMeter = (updatedObj as Meter)
45             val nodeInstanceId = instanceId.firstIdentifierOf(Node);
46             val builder = new UpdateMeterInputBuilder();
47             builder.setNode(new NodeRef(nodeInstanceId));
48             builder.setMeterRef(new MeterRef(instanceId));
49             val ufb = new UpdatedMeterBuilder(updatedMeter);
50             builder.setUpdatedMeter((ufb.build()));
51             builder.setTransactionUri(new Uri(modification.getIdentifier() as String));
52             val ofb = new OriginalMeterBuilder(originalMeter);
53             builder.setOriginalMeter(ofb.build());      
54             _salMeterService.updateMeter(builder.build());
55            
56         }
57     }
58     
59     override add(InstanceIdentifier<?> instanceId, DataObject obj) {
60         if(obj instanceof Meter) {
61             val meter = (obj as Meter)
62             val nodeInstanceId = instanceId.firstIdentifierOf(Node);
63             val builder = new AddMeterInputBuilder(meter);
64             builder.setNode(new NodeRef(nodeInstanceId));
65             builder.setMeterRef(new MeterRef(instanceId));
66             builder.setTransactionUri(new Uri(modification.getIdentifier() as String));
67             _salMeterService.addMeter(builder.build());            
68         }
69     }
70     
71     override validate() throws IllegalStateException {
72         MeterTransactionValidator.validate(this)
73     }  
74 }