コード例 #1
0
ファイル: JE_Table.java プロジェクト: HashZhang/tddl
  @Override
  public CloneableRecord get(
      ExecutionContext context, CloneableRecord key, IndexMeta indexMeta, String dbName) {

    DatabaseEntry keyEntry = new DatabaseEntry();
    DatabaseEntry valueEntry = new DatabaseEntry();
    keyEntry.setData(indexCodecMap.get(indexMeta.getName()).getKey_codec().encode(key));
    OperationStatus status =
        getDatabase(dbName)
            .get(
                context.getTransaction() == null
                    ? null
                    : ((JE_Transaction) context.getTransaction()).txn,
                keyEntry,
                valueEntry,
                LockMode.DEFAULT);
    if (OperationStatus.SUCCESS != status) {
      return null;
    }
    if (valueEntry.getSize() != 0) {
      return indexCodecMap.get(indexMeta.getName()).getValue_codec().decode(valueEntry.getData());
    } else {
      return null;
    }
  }
コード例 #2
0
  @Override
  public Object scalarCalucate(IRowSet kvPair, ExecutionContext ec) throws TddlRuntimeException {

    IQueryTree query = this.getQueryPlan();

    if (query.isCorrelatedSubquery()) {
      QueryTreeNode ast = OptimizerUtils.convertPlanToAst(query);
      // 获取correlated column
      List<ISelectable> columnsCorrelated = ast.getColumnsCorrelated();
      Map<Long, Object> correlatedValues = new HashMap();
      // 从rowset中找到对应的column进行替换
      for (ISelectable column : columnsCorrelated) {
        Object value = getValueByIColumnWithException(kvPair, column);
        correlatedValues.put(column.getCorrelateOnFilterId(), value);
      }
      // 替换correlated column
      OptimizerContext.getContext()
          .getOptimizer()
          .assignmentSubquery(ast, correlatedValues, ec.getExtraCmds());
      // 重新生成执行计划
      query =
          (IQueryTree)
              OptimizerContext.getContext()
                  .getOptimizer()
                  .optimizePlan(ast, ec.getParams(), ec.getExtraCmds());
    }

    return this.compute(new Object[] {query}, ec);
  }
コード例 #3
0
ファイル: JE_Table.java プロジェクト: HashZhang/tddl
 @Override
 public ISchematicCursor getCursor(
     ExecutionContext executionContext, IndexMeta meta, IQuery iQuery) throws TddlException {
   String actualTable = iQuery.getTableName();
   return getCursor(
       executionContext.getTransaction(), meta, executionContext.getIsolation(), actualTable);
 }
コード例 #4
0
  @Override
  public ITempTableSortCursor tempTableSortCursor(
      ExecutionContext executionContext,
      ISchematicCursor cursor,
      List<IOrderBy> orderBys,
      boolean sortedDuplicates,
      long requestID)
      throws TddlException {
    try {
      if (GeneralUtil.getExtraCmdBoolean(
          executionContext.getExtraCmds(), ConnectionProperties.ALLOW_TEMPORARY_TABLE, false)) {

        IRepository bdbRepo =
            ExecutorContext.getContext()
                .getRepositoryHolder()
                .getOrCreateRepository(
                    Group.GroupType.BDB_JE.name(),
                    Collections.EMPTY_MAP,
                    executionContext.getExtraCmds());
        return new TempTableSortCursor(
            this, bdbRepo, cursor, orderBys, sortedDuplicates, requestID, executionContext);
      }

      throw new IllegalStateException("not allow to use temporary table . allow first");

    } catch (Exception e) {
      closeParentCursor(cursor);
      throw new TddlException(e);
    }
  }
コード例 #5
0
ファイル: JE_Table.java プロジェクト: HashZhang/tddl
  @Override
  public ISchematicCursor getCursor(
      ExecutionContext executionContext, IndexMeta indexMeta, String actualTableName)
      throws TddlException {
    Database db = getDatabase(actualTableName);
    if (db == null) {
      throw new TddlException("table don't contains indexName:" + actualTableName);
    }
    ITransaction txn = executionContext.getTransaction();
    CursorConfig cc = CursorConfig.DEFAULT;
    LockMode lm = LockMode.DEFAULT;
    if (txn != null) {
      com.sleepycat.je.TransactionConfig _config = ((JE_Transaction) txn).config;
      if (_config.getReadUncommitted()) {
        cc = CursorConfig.READ_UNCOMMITTED;
        lm = LockMode.READ_UNCOMMITTED;
      } else if (_config.getReadCommitted()) {
        cc = CursorConfig.READ_COMMITTED;
        // lm = LockMode.READ_COMMITTED;
      }
    } else {

      cc = CursorConfig.READ_COMMITTED;
    }
    JE_Cursor je_cursor =
        new JE_Cursor(
            indexMeta, db.openCursor(txn == null ? null : ((JE_Transaction) txn).txn, cc), lm);
    if (txn != null) {
      ((JE_Transaction) txn).addCursor(je_cursor);
    }
    return new SchematicCursor(
        je_cursor, je_cursor.getiCursorMeta(), ExecUtils.getOrderBy(indexMeta));
  }
コード例 #6
0
  @Override
  public TempTableCursor tempTableCursor(
      ExecutionContext executionContext,
      ISchematicCursor cursor,
      List<IOrderBy> orderBys,
      boolean sortedDuplicates,
      long requestID)
      throws TddlException {
    try {

      IRepository bdbRepo =
          ExecutorContext.getContext()
              .getRepositoryHolder()
              .getOrCreateRepository(
                  Group.GroupType.BDB_JE.name(),
                  Collections.EMPTY_MAP,
                  executionContext.getExtraCmds());
      return new TempTableCursor(
          bdbRepo, cursor, orderBys, sortedDuplicates, requestID, executionContext);

    } catch (Exception e) {
      closeParentCursor(cursor);
      throw new TddlException(e);
    }
  }
コード例 #7
0
ファイル: JE_Table.java プロジェクト: HashZhang/tddl
  @Override
  public void delete(
      ExecutionContext context, CloneableRecord key, IndexMeta indexMeta, String dbName)
      throws TddlException {

    DatabaseEntry keyEntry = new DatabaseEntry();
    keyEntry.setData(indexCodecMap.get(indexMeta.getName()).getKey_codec().encode(key));
    try {
      getDatabase(dbName)
          .delete(
              context.getTransaction() == null
                  ? null
                  : ((JE_Transaction) context.getTransaction()).txn,
              keyEntry);
    } catch (LockTimeoutException ex) {

      throw ex;
    } catch (ReplicaWriteException ex) {
      throw new TddlException(ExceptionErrorCodeUtils.Read_only, ex);
    }
  }
コード例 #8
0
  @Override
  public TempTableCursor tempTableCursor(
      ExecutionContext executionContext,
      IQueryTree plan,
      List<IOrderBy> orderBys,
      boolean sortedDuplicates,
      long requestID)
      throws TddlException {

    IRepository bdbRepo =
        ExecutorContext.getContext()
            .getRepositoryHolder()
            .getOrCreateRepository(
                Group.GroupType.BDB_JE.name(),
                Collections.EMPTY_MAP,
                executionContext.getExtraCmds());
    return new TempTableCursor(bdbRepo, plan, sortedDuplicates, requestID, executionContext);
  }
コード例 #9
0
ファイル: JE_Table.java プロジェクト: HashZhang/tddl
  /**
   * todo:触发更新二级索引
   *
   * @param key
   * @param value
   * @throws TddlException
   */
  @Override
  public void put(
      ExecutionContext context,
      CloneableRecord key,
      CloneableRecord value,
      IndexMeta indexMeta,
      String dbName)
      throws TddlException {
    DatabaseEntry keyEntry = new DatabaseEntry();
    DatabaseEntry valueEntry = new DatabaseEntry();
    keyEntry.setData(indexCodecMap.get(indexMeta.getName()).getKey_codec().encode(key));

    // 当临时表排序时候可能会用到临时表join,主键插入值为null
    // if (keyEntry.getData().length == 1 && !isTempTable) {
    // throw new RuntimeException("pk must not null.");
    // }

    if (value != null) {
      valueEntry.setData(indexCodecMap.get(indexMeta.getName()).getValue_codec().encode(value));
    } else {
      valueEntry = emptyValueEntry;
    }
    try {
      ITransaction transaction = context.getTransaction();
      com.sleepycat.je.Transaction txn = null;
      if (transaction != null && transaction instanceof JE_Transaction) {
        txn = ((JE_Transaction) transaction).txn;
      }
      OperationStatus operationStatus = getDatabase(dbName).put(txn, keyEntry, valueEntry);
      if (operationStatus.equals(OperationStatus.SUCCESS)) {
        return;
      }
    } catch (LockTimeoutException ex) {

      throw ex;
    } catch (ReplicaWriteException ex) {
      throw new TddlException(ExceptionErrorCodeUtils.Read_only, ex);
    }
  }