actually adding randomintentcommand.java

Change-Id: Ic156b8553be3417e0b7f4b27a85fcd503efe55b8

+/*
+ * Copyright 2014 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.onlab.onos.cli.net;
+
+import com.google.common.collect.Lists;
+import org.apache.karaf.shell.commands.Argument;
+import org.apache.karaf.shell.commands.Command;
+import org.onlab.onos.cli.AbstractShellCommand;
+import org.onlab.onos.core.ApplicationId;
+import org.onlab.onos.core.CoreService;
+import org.onlab.onos.net.Host;
+import org.onlab.onos.net.flow.DefaultTrafficSelector;
+import org.onlab.onos.net.flow.DefaultTrafficTreatment;
+import org.onlab.onos.net.flow.TrafficSelector;
+import org.onlab.onos.net.flow.TrafficTreatment;
+import org.onlab.onos.net.host.HostService;
+import org.onlab.onos.net.intent.HostToHostIntent;
+import org.onlab.onos.net.intent.Intent;
+import org.onlab.onos.net.intent.IntentOperations;
+import org.onlab.onos.net.intent.IntentService;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * Installs point-to-point connectivity intents.
+ */
+@Command(scope = "onos", name = "push-random-intents",
+         description = "Installs random intents to test throughput")
+public class RandomIntentCommand extends AbstractShellCommand {
+
+    @Argument(index = 0, name = "count",
+              description = "Number of intents to push",
+              required = true, multiValued = false)
+    String countString = null;
+
+    private IntentService service;
+    private HostService hostService;
+    private int count;
+
+    @Override
+    protected void execute() {
+        service = get(IntentService.class);
+        hostService = get(HostService.class);
+
+        count = Integer.parseInt(countString);
+
+        if (count > 0) {
+            Collection<Intent> intents = generateIntents();
+            submitIntents(intents);
+        } else {
+            withdrawIntents();
+        }
+    }
+
+    private Collection<Intent> generateIntents() {
+        TrafficSelector selector = DefaultTrafficSelector.builder().build();
+        TrafficTreatment treatment = DefaultTrafficTreatment.builder().build();
+
+        List<Host> hosts = Lists.newArrayList(hostService.getHosts());
+        List<Intent> fullMesh = Lists.newArrayList();
+        for (int i = 0; i < hosts.size(); i++) {
+            for (int j = i + 1; j < hosts.size(); j++) {
+                fullMesh.add(new HostToHostIntent(appId(),
+                                                  hosts.get(i).id(),
+                                                  hosts.get(j).id(),
+                                                  selector, treatment));
+            }
+        }
+        Collections.shuffle(fullMesh);
+        return fullMesh.subList(0, Math.min(count, fullMesh.size()));
+    }
+
+    private void submitIntents(Collection<Intent> intents) {
+        IntentOperations.Builder builder = IntentOperations.builder();
+        for (Intent intent : intents) {
+            builder.addSubmitOperation(intent);
+        }
+        service.execute(builder.build());
+        print("Submitted %d host to host intents.", intents.size());
+    }
+
+    private void withdrawIntents() {
+        IntentOperations.Builder builder = IntentOperations.builder();
+        for (Intent intent : service.getIntents()) {
+            if (appId().equals(intent.appId())) {
+                builder.addWithdrawOperation(intent.id());
+            }
+        }
+        service.execute(builder.build());
+        print("Withdrew all randomly generated host to host intents.");
+    }
+
+    @Override
+    protected ApplicationId appId() {
+        return get(CoreService.class).registerApplication("org.onlab.onos.cli-random");
+    }
+}