com.aliyun.odps.sqa.commandapi.CompactCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of odps-sdk-core Show documentation
The newest version!
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

package com.aliyun.odps.sqa.commandapi;

import java.util.HashMap;
import java.util.Map;

import com.aliyun.odps.Odps;
import com.aliyun.odps.OdpsException;
import com.aliyun.odps.data.Record;
import com.aliyun.odps.utils.StringUtils;

class CompactCommand extends MergeTaskCommand {

  private String compactType;

  public CompactCommand(String tableName, String partition, String compactType) {
    super(tableName, partition);
    this.compactType = compactType;
  }

  private boolean checkTransactional(String tablePart, Odps odps) {
    String projectName = odps.getDefaultProject();
    String tableName = tablePart.split("\\s+")[0];

    if (tableName.contains(".")) {
      tableName = tableName.split("\\.")[1];
    }

    return odps.tables().get(projectName, tableName).isTransactional();
  }

  @Override
  public boolean isSync() {
    return false;
  }

  @Override
  public RecordIter run(Odps odps, CommandInfo commandInfo) throws OdpsException {
    String tableInfo = getTableName() + " " + getPartition();
    if (!checkTransactional(tableInfo, odps)) {
      throw new OdpsException(tableInfo + " is not a transactional table.");
    }

    compactType = getCompactType(compactType.trim());
    if (compactType == null) {
      throw new RuntimeException("Compact type should be MAJOR or MINOR.");
    }

    String taskNamePrefix = "console_merge_task_";

    Map settingMap = new HashMap<>();
    settingMap.put("odps.merge.txn.table.compact", compactType);
    settingMap.put("odps.merge.restructure.action", "hardlink");

    return mergeTaskRun(odps, commandInfo, taskNamePrefix, settingMap);
  }

  private static String getCompactType(String type) {
    if (StringUtils.isNullOrEmpty(type)) {
      return null;
    }

    if (type.equalsIgnoreCase("major")) {
      return "major_compact";
    } else if (type.equalsIgnoreCase("minor")) {
      return "minor_compact";
    }

    return null;
  }
}