Skip to content

Commit c3b7b19

Browse files
committed
Change node internalId in node customerId for nodes if specified
1 parent dcb4ced commit c3b7b19

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

twingraph/src/main/kotlin/com/cosmotech/twingraph/utils/TwingraphUtils.kt

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,26 @@ import redis.clients.jedis.graph.entities.Property
88

99
object TwingraphUtils {
1010

11+
private const val NODE_ID_PROPERTY_NAME = "id"
12+
1113
@JvmStatic
1214
fun getNodeJson(node: Node): String {
15+
var nodeId = node.id.toString()
16+
17+
if (node.entityPropertyNames.contains(NODE_ID_PROPERTY_NAME)) {
18+
nodeId = node.getProperty(NODE_ID_PROPERTY_NAME).value.toString()
19+
}
20+
1321
val graphNode =
14-
GraphNode(node.getLabel(0), node.id, node.entityPropertyNames.map { node.getProperty(it) })
22+
GraphNode(
23+
node.getLabel(0),
24+
nodeId,
25+
node.entityPropertyNames.filter { it != NODE_ID_PROPERTY_NAME }.map {
26+
node.getProperty(it)
27+
})
1528
return objectMapper().writeValueAsString(graphNode)
1629
}
30+
1731
@JvmStatic
1832
fun isReadOnlyQuery(query: String): Boolean {
1933
val queryNormalized = query.trim().lowercase()
@@ -23,4 +37,4 @@ object TwingraphUtils {
2337
}
2438
}
2539

26-
data class GraphNode(var label: String, var id: Long, var properties: List<Property<Any>>)
40+
data class GraphNode(var label: String, var id: String, var properties: List<Property<Any>>)

0 commit comments

Comments
 (0)