Skip to content

Commit

Permalink
refac: rename method append
Browse files Browse the repository at this point in the history
  • Loading branch information
kasakrisz committed Jul 12, 2022
1 parent 7d7fd1a commit 0c291c4
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ WHEN NOT MATCHED THEN INSERT VALUES (source.a2, source.b2)
String subQueryAlias = isAliased(targetNameNode) ? targetName : targetTable.getTTable().getTableName();
ColumnAppender columnAppender = nonNativeAcid ? new NonNativeAcidColumnAppender(targetTable, conf, subQueryAlias) :
new NativeAcidColumnAppender(targetTable, conf, subQueryAlias);
columnAppender.append(rewrittenQueryStr, Context.Operation.UPDATE);
columnAppender.appendAcidSelectColumns(rewrittenQueryStr, Context.Operation.UPDATE);

rewrittenQueryStr.deleteCharAt(rewrittenQueryStr.length() - 1); // remove last ','
addColsToSelect(targetTable.getCols(), rewrittenQueryStr);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -617,7 +617,7 @@ protected String getMatchedText(ASTNode n) {
public static final String SUB_QUERY_ALIAS = "s";

protected interface ColumnAppender {
void append(StringBuilder stringBuilder, Context.Operation operation);
void appendAcidSelectColumns(StringBuilder stringBuilder, Context.Operation operation);
List<String> getDeleteValues(Context.Operation operation);
List<String> getSortKeys();
}
Expand All @@ -635,7 +635,7 @@ public NativeAcidColumnAppender(Table table, HiveConf conf, String subQueryAlias
}

@Override
public void append(StringBuilder stringBuilder, Context.Operation operation) {
public void appendAcidSelectColumns(StringBuilder stringBuilder, Context.Operation operation) {
stringBuilder.append("ROW__ID,");
for (FieldSchema fieldSchema : table.getPartCols()) {
String identifier = HiveUtils.unparseIdentifier(fieldSchema.getName(), this.conf);
Expand Down Expand Up @@ -673,7 +673,7 @@ public NonNativeAcidColumnAppender(Table table, HiveConf conf, String subQueryAl
}

@Override
public void append(StringBuilder stringBuilder, Context.Operation operation) {
public void appendAcidSelectColumns(StringBuilder stringBuilder, Context.Operation operation) {
List<FieldSchema> acidSelectColumns = table.getStorageHandler().acidSelectColumns(table, operation);
for (FieldSchema fieldSchema : acidSelectColumns) {
String identifier = HiveUtils.unparseIdentifier(fieldSchema.getName(), this.conf);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,7 @@ private void analyzeSplitUpdate(ASTNode tree, Table mTable, ASTNode tabNameNode)
boolean nonNativeAcid = AcidUtils.isNonNativeAcidTable(mTable);
ColumnAppender columnAppender = nonNativeAcid ? new NonNativeAcidColumnAppender(mTable, conf, SUB_QUERY_ALIAS) :
new NativeAcidColumnAppender(mTable, conf, SUB_QUERY_ALIAS);
columnAppender.append(rewrittenQueryStr, operation);
columnAppender.appendAcidSelectColumns(rewrittenQueryStr, operation);
List<String> deleteValues = columnAppender.getDeleteValues(operation);
int columnOffset = deleteValues.size();

Expand Down

0 comments on commit 0c291c4

Please sign in to comment.