Function Graph_BulkInsert

src/commands/cmd_bulk_insert.c:51–117

}

int Graph_BulkInsert(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
	if(argc < 3) return RedisModule_WrongArity(ctx);

	// bulk-insert process a batch of node/edge creation generated by the
	// bulk-insert utility

	GraphContext *gc     = NULL;
	long long node_count = 0;  // number of declared nodes
	long long edge_count = 0;  // number of declared edges

	// get graph name
	argv += 1; // skip "GRAPH.BULK"
	RedisModuleString *rs_graph_name = *argv++;
	const char *graphname = RedisModule_StringPtrLen(rs_graph_name, NULL);
	argc -= 2; // skip "GRAPH.BULK [GRAPHNAME]"

	bool begin = false;
	if(_Graph_Bulk_Begin(ctx, &argv, &argc, rs_graph_name, graphname, &begin)
			!= BULK_OK) goto cleanup;

	gc = GraphContext_Retrieve(ctx, rs_graph_name, false, begin);

	// failed to retrieve GraphContext; an error has been emitted
	if(gc == NULL) goto cleanup;

	// read the user-provided counts for nodes and edges in the current query
	if(RedisModule_StringToLongLong(*argv++, &node_count) != REDISMODULE_OK) {
		RedisModule_ReplyWithError(ctx, "Error parsing node count.");
		goto cleanup;
	}

	if(RedisModule_StringToLongLong(*argv++, &edge_count) != REDISMODULE_OK) {
		RedisModule_ReplyWithError(ctx, "Error parsing relation count.");
		goto cleanup;
	}

	argc -= 2; // already read node count and edge count

	int rc = BulkInsert(ctx, gc, argv, argc, node_count, edge_count);

	if(rc == BULK_FAIL) {
		// if insertion failed, clean up keyspace and free added entities
		GraphContext_DecreaseRefCount(gc);
		RedisModuleKey *key = NULL;

		key = RedisModule_OpenKey(ctx, rs_graph_name, REDISMODULE_WRITE);
		RedisModule_DeleteKey(key);
		RedisModule_CloseKey(key);

		gc = NULL;
		goto cleanup;
	}

	// successful bulk commands should always modify slaves
	RedisModule_ReplicateVerbatim(ctx);

	// replay to caller
	char reply[1024];