[!] add edge_type for identityConnection and do some cutoff about e…

…dges
NextDotID · Feb 1, 2024 · 5ad3665 · 5ad3665
1 parent 818a8ca
commit 5ad3665
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 5 deletions.
diff --git a/src/config/tdb/migrations/LoadingJob_SocialGraph.gsql b/src/config/tdb/migrations/LoadingJob_SocialGraph.gsql
@@ -331,7 +331,7 @@ CREATE OR REPLACE QUERY find_identity_graph_by_vertex(VERTEX<Identities> p) FOR
 
 
 CREATE OR REPLACE QUERY query_identity_graph_by_ids(SET<STRING> ids) FOR GRAPH SocialGraph {
-  TYPEDEF TUPLE< VERTEX source_v, VERTEX target_v, STRING data_source > IdentityConnection;
+  TYPEDEF TUPLE< VERTEX source_v, VERTEX target_v, STRING data_source, STRING edge_type > IdentityConnection;
   SetAccum<IdentityConnection> @@edges;
   SetAccum<VERTEX> @@hyper_vset;
   @@hyper_vset = to_vertex_set(ids, "IdentitiesGraph");
@@ -341,10 +341,26 @@ CREATE OR REPLACE QUERY query_identity_graph_by_ids(SET<STRING> ids) FOR GRAPH S
     @@edges.clear();
     hyper_v = {hv};
     vset = SELECT v FROM Identities:v-((PartOfIdentitiesGraph>):e)-hyper_v;
-    tmp1 = SELECT v1 FROM vset:v1-((Proof_Forward>|Proof_Backward>|Hold_Identity>|Resolve>|Reverse_Resolve>):e1)-vset:v2
-          ACCUM @@edges += IdentityConnection(v1, v2, e1.source);
-    tmp2 = SELECT v1 FROM vset:v1-((<Proof_Forward|<Proof_Backward|<Hold_Identity|<Resolve|<Reverse_Resolve):e2)-vset:v2
-          ACCUM @@edges += IdentityConnection(v2, v1, e2.source);
+
+    tmp1 = SELECT v1 FROM vset:v1-((Proof_Forward>|Proof_Backward>):e1)-vset:v2
+          ACCUM @@edges += IdentityConnection(v1, v2, e1.source, "Proof");
+    tmp2 = SELECT v1 FROM vset:v1-((<Proof_Forward|<Proof_Backward):e2)-vset:v2
+          ACCUM @@edges += IdentityConnection(v2, v1, e2.source, "Proof");
+    tmp3 = SELECT v1  FROM vset:v1-((Hold_Identity>):e1)-vset:i-((Resolve>):e2)-vset:v2
+          WHERE i.platform == "ens"
+          ACCUM
+            IF v1 == v2 THEN
+              @@edges += IdentityConnection(v1, i, e1.source, "Hold")
+            ELSE
+              @@edges += IdentityConnection(v1, i, e1.source, "Hold"),
+              @@edges += IdentityConnection(i, v2, e2.source, "Resolve")
+            END;
+    tmp4 = SELECT v1  FROM vset:v1-((Hold_Identity>):e1)-vset:i-((Resolve>):e2)-vset:v2
+          WHERE i.platform != "ens"
+          ACCUM @@edges += IdentityConnection(v1, i, e1.source, "Hold");
+
+    tmp5 = SELECT v1 FROM vset:v1-((Reverse_Resolve>):e1)-vset:v2
+          ACCUM @@edges += IdentityConnection(v1, v2, e1.source, "Reverse_Resolve");
 
     PRINT hv as graph_id, vset as vertices, @@edges as edges;
   END;

diff --git a/src/controller/tigergraphql/identity_graph.rs b/src/controller/tigergraphql/identity_graph.rs
@@ -65,6 +65,11 @@ impl IdentityGraph {
 
 #[Object]
 impl IdentityConnection {
+    /// Returns edge type connects start node and end node.
+    async fn edge_type(&self) -> String {
+        self.edge_type.clone()
+    }
+
     /// Returns data sources from upstreams supported by RelationService.
     async fn data_source(&self) -> DataSource {
         self.data_source

diff --git a/src/tigergraph/vertex/identity_graph.rs b/src/tigergraph/vertex/identity_graph.rs
@@ -31,6 +31,7 @@ struct IdentityGraphResponse {
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct IdentityConnection {
+    pub edge_type: String,
     pub data_source: DataSource,
     #[serde(rename = "source_v")]
     pub source: String,