imay commented on a change in pull request #1695: Refactor alter job
URL: https://github.com/apache/incubator-doris/pull/1695#discussion_r323542395
##########
File path: fe/src/main/java/org/apache/doris/planner/StreamLoadScanNode.java
##########
@@ -120,72 +119,9 @@ public void init(Analyzer analyzer) throws UserException {
TBrokerScanRangeParams params = new TBrokerScanRangeParams();
params.setStrict_mode(streamLoadTask.isStrictMode());
- // parse columns header. this contain map from input column to column
of destination table
- // columns: k1, k2, v1, v2=k1 + k2
- // this means that there are three columns(k1, k2, v1) in source file,
- // and v2 is derived from (k1 + k2)
- if (streamLoadTask.getColumnExprDesc() != null &&
!streamLoadTask.getColumnExprDesc().isEmpty()) {
- for (ImportColumnDesc importColumnDesc :
streamLoadTask.getColumnExprDesc()) {
- // make column name case match with real column name
- String columnName = importColumnDesc.getColumnName();
- String realColName = dstTable.getColumn(columnName) == null ?
columnName
- : dstTable.getColumn(columnName).getName();
- if (importColumnDesc.getExpr() != null) {
- exprsByName.put(realColName, importColumnDesc.getExpr());
- } else {
- SlotDescriptor slotDesc =
analyzer.getDescTbl().addSlotDescriptor(srcTupleDesc);
-
slotDesc.setType(ScalarType.createType(PrimitiveType.VARCHAR));
- slotDesc.setIsMaterialized(true);
- // ISSUE A: src slot should be nullable even if the column
is not nullable.
- // because src slot is what we read from file, not
represent to real column value.
- // If column is not nullable, error will be thrown when
filling the dest slot,
- // which is not nullable
- slotDesc.setIsNullable(true);
- params.addToSrc_slot_ids(slotDesc.getId().asInt());
- slotDescByName.put(realColName, slotDesc);
- }
- }
-
- // analyze all exprs
- for (Map.Entry<String, Expr> entry : exprsByName.entrySet()) {
- ExprSubstitutionMap smap = new ExprSubstitutionMap();
- List<SlotRef> slots = Lists.newArrayList();
- entry.getValue().collect(SlotRef.class, slots);
- for (SlotRef slot : slots) {
- SlotDescriptor slotDesc =
slotDescByName.get(slot.getColumnName());
- if (slotDesc == null) {
- throw new UserException("unknown reference column,
column=" + entry.getKey()
- + ", reference=" + slot.getColumnName());
- }
- smap.getLhs().add(slot);
- smap.getRhs().add(new SlotRef(slotDesc));
- }
- Expr expr = entry.getValue().clone(smap);
- expr.analyze(analyzer);
-
- // check if contain aggregation
- List<FunctionCallExpr> funcs = Lists.newArrayList();
- expr.collect(FunctionCallExpr.class, funcs);
- for (FunctionCallExpr fn : funcs) {
- if (fn.isAggregateFunction()) {
- throw new AnalysisException("Don't support aggregation
function in load expression");
- }
- }
-
- exprsByName.put(entry.getKey(), expr);
- }
- } else {
- for (Column column : dstTable.getBaseSchema()) {
- SlotDescriptor slotDesc =
analyzer.getDescTbl().addSlotDescriptor(srcTupleDesc);
- slotDesc.setType(ScalarType.createType(PrimitiveType.VARCHAR));
- slotDesc.setIsMaterialized(true);
- // same as ISSUE A
- slotDesc.setIsNullable(true);
- params.addToSrc_slot_ids(slotDesc.getId().asInt());
-
- slotDescByName.put(column.getName(), slotDesc);
- }
- }
+ boolean specifyFileFieldNames =
streamLoadTask.getColumnExprDescs().stream().anyMatch(p -> p.isColumn());
Review comment:
why not put this in Load.initColumns?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
