added add-flows command
Change-Id: I240da77d78eebd54eeaa5a0864244920278429a2
diff --git a/cli/src/main/java/org/onlab/onos/cli/net/AddFlowsCommand.java b/cli/src/main/java/org/onlab/onos/cli/net/AddFlowsCommand.java
new file mode 100644
index 0000000..3c21343
--- /dev/null
+++ b/cli/src/main/java/org/onlab/onos/cli/net/AddFlowsCommand.java
@@ -0,0 +1,103 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.onlab.onos.cli.net;
+
+import com.google.common.collect.Sets;
+import org.apache.karaf.shell.commands.Argument;
+import org.apache.karaf.shell.commands.Command;
+import org.onlab.onos.cli.AbstractShellCommand;
+import org.onlab.onos.net.Device;
+import org.onlab.onos.net.PortNumber;
+import org.onlab.onos.net.device.DeviceService;
+import org.onlab.onos.net.flow.CompletedBatchOperation;
+import org.onlab.onos.net.flow.DefaultFlowRule;
+import org.onlab.onos.net.flow.DefaultTrafficSelector;
+import org.onlab.onos.net.flow.DefaultTrafficTreatment;
+import org.onlab.onos.net.flow.FlowRuleBatchEntry;
+import org.onlab.onos.net.flow.FlowRuleBatchOperation;
+import org.onlab.onos.net.flow.FlowRuleService;
+import org.onlab.onos.net.flow.TrafficSelector;
+import org.onlab.onos.net.flow.TrafficTreatment;
+import org.onlab.packet.MacAddress;
+
+import java.util.Set;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Future;
+
+/**
+ * Installs many many flows.
+ */
+@Command(scope = "onos", name = "add-flows",
+ description = "Installs a flow rules")
+public class AddFlowsCommand extends AbstractShellCommand {
+
+ @Argument(index = 0, name = "flowPerDevice", description = "Number of flows to add per device",
+ required = true, multiValued = false)
+ String flows = null;
+
+
+ @Override
+ protected void execute() {
+
+ FlowRuleService flowService = get(FlowRuleService.class);
+ DeviceService deviceService = get(DeviceService.class);
+
+ int flowsPerDevice = Integer.parseInt(flows);
+
+ Iterable<Device> devices = deviceService.getDevices();
+ TrafficTreatment treatment = DefaultTrafficTreatment.builder()
+ .setOutput(PortNumber.portNumber(1)).build();
+ TrafficSelector.Builder sbuilder;
+ Set<FlowRuleBatchEntry> rules = Sets.newHashSet();
+ Set<FlowRuleBatchEntry> remove = Sets.newHashSet();
+ for (Device d : devices) {
+ for (int i = 0; i < flowsPerDevice; i++) {
+ sbuilder = DefaultTrafficSelector.builder();
+ sbuilder.matchEthSrc(MacAddress.valueOf(i))
+ .matchEthDst(MacAddress.valueOf(Integer.MAX_VALUE - i));
+ rules.add(new FlowRuleBatchEntry(FlowRuleBatchEntry.FlowRuleOperation.ADD,
+ new DefaultFlowRule(d.id(), sbuilder.build(), treatment,
+ 100, (long) 0, 10, false)));
+ remove.add(new FlowRuleBatchEntry(FlowRuleBatchEntry.FlowRuleOperation.REMOVE,
+ new DefaultFlowRule(d.id(), sbuilder.build(), treatment,
+ 100, (long) 0, 10, false)));
+
+ }
+ }
+ long startTime = System.currentTimeMillis();
+ Future<CompletedBatchOperation> op = flowService.applyBatch(
+ new FlowRuleBatchOperation(rules));
+ CompletedBatchOperation completed = null;
+ try {
+ completed = op.get();
+ } catch (InterruptedException e) {
+ e.printStackTrace();
+ } catch (ExecutionException e) {
+ e.printStackTrace();
+ }
+ long endTime = System.currentTimeMillis();
+ print("%s AFTER ELAPSED TIME %s", completed.isSuccess() ? "SUCCESS" : "FAILURE",
+ endTime - startTime);
+
+ flowService.applyBatch(
+ new FlowRuleBatchOperation(remove));
+
+ }
+
+}
diff --git a/cli/src/main/resources/OSGI-INF/blueprint/shell-config.xml b/cli/src/main/resources/OSGI-INF/blueprint/shell-config.xml
index 418ee26..6fa500e 100644
--- a/cli/src/main/resources/OSGI-INF/blueprint/shell-config.xml
+++ b/cli/src/main/resources/OSGI-INF/blueprint/shell-config.xml
@@ -183,6 +183,10 @@
</command>
<command>
+ <action class="org.onlab.onos.cli.net.AddFlowsCommand"/>
+ </command>
+
+ <command>
<action class="org.onlab.onos.cli.net.WipeOutCommand"/>
</command>
</command-bundle>