/* * 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 2011-11-1) */ package com.alibaba.cobar.route.config; import java.util.Collections; import java.util.HashSet; import java.util.Set; import com.alibaba.cobar.route.config.TableRuleConfig.RuleConfig; import com.alibaba.cobar.util.SplitUtil; /** * @author <a href="mailto:shuo.qius@alibaba-inc.com">QIU Shuo</a> * @author xianmao.hexm */ public final class TableConfig { private final String name; private final String[] dataNodes; private final TableRuleConfig rule; private final Set<String> columnIndex; private final boolean ruleRequired; public TableConfig(String name, String dataNode, TableRuleConfig rule, boolean ruleRequired) { this.name = name; this.dataNodes = SplitUtil.split(dataNode, ',', '$', '-', '[', ']'); if (this.dataNodes == null || this.dataNodes.length <= 0) { throw new IllegalArgumentException("invalid table dataNodes: " + dataNode); } this.rule = rule; this.columnIndex = buildColumnIndex(rule); this.ruleRequired = ruleRequired; } public boolean existsColumn(String columnNameUp) { return columnIndex.contains(columnNameUp); } public String getName() { return name; } public String[] getDataNodes() { return dataNodes; } public boolean isRuleRequired() { return ruleRequired; } public TableRuleConfig getRule() { return rule; } private static Set<String> buildColumnIndex(TableRuleConfig rule) { if (rule == null) { return Collections.emptySet(); } RuleConfig[] rs = rule.getRules(); if (rs == null || rs.length <= 0) { return Collections.emptySet(); } Set<String> columnIndex = new HashSet<String>(); for (RuleConfig r : rs) { String[] columns = r.getColumns(); if (columns != null) for (String col : columns) { if (col != null) { columnIndex.add(col.toUpperCase()); } } } return columnIndex; } }