LKE-14350: Add the bulk create nodes/edges API

Author: gmasclet

src/api/GraphEdge/index.ts

@@ -9,7 +9,13 @@ import {LkErrorKey} from '../../http/response';
 import {LkEdge, LkSubGraph} from '../graphItemTypes';
 import {IDataSourceParams} from '../commonTypes';
 
-import {ICreateEdgeParams, IDeleteEdgeParams, IGetEdgeParams, IUpdateEdgeParams} from './types';
+import {
+  BulkCreateEdgesParams,
+  ICreateEdgeParams,
+  IDeleteEdgeParams,
+  IGetEdgeParams,
+  IUpdateEdgeParams
+} from './types';
 
 export * from './types';
 
@@ -68,6 +74,18 @@ export class GraphEdgeAPI extends Request {
     });
   }
 
+  /**
+   * Add a chunk of edges to the graph.
+   */
+  public bulkCreateEdges(this: Request<{items: LkEdge[]}>, params: BulkCreateEdgesParams) {
+    return this.request({
+      errors: [UNAUTHORIZED, DATA_SOURCE_UNAVAILABLE, FORBIDDEN],
+      url: '/:sourceKey/graph/bulk/edges',
+      method: 'POST',
+      params: params
+    });
+  }
+
   /**
    * Get the number of edges in the graph.
    */

src/api/GraphEdge/types.ts

@@ -10,13 +10,15 @@ export interface IGetEdgeParams extends IGetSubGraphParams {
   id: string;
 }
 
-export interface ICreateEdgeParams extends IDataSourceParams {
+interface EdgeParams {
   source: string;
   target: string;
   type: string;
   properties?: GenericObject;
 }
 
+export interface ICreateEdgeParams extends EdgeParams, IDataSourceParams {}
+
 export interface IUpdateEdgeParams extends IDataSourceParams {
   id: string;
   properties?: GenericObject;
@@ -27,3 +29,7 @@ export interface IUpdateEdgeParams extends IDataSourceParams {
 export interface IDeleteEdgeParams extends IDataSourceParams {
   id: string;
 }
+
+export interface BulkCreateEdgesParams extends IDataSourceParams {
+  edges: EdgeParams[];
+}

src/api/GraphNode/index.ts

@@ -10,6 +10,7 @@ import {LkNode, LkNodeStatistics, LkSubGraph} from '../graphItemTypes';
 import {IDataSourceParams} from '../commonTypes';
 
 import {
+  BulkCreateNodesParams,
   ICreateNodeParams,
   IDeleteNodeParams,
   IGetAdjacentNodesParams,
@@ -98,6 +99,24 @@ export class GraphNodeAPI extends Request {
     });
   }
 
+  /**
+   * Add a chunk of nodes to the graph.
+   */
+  public bulkCreateNodes(this: Request<{items: LkNode[]}>, params: BulkCreateNodesParams) {
+    return this.request({
+      errors: [
+        UNAUTHORIZED,
+        DATA_SOURCE_UNAVAILABLE,
+        FORBIDDEN,
+        INVALID_PARAMETER,
+        CONSTRAINT_VIOLATION
+      ],
+      url: '/:sourceKey/graph/bulk/nodes',
+      method: 'POST',
+      params: params
+    });
+  }
+
   /**
    * Get the number of nodes in the graph.
    */

src/api/GraphNode/types.ts

@@ -10,12 +10,14 @@ export interface IGetNodeParams extends IGetSubGraphParams {
   id: string;
 }
 
-export interface ICreateNodeParams extends IDataSourceParams {
+interface NodeParams {
   categories: string[];
   // TODO type it as LkProperties
   properties?: GenericObject;
 }
 
+export interface ICreateNodeParams extends NodeParams, IDataSourceParams {}
+
 export interface IUpdateNodeParams extends IDataSourceParams {
   id: string;
   addedCategories?: string[];
@@ -29,6 +31,10 @@ export interface IDeleteNodeParams extends IDataSourceParams {
   id: string;
 }
 
+export interface BulkCreateNodesParams extends IDataSourceParams {
+  nodes: NodeParams[];
+}
+
 export interface IGetStatisticsParams extends IDataSourceParams {
   ids: string[];
   withDigest?: boolean;