diff --git ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java index a3da075817..aadbf38fda 100644 --- ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java +++ ql/src/java/org/apache/hadoop/hive/ql/parse/SemanticAnalyzer.java @@ -8013,16 +8013,18 @@ private void handleLineage(LoadTableDesc ltd, Operator output) } private void setWriteIdForSurrogateKeys(LoadTableDesc ltd, Operator input) throws SemanticException { - Map columnExprMap = input.getConf().getColumnExprMap(); - if (ltd == null || columnExprMap == null) { + if (ltd == null) { return; } - for (ExprNodeDesc desc : columnExprMap.values()) { - if (desc instanceof ExprNodeGenericFuncDesc) { - GenericUDF genericUDF = ((ExprNodeGenericFuncDesc)desc).getGenericUDF(); - if (genericUDF instanceof GenericUDFSurrogateKey) { - ((GenericUDFSurrogateKey)genericUDF).setWriteId(ltd.getWriteId()); + Map columnExprMap = input.getConf().getColumnExprMap(); + if (columnExprMap != null) { + for (ExprNodeDesc desc : columnExprMap.values()) { + if (desc instanceof ExprNodeGenericFuncDesc) { + GenericUDF genericUDF = ((ExprNodeGenericFuncDesc)desc).getGenericUDF(); + if (genericUDF instanceof GenericUDFSurrogateKey) { + ((GenericUDFSurrogateKey)genericUDF).setWriteId(ltd.getWriteId()); + } } } }