corrected update statements

Author: ramitg254

ql/src/java/org/apache/hadoop/hive/ql/parse/RewriteSemanticAnalyzer.java

@@ -120,7 +120,7 @@ protected void checkValidSetClauseTarget(ASTNode colName, Table targetTable) thr
 
     // Make sure this isn't one of the partitioning columns, that's not supported.
     for (FieldSchema fschema : targetTable.getPartCols()) {
-      if (fschema.getName().equalsIgnoreCase(columnName)) {
+      if (fschema.getName().equalsIgnoreCase(columnName) && !targetTable.hasNonNativePartitionSupport()) {
         throw new SemanticException(ErrorMsg.UPDATE_CANNOT_UPDATE_PART_VALUE.getMsg());
       }
     }

ql/src/java/org/apache/hadoop/hive/ql/parse/rewrite/MergeRewriter.java

@@ -250,8 +250,10 @@ protected void addValues(Table targetTable, String targetAlias, Map<String, Stri
         }
       }
 
-      targetTable.getPartCols().forEach(fieldSchema -> values.add(
-          formatter.apply(fieldSchema.getName())));
+      if (!targetTable.hasNonNativePartitionSupport()) {
+        targetTable.getPartCols().forEach(fieldSchema -> values.add(
+            formatter.apply(fieldSchema.getName())));
+      }
     }
 
     protected String getRhsExpValue(String newValue, String alias) {

ql/src/java/org/apache/hadoop/hive/ql/parse/rewrite/SplitUpdateRewriter.java

@@ -98,7 +98,7 @@ public ParseUtils.ReparseResult rewrite(Context context, UpdateStatement updateB
 
       insertValues.add(sqlGenerator.qualify(identifier));
     }
-    if (updateBlock.getTargetTable().getPartCols() != null) {
+    if (!updateBlock.getTargetTable().hasNonNativePartitionSupport()) {
       updateBlock.getTargetTable().getPartCols().forEach(
           fieldSchema -> insertValues.add(sqlGenerator.qualify(HiveUtils.unparseIdentifier(fieldSchema.getName(), conf))));
     }

ql/src/java/org/apache/hadoop/hive/ql/parse/rewrite/sql/MultiInsertSqlGenerator.java

@@ -111,6 +111,9 @@ public void appendPartitionColsOfTarget() {
    */
   public void appendPartitionCols(Table table) {
     // If the table is partitioned we have to put the partition() clause in
+    if (table.hasNonNativePartitionSupport()) {
+      return;
+    }
     List<FieldSchema> partCols = table.getPartCols();
     if (partCols == null || partCols.isEmpty()) {
       return;
@@ -148,7 +151,10 @@ public void removeLastChar() {
   }
 
   public void appendPartColsOfTargetTableWithComma(String alias) {
-    if (targetTable.getPartCols() == null || targetTable.getPartCols().isEmpty()) {
+    if (targetTable.hasNonNativePartitionSupport()) {
+      return;
+    }
+    if (targetTable.getPartCols().isEmpty()) {
       return;
     }
     queryStr.append(',');