Exemplo n.º 1
0
  /**
   * 根据输入的path地址获取含有所有dbrule和tableRule的Map. 只在初始化的时候调用一次.
   *
   * @param path 路径名系统内采用 getClass().getResourceAsStream(path);的方式获取对应的stream资源
   * @return
   */
  @SuppressWarnings("unchecked")
  public synchronized void initLogicTabMarticMap(String path, Map<String, LogicTabMatrix> map) {
    Document doc = getDocument(path);
    Element ele = doc.getRootElement();
    String dbType = trim(ele.attributeValue("dbType"));
    Iterator dbTabItr = ele.elementIterator("table");
    int i = 1;
    while (dbTabItr.hasNext()) {
      Element aVtab = (Element) dbTabItr.next();
      LogicTabMatrix aVtabMatrix = new LogicTabMatrix();
      // globalTableRule,如果有则储存起来
      Element globeRule = aVtab.element("globalTableRule");
      if (globeRule != null) {
        TabRule globTabRule = this.getTabRule(globeRule);
        aVtabMatrix.setGlobalTableRule(globTabRule);
      }
      String logicName =
          validAndTrim(aVtab.attributeValue("logicName"), "无法找到第" + i + "个虚拟表的logicName");
      // martrix里面的TableName是返还给前端用于表名替换的,因此要保持原样
      aVtabMatrix.setTableName(logicName);
      String needRowCopy = trim(aVtab.attributeValue("rowCopy"));
      if (needRowCopy != null && !needRowCopy.equals("")) {
        aVtabMatrix.setNeedRowCopy(Boolean.valueOf(needRowCopy));
      }
      String reverseOutput = trim(aVtab.attributeValue("reverseOutput"));
      if (reverseOutput != null && !reverseOutput.equals("")) {
        aVtabMatrix.setAllowReverseOutput(Boolean.valueOf(reverseOutput));
      }
      aVtabMatrix.setDBType(dbType);
      String tabFactor = trim(aVtab.elementText("tableFactor"));
      aVtabMatrix.setTableFactor(tabFactor);
      Element dbRules = (Element) aVtab.element("dbRules");
      Iterator rulesItr = dbRules.elementIterator("dbRule");
      Map<String, DBRule> ruleMap = getRuleList(rulesItr, aVtabMatrix);
      // 当一个Rule对象有expression的时候放在depositedRule和allRule
      // 如果没有expressionString串,则只放在allRule待选
      Map<String, DBRule> depositedRules = getDepositedRule(ruleMap);
      aVtabMatrix.setAllRules(ruleMap);
      aVtabMatrix.setDepositedRules(depositedRules);
      List<DBRule> defaultRuleList = getDefaultRuleList(aVtab, aVtabMatrix);
      aVtabMatrix.setDefaultRules(defaultRuleList);

      map.put(logicName.toLowerCase(), aVtabMatrix);
      i++;
    }
  }
Exemplo n.º 2
0
  /**
   * 解析每一个rule
   *
   * @param rulesItr
   * @return
   */
  @SuppressWarnings("unchecked")
  private Map<String, DBRule> getRuleList(Iterator rulesItr, LogicTabMatrix aVtabMatrix) {
    Map<String, DBRule> rules = new HashMap<String, DBRule>();

    while (rulesItr.hasNext()) {

      Element ruleEle = (Element) rulesItr.next();
      String id = validAndTrim(ruleEle.attributeValue("id"), "必须指定Rule的id");
      DBRule rule = new DBRule();
      String exp = trim(ruleEle.elementText("expression"));
      rule.setExpression(exp);
      String primaryKey = trim(ruleEle.elementText("primaryKey"));
      rule.setPrimaryKey(primaryKey);
      String primaryKeyExp = trim(ruleEle.elementText("primaryKeyExp"));
      rule.setPrimaryKeyExp(primaryKeyExp);
      String parameters = trim(ruleEle.elementText("parameters"));
      rule.setParameters(parameters);
      String readPoolStr = ruleEle.elementText("readPools");
      String writePoolStr = ruleEle.elementText("writePools");
      if (readPoolStr == null
          || writePoolStr == null
          || readPoolStr.trim().equals("")
          || writePoolStr.trim().equals("")) {
        throw new TDLRunTimeException(
            "readPool和writePool必须同时指名" + "readPool可以和writePool同名,但writePool不能为多个,也不能为虚拟池");
      }
      String[] readPools = readPoolStr.trim().split(",");
      String[] writePools = writePoolStr.trim().split(",");

      rule.setReadPool(readPools);
      rule.setWritePool(writePools);
      Element subTableRuleElement = ruleEle.element("tableRule");
      if (subTableRuleElement != null) {
        TabRule tabRule = getTabRule(subTableRuleElement);

        tabRule.setPrimaryKey(primaryKey);
        rule.setDBSubTabRule(tabRule);
        log.debug("id:" + id + "的DBRule有subTableRule,因此使用subTableRule");
      }
      // add by shenxun 现在的逻辑是直接替换subRuleEle,以后不会再
      // 使用globalRule这个项目了,它只在载入的时候临时起作用
      else {
        TabRule tabRule = aVtabMatrix.getGlobalTableRule();
        if (tabRule != null) {
          tabRule.setPrimaryKey(primaryKey);
        }
        rule.setDBSubTabRule(tabRule);
        log.debug("id:" + id + "的DBRule没有subTableRule,因此使用globalTableRule");
      }
      rules.put(id, rule);
    }
    return rules;
  }
Exemplo n.º 3
0
 /**
  * 根据DefaultPool字段(允许用“,”分隔),获取DBRule中含有指定写库的DBRule
  *
  * @param aVtab
  * @param aVtabMatrix
  * @param tempSet
  * @return
  */
 private List<DBRule> getDefaultRuleList(Element aVtab, LogicTabMatrix aVtabMatrix) {
   List<DBRule> defaultList = new ArrayList<DBRule>();
   String defaultPools = trim(aVtab.attributeValue("defaultWritePool"));
   if (defaultPools != null) {
     String[] defaultPoolsStrArray = defaultPools.split(",");
     Map<String, DBRule> map = aVtabMatrix.getAllRules();
     for (String str : defaultPoolsStrArray) {
       DBRule dbrule = map.get(str.trim());
       if (dbrule != null) {
         defaultList.add(dbrule);
       } else {
         throw new IllegalArgumentException(
             "defaultRule中id为:" + str + " 的字段不能找到" + "一个对应的规则,请确认该id对应一个dbRule的id参数");
       }
     }
   }
   return defaultList;
 }