io.galeb.core.entity.BalancePolicy.java Source code

Introduction

Here is the source code for io.galeb.core.entity.BalancePolicy.java

Source

/*
 * Copyright (c) 2014-2017 Globo.com - ATeam
 * All rights reserved.
 *
 * This source is subject to the Apache License, Version 2.0.
 * Please see the LICENSE file for more information.
 *
 * Authors: See AUTHORS file
 *
 * 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 io.galeb.core.entity;

import com.fasterxml.jackson.annotation.JsonProperty;
import org.springframework.util.Assert;

import javax.persistence.Entity;
import javax.persistence.ForeignKey;
import javax.persistence.JoinColumn;
import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;
import javax.persistence.Table;
import javax.persistence.UniqueConstraint;
import java.util.HashSet;
import java.util.Set;

@Entity
@Table(name = "balancepolicy", uniqueConstraints = {
        @UniqueConstraint(name = "UK_name_balancepolicy", columnNames = { "name" }) })
public class BalancePolicy extends AbstractEntity<BalancePolicy> {

    private static final long serialVersionUID = 5596582746795373030L;

    @ManyToOne
    @JoinColumn(name = "balancepolicytype_id", nullable = false, foreignKey = @ForeignKey(name = "FK_balancepolicy_balancepolicytype"))
    @JsonProperty(required = true)
    private BalancePolicyType balancePolicyType;

    @OneToMany(mappedBy = "balancePolicy")
    private final Set<Pool> pools = new HashSet<>();

    public BalancePolicy(String name, BalancePolicyType balancePolicyType) {
        Assert.hasText(name,
                "[Assertion failed] - this String argument must have text; it must not be null, empty, or blank");
        Assert.notNull(balancePolicyType, "[Assertion failed] - this argument is required; it must not be null");
        setName(name);
        this.balancePolicyType = balancePolicyType;
    }

    protected BalancePolicy() {
        // Hibernate Requirement
    }

    public BalancePolicyType getBalancePolicyType() {
        return balancePolicyType;
    }

    public BalancePolicy setBalancePolicyType(BalancePolicyType balancePolicyType) {
        Assert.notNull(balancePolicyType, "[Assertion failed] - this argument is required; it must not be null");
        updateHash();
        this.balancePolicyType = balancePolicyType;
        return this;
    }

    public Set<Pool> getPools() {
        return pools;
    }

    public BalancePolicy setPools(Set<Pool> pools) {
        if (pools != null) {
            updateHash();
            this.pools.clear();
            this.pools.addAll(pools);
        }
        return this;
    }
}