apps/vtn/vtnrsc/src/main/java/org/onosproject/vtnrsc/cli/virtualport/VirtualPortExGwUpdateCommand.java

/*
 * Copyright 2015-present Open Networking Foundation
 *
 * 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.vtnrsc.cli.virtualport;

import java.util.Map;
import java.util.Set;

import org.apache.karaf.shell.commands.Command;
import org.apache.karaf.shell.commands.Option;
import org.onlab.packet.IpAddress;
import org.onlab.packet.MacAddress;
import org.onosproject.cli.AbstractShellCommand;
import org.onosproject.net.DeviceId;
import org.onosproject.vtnrsc.BindingHostId;
import org.onosproject.vtnrsc.DefaultVirtualPort;
import org.onosproject.vtnrsc.FixedIp;
import org.onosproject.vtnrsc.Subnet;
import org.onosproject.vtnrsc.TenantNetwork;
import org.onosproject.vtnrsc.VirtualPort;
import org.onosproject.vtnrsc.VirtualPortId;
import org.onosproject.vtnrsc.subnet.SubnetService;
import org.onosproject.vtnrsc.tenantnetwork.TenantNetworkService;
import org.onosproject.vtnrsc.virtualport.VirtualPortService;

import com.google.common.collect.Maps;
import com.google.common.collect.Sets;

/**
 * Supports for updating the external gateway virtualPort.
 */
@Command(scope = "onos", name = "externalgateway-update",
        description = "Supports for updating the external gateway virtualPort.")
public class VirtualPortExGwUpdateCommand extends AbstractShellCommand {

    @Option(name = "-m", aliases = "--macAddress", description = "MAC address.", required = true,
            multiValued = false)
    String macAddress = "";

    @Override
    protected void execute() {
        VirtualPortService service = get(VirtualPortService.class);
        SubnetService subnetService = get(SubnetService.class);
        TenantNetworkService tenantNetworkService = get(TenantNetworkService.class);
        Iterable<TenantNetwork> networks = tenantNetworkService.getNetworks();
        if (networks != null) {
            for (TenantNetwork network : networks) {
                if (network.routerExternal()) {
                    Iterable<Subnet> subnets = subnetService.getSubnets();
                    if (subnets != null) {
                        for (Subnet subnet : subnets) {
                            if (network.id().networkId().equals(subnet.networkId().networkId())) {
                                IpAddress exgwip = subnet.gatewayIp();
                                FixedIp fixedGwIp = FixedIp.fixedIp(subnet.id(), exgwip);
                                VirtualPort exgwPort = service.getPort(fixedGwIp);
                                if (exgwPort == null) {
                                    createExGwPort(network, subnet, fixedGwIp);
                                } else {
                                    updateExGwPort(exgwPort);
                                }
                            }
                        }
                    }
                }
            }
        }
    }

    private void createExGwPort(TenantNetwork network, Subnet subnet, FixedIp fixedGwIp) {
        VirtualPortService service = get(VirtualPortService.class);
        Map<String, String> strMap = Maps.newHashMap();
        VirtualPort virtualPort = new DefaultVirtualPort(VirtualPortId.portId("externalgateway-update-id"),
                                                         network.id(),
                                                         false, strMap,
                                                         VirtualPort.State.DOWN,
                                                         MacAddress.valueOf(macAddress),
                                                         subnet.tenantId(),
                                                         DeviceId.deviceId(""),
                                                         Sets.newHashSet(fixedGwIp),
                                                         BindingHostId.bindingHostId(""),
                                                         Sets.newHashSet(),
                                                         Sets.newHashSet());
        Set<VirtualPort> virtualPorts = Sets.newHashSet(virtualPort);
        service.createPorts(virtualPorts);
    }

    private void updateExGwPort(VirtualPort exgwPort) {
        VirtualPortService service = get(VirtualPortService.class);
        Map<String, String> strMap = Maps.newHashMap();
        strMap.putIfAbsent("name", exgwPort.name());
        strMap.putIfAbsent("deviceOwner", exgwPort.deviceOwner());
        strMap.putIfAbsent("bindingvnicType", exgwPort.bindingVnicType());
        strMap.putIfAbsent("bindingvifType", exgwPort.bindingVifType());
        strMap.putIfAbsent("bindingvnicDetails", exgwPort.bindingVifDetails());
        VirtualPort virtualPort = new DefaultVirtualPort(exgwPort.portId(),
                                                         exgwPort.networkId(),
                                                         false, strMap,
                                                         VirtualPort.State.DOWN,
                                                         MacAddress.valueOf(macAddress),
                                                         exgwPort.tenantId(),
                                                         exgwPort.deviceId(),
                                                         exgwPort.fixedIps(),
                                                         exgwPort.bindingHostId(),
                                                         Sets.newHashSet(exgwPort
                                                                .allowedAddressPairs()),
                                                         Sets.newHashSet(exgwPort
                                                                .securityGroups()));
        Set<VirtualPort> virtualPorts = Sets.newHashSet(virtualPort);
        service.updatePorts(virtualPorts);
    }
}