diff options
Diffstat (limited to 'framework/src/onos/cli/src/main/java/org/onosproject/cli/net/AddTestFlowsCommand.java')
| -rw-r--r-- | framework/src/onos/cli/src/main/java/org/onosproject/cli/net/AddTestFlowsCommand.java | 170 |
1 files changed, 170 insertions, 0 deletions
diff --git a/framework/src/onos/cli/src/main/java/org/onosproject/cli/net/AddTestFlowsCommand.java b/framework/src/onos/cli/src/main/java/org/onosproject/cli/net/AddTestFlowsCommand.java new file mode 100644 index 00000000..e1e158ab --- /dev/null +++ b/framework/src/onos/cli/src/main/java/org/onosproject/cli/net/AddTestFlowsCommand.java @@ -0,0 +1,170 @@ + +/* + * Copyright 2014-2015 Open Networking Laboratory + * + * Licensed 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.onosproject.cli.net; + +import com.fasterxml.jackson.databind.ObjectMapper; +import com.fasterxml.jackson.databind.node.ArrayNode; +import com.fasterxml.jackson.databind.node.ObjectNode; +import com.google.common.base.Stopwatch; +import com.google.common.collect.Lists; +import org.apache.commons.lang.math.RandomUtils; +import org.apache.karaf.shell.commands.Argument; +import org.apache.karaf.shell.commands.Command; +import org.onlab.packet.MacAddress; +import org.onosproject.cli.AbstractShellCommand; +import org.onosproject.core.ApplicationId; +import org.onosproject.core.CoreService; +import org.onosproject.net.Device; +import org.onosproject.net.PortNumber; +import org.onosproject.net.device.DeviceService; +import org.onosproject.net.flow.DefaultFlowRule; +import org.onosproject.net.flow.DefaultTrafficSelector; +import org.onosproject.net.flow.DefaultTrafficTreatment; +import org.onosproject.net.flow.FlowRule; +import org.onosproject.net.flow.FlowRuleOperations; +import org.onosproject.net.flow.FlowRuleOperationsContext; +import org.onosproject.net.flow.FlowRuleService; +import org.onosproject.net.flow.TrafficSelector; +import org.onosproject.net.flow.TrafficTreatment; + +import java.util.ArrayList; +import java.util.concurrent.CountDownLatch; +import java.util.concurrent.TimeUnit; + +/** + * Installs bulk flows. + */ +@Command(scope = "onos", name = "add-test-flows", + description = "Installs a number of test flow rules - for testing only") +public class AddTestFlowsCommand extends AbstractShellCommand { + + private CountDownLatch latch; + + @Argument(index = 0, name = "flowPerDevice", description = "Number of flows to add per device", + required = true, multiValued = false) + String flows = null; + + @Argument(index = 1, name = "numOfRuns", description = "Number of iterations", + required = true, multiValued = false) + String numOfRuns = null; + + @Override + protected void execute() { + FlowRuleService flowService = get(FlowRuleService.class); + DeviceService deviceService = get(DeviceService.class); + CoreService coreService = get(CoreService.class); + + ApplicationId appId = coreService.registerApplication("onos.test.flow.installer"); + + int flowsPerDevice = Integer.parseInt(flows); + int num = Integer.parseInt(numOfRuns); + + ArrayList<Long> results = Lists.newArrayList(); + Iterable<Device> devices = deviceService.getDevices(); + TrafficTreatment treatment = DefaultTrafficTreatment.builder() + .setOutput(PortNumber.portNumber(RandomUtils.nextInt())).build(); + TrafficSelector.Builder sbuilder; + FlowRuleOperations.Builder rules = FlowRuleOperations.builder(); + FlowRuleOperations.Builder remove = FlowRuleOperations.builder(); + + for (Device d : devices) { + for (int i = 0; i < flowsPerDevice; i++) { + sbuilder = DefaultTrafficSelector.builder(); + + sbuilder.matchEthSrc(MacAddress.valueOf(RandomUtils.nextInt() * i)) + .matchEthDst(MacAddress.valueOf((Integer.MAX_VALUE - i) * RandomUtils.nextInt())); + + + int randomPriority = RandomUtils.nextInt(); + + FlowRule addRule = DefaultFlowRule.builder() + .forDevice(d.id()) + .withSelector(sbuilder.build()) + .withTreatment(treatment) + .withPriority(randomPriority) + .fromApp(appId) + .makeTemporary(10) + .build(); + FlowRule removeRule = DefaultFlowRule.builder() + .forDevice(d.id()) + .withSelector(sbuilder.build()) + .withTreatment(treatment) + .withPriority(randomPriority) + .fromApp(appId) + .makeTemporary(10) + .build(); + + rules.add(addRule); + remove.remove(removeRule); + + } + } + + for (int i = 0; i < num; i++) { + latch = new CountDownLatch(2); + flowService.apply(rules.build(new FlowRuleOperationsContext() { + + private final Stopwatch timer = Stopwatch.createStarted(); + + @Override + public void onSuccess(FlowRuleOperations ops) { + + timer.stop(); + results.add(timer.elapsed(TimeUnit.MILLISECONDS)); + if (results.size() == num) { + if (outputJson()) { + print("%s", json(new ObjectMapper(), true, results)); + } else { + printTime(true, results); + } + } + latch.countDown(); + } + })); + + flowService.apply(remove.build(new FlowRuleOperationsContext() { + @Override + public void onSuccess(FlowRuleOperations ops) { + latch.countDown(); + } + })); + try { + latch.await(); + } catch (InterruptedException e) { + e.printStackTrace(); + } + + } + } + + private Object json(ObjectMapper mapper, boolean isSuccess, ArrayList<Long> elapsed) { + ObjectNode result = mapper.createObjectNode(); + result.put("Success", isSuccess); + ArrayNode node = result.putArray("elapsed-time"); + for (Long v : elapsed) { + node.add(v); + } + return result; + } + + private void printTime(boolean isSuccess, ArrayList<Long> elapsed) { + print("Run is %s.", isSuccess ? "success" : "failure"); + for (int i = 0; i < elapsed.size(); i++) { + print(" Run %s : %s", i, elapsed.get(i)); + } + } +} |