Function BulkInsert

src/bulk_insert/bulk_insert.c:336–418

}

int BulkInsert
(
	RedisModuleCtx* ctx,
	GraphContext* gc,
	RedisModuleString** argv,
	int argc,
	uint node_count,
	uint edge_count
) {
	ASSERT(gc    !=  NULL);
	ASSERT(ctx   !=  NULL);
	ASSERT(argv  !=  NULL);

	if (argc < 2) {
		RedisModule_ReplyWithError(ctx, "Bulk insert format error, \
				failed to parse bulk insert sections.");
		return BULK_FAIL;
	}

	// read the number of node tokens
	long long node_token_count;
	long long relation_token_count;

	if (RedisModule_StringToLongLong(*argv++, &node_token_count) != REDISMODULE_OK) {
		RedisModule_ReplyWithError(ctx, "Error parsing number of node \
				descriptor tokens.");
		return BULK_FAIL;
	}

	// read the number of relation tokens
	if (RedisModule_StringToLongLong(*argv++, &relation_token_count) != REDISMODULE_OK) {
		RedisModule_ReplyWithError(ctx, "Error parsing number of relation \
				descriptor tokens.");
		return BULK_FAIL;
	}

	Graph* g = gc->g;
	int res = BULK_OK;

	// lock graph under write lock
	// allocate space for new nodes and edges
	// set graph sync policy to resize only
	Graph_AcquireWriteLock(g);
	Graph_SetMatrixPolicy(g, SYNC_POLICY_RESIZE);
	Graph_AllocateNodes(g, node_count);
	Graph_AllocateEdges(g, edge_count);

	argc -= 2;

	if (node_token_count > 0) {
		ASSERT(argc >= node_token_count);
		// process all node files
		if (_BulkInsert_ProcessTokens(gc, node_token_count, argv,
					SCHEMA_NODE)
				!= BULK_OK) {
			res = BULK_FAIL;
			goto cleanup;
		}