drivers/microsemi/ea1000yang/src/main/java/org/onosproject/yang/gen/v1/http/www/microsemi/com/microsemi/edge/assure/msea/uni/evc/service/rev20160317/mseaunievcservice/mefservices/uni/evc/evcperuni/CustomEvcPerUnin.java

/*
 * Copyright 2017-present 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.yang.gen.v1.http.www.microsemi.com.microsemi.edge.assure.msea.uni.evc.service
    .rev20160317.mseaunievcservice.mefservices.uni.evc.evcperuni;

import org.onosproject.drivers.microsemi.yang.utils.CeVlanMapUtils;
import org.onosproject.yang.gen.v1.http.www.microsemi.com.microsemi.edge.assure.msea.types
    .rev20160229.mseatypes.ServiceListType;

/**
 * A custom implementation of the DefaultEvcPerUnin - especially its Builder.
 *
 * This allows the EvcPerUniN to be modified after creation. These additions to the
 * builder are necessary because in the EA1000 YANG model the EvcPerUniN can
 * be added at separate stages
 */
public class CustomEvcPerUnin extends DefaultEvcPerUnin {
    public static EvcPerUninBuilder builder(EvcPerUnin evcPerUnin) {
        return new EvcPerUninBuilder(evcPerUnin);
    }

    public static EvcPerUninBuilder builder() {
        return new EvcPerUninBuilder();
    }

    public static class EvcPerUninBuilder extends DefaultEvcPerUnin.EvcPerUninBuilder {

        /**
         * Allow a new builder to be constructed
         */
        public EvcPerUninBuilder() {
            valueLeafFlags.set(LeafIdentifier.CEVLANMAP.getLeafIndex());
            this.ceVlanMap = new ServiceListType("0");
            valueLeafFlags.set(LeafIdentifier.INGRESSBWPGROUPINDEX.getLeafIndex());
            this.ingressBwpGroupIndex = 0;
        }

        /**
         * Allow a builder to be constructed from an existing EvcPerUnin
         * @param evcPerUnin An existing EvcPerUnin
         */
        public EvcPerUninBuilder(EvcPerUnin evcPerUnin) {
            this.ceVlanMap = evcPerUnin.ceVlanMap();
            this.evcPerUniServiceType = evcPerUnin.evcPerUniServiceType();
            this.ingressBwpGroupIndex = evcPerUnin.ingressBwpGroupIndex();
            this.tagManipulation = evcPerUnin.tagManipulation();
            this.flowMapping = evcPerUnin.flowMapping();
            this.yangEvcPerUninOpType = evcPerUnin.yangEvcPerUninOpType();
            this.yangAugmentedInfoMap = evcPerUnin.yangAugmentedInfoMap();
            this.selectLeafFlags = evcPerUnin.selectLeafFlags();
            this.valueLeafFlags = evcPerUnin.valueLeafFlags();
        }

        /**
         * Method to allow ceVlanMap to be modified.
         * @param additionalCeVlanMap An addition to the existing ceVlanMap
         * @return The updated builder
         */
        public EvcPerUninBuilder addToCeVlanMap(ServiceListType additionalCeVlanMap) {
            String combinedCeVlanMap =
                    CeVlanMapUtils.combineVlanSets(ceVlanMap.string(), additionalCeVlanMap.string());
            //If it contains 0 we should remove it
            ceVlanMap = new ServiceListType(
                    CeVlanMapUtils.removeZeroIfPossible(combinedCeVlanMap));
            return this;
        }
    }
}