/*
 * Copyright 1999-2012 Alibaba Group.
 *  
 * 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.
 */
/**
 * (created at 2012-6-13)
 */
package com.alibaba.cobar.config.model;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * @author <a href="mailto:shuo.qius@alibaba-inc.com">QIU Shuo</a>
 */
public class SchemaConfig {

    private final String name;
    private final String dataNode;
    private final String group;
    private final Map<String, TableConfig> tables;
    private final boolean noSharding;
    private final String[] metaDataNodes;
    private final boolean keepSqlSchema;
    private final Set<String> allDataNodes;

    public SchemaConfig(String name, String dataNode, String group, boolean keepSqlSchema,
                        Map<String, TableConfig> tables) {
        this.name = name;
        this.dataNode = dataNode;
        this.group = group;
        this.tables = tables;
        this.noSharding = (tables == null || tables.isEmpty()) ? true : false;
        this.metaDataNodes = buildMetaDataNodes();
        this.allDataNodes = buildAllDataNodes();
        this.keepSqlSchema = keepSqlSchema;
    }

    public boolean isKeepSqlSchema() {
        return keepSqlSchema;
    }

    public String getName() {
        return name;
    }

    public String getDataNode() {
        return dataNode;
    }

    public String getGroup() {
        return group;
    }

    public Map<String, TableConfig> getTables() {
        return tables;
    }

    public boolean isNoSharding() {
        return noSharding;
    }

    public String[] getMetaDataNodes() {
        return metaDataNodes;
    }

    public Set<String> getAllDataNodes() {
        return allDataNodes;
    }

    public String getRandomDataNode() {
        if (allDataNodes == null || allDataNodes.isEmpty()) {
            return null;
        }
        return allDataNodes.iterator().next();
    }

    /**
     * 取得含有不同Meta信息的数据节点,比如表和表结构。
     */
    private String[] buildMetaDataNodes() {
        Set<String> set = new HashSet<String>();
        if (!isEmpty(dataNode)) {
            set.add(dataNode);
        }
        if (!noSharding) {
            for (TableConfig tc : tables.values()) {
                set.add(tc.getDataNodes()[0]);
            }
        }
        return set.toArray(new String[set.size()]);
    }

    /**
     * 取得该schema的所有数据节点
     */
    private Set<String> buildAllDataNodes() {
        Set<String> set = new HashSet<String>();
        if (!isEmpty(dataNode)) {
            set.add(dataNode);
        }
        if (!noSharding) {
            for (TableConfig tc : tables.values()) {
                set.addAll(Arrays.asList(tc.getDataNodes()));
            }
        }
        return set;
    }

    private static boolean isEmpty(String str) {
        return ((str == null) || (str.length() == 0));
    }

}