Add support for fields starting with underscore

Peter Ableda · Peter Ableda · commit 061016fbbaa8 · 2014-06-11T14:51:14.000+02:00
diff --git a/pig/src/main/java/com/mongodb/hadoop/pig/MongoInsertStorage.java b/pig/src/main/java/com/mongodb/hadoop/pig/MongoInsertStorage.java
@@ -74,7 +74,11 @@ protected void writeField(final BasicDBObjectBuilder builder,
         if (field.getName() != null && field.getName().equals(this.idField)) {
             builder.add("_id", convertedType);
         } else {
-            builder.add(field.getName(), convertedType);
+            if (field.getName() != null && field.getName().startsWith("underscore_")) {
+                builder.add(field.getName().replace("underscore", ""), convertedType);
+            } else {
+            	builder.add(field.getName(), convertedType);
+            }
         }
 
     }

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,11 @@ protected void writeField(final BasicDBObjectBuilder builder,`
`74`	`74`	`if (field.getName() != null && field.getName().equals(this.idField)) {`
`75`	`75`	`builder.add("_id", convertedType);`
`76`	`76`	`} else {`
`77`		`- builder.add(field.getName(), convertedType);`
	`77`	`+ if (field.getName() != null && field.getName().startsWith("underscore_")) {`
	`78`	`+ builder.add(field.getName().replace("underscore", ""), convertedType);`
	`79`	`+ } else {`
	`80`	`+ builder.add(field.getName(), convertedType);`
	`81`	`+ }`
`78`	`82`	`}`
`79`	`83`
`80`	`84`	`}`