[Issue mpollmeier#166] Make ScalaElement.updateWith(x) remove propert…

…ies which are None in x
SmedbergM · Aug 19, 2016 · c87b542 · c87b542
1 parent 30a8b63
commit c87b542
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gremlin-scala/src/main/scala/gremlin/scala/ScalaElement.scala b/gremlin-scala/src/main/scala/gremlin/scala/ScalaElement.scala
@@ -56,7 +56,11 @@ trait ScalaElement[ElementType <: Element] {
 
   def updateWith[CC <: Product: Marshallable](update: CC): ElementType = {
     val propMap = implicitly[Marshallable[CC]].fromCC(update).valueMap
-    propMap foreach {case (prop, value) => element.property(prop, value)}
+    this.valueMap.keySet.diff(propMap.keySet) foreach { key =>
+      val prop = element.property(key)
+      if (prop.isPresent) prop.remove()
+    }
+    propMap foreach {case (key, value) => element.property(key, value)}
 
     element
   }