Add Long and Boolean types to toPropertyType method

Author: michel-guillon

dataset/src/main/kotlin/com/cosmotech/dataset/bulk/PropertyType.kt

@@ -21,6 +21,8 @@ fun Any?.toPropertyType(): PropertyType {
   when (this) {
     is Int -> type = PropertyType.BI_LONG
     is Double -> type = PropertyType.BI_DOUBLE
+    is Long -> type = PropertyType.BI_LONG
+    is Boolean -> type = PropertyType.BI_BOOL
     is String -> {
       this.toIntOrNull()?.let {
         return PropertyType.BI_LONG

dataset/src/main/kotlin/com/cosmotech/dataset/service/DatasetServiceImpl.kt

@@ -279,9 +279,8 @@ class DatasetServiceImpl(
       record?.values()?.forEach { element ->
         when (element) {
           is Node -> {
-            val properties =
-                element.entityPropertyNames.associateWith { it -> element.getProperty(it).value }
-            queryBuffer.addNode(element.getLabel(0), element.id.toString(), properties)
+            val csmGraphEntity = element.toCsmGraphEntity(CsmGraphEntityType.NODE)
+            queryBuffer.addNode(csmGraphEntity.label, csmGraphEntity.id, csmGraphEntity.properties)
           }
           is Edge -> {
             val csmGraphEntity = element.toCsmGraphEntity(CsmGraphEntityType.RELATION)