Function Graph_Constraint

src/commands/cmd_constraint.c:380–439

command handler for GRAPH.CONSTRAINT command GRAPH.CONSTRAINT CREATE UNIQUE/MANDATORY [NODE label / RELATIONSHIP type] PROPERTIES prop_count prop0, prop1... GRAPH.CONSTRAINT DROP UNIQUE/MANDATORY [NODE label / RELATIONSHIP type] PROPERTIES prop_count prop0, prop1...

// GRAPH.CONSTRAINT CREATE <key> UNIQUE/MANDATORY [NODE label / RELATIONSHIP type] PROPERTIES prop_count prop0, prop1...
// GRAPH.CONSTRAINT DROP <key> UNIQUE/MANDATORY [NODE label / RELATIONSHIP type] PROPERTIES prop_count prop0, prop1...
int Graph_Constraint
(
	RedisModuleCtx *ctx,
	RedisModuleString **argv,
	int argc
) {
	if(argc < 8) {
		return RedisModule_WrongArity(ctx);
	}

	ConstraintOp op;
	ConstraintType ct;
	const char *label;
	uint8_t prop_count;
	RedisModuleString **props;
	GraphEntityType entity_type;
	RedisModuleString *key_name;

	//--------------------------------------------------------------------------
	// parse command arguments
	//--------------------------------------------------------------------------

	int res = Constraint_Parse(ctx, argv+1, argc-1, &key_name, &op, &ct,
			&entity_type, &label, &prop_count, &props);

	// command parsing error, abort
	if(res != REDISMODULE_OK) {
		return REDISMODULE_ERR;
	}

	// extract constraint properties and validate property name
	const char *props_cstr[prop_count];
	for(uint8_t i = 0; i < prop_count; i++) {
		props_cstr[i] = RedisModule_StringPtrLen(props[i], NULL);
		if(str_MatchRegex(PROPERTY_NAME_PATTERN, props_cstr[i]) == false) {
			RedisModule_ReplyWithErrorFormat(ctx, "Property name %s is invalid", props_cstr[i]);
			return REDISMODULE_ERR;
		}
	}

	bool success = false;

	if(op == CT_CREATE) {
		// try to create constraint
		success = _Constraint_Create(ctx, key_name, ct, entity_type, label,
				prop_count, props_cstr);
	} else {
		// CT_DROP
		success = _Constraint_Drop(ctx, key_name, ct, entity_type, label,
				prop_count, props_cstr);
	}

	if(success == true) {
		RedisModule_ReplyWithSimpleString(ctx, op == CT_CREATE ? "PENDING" : "OK");
		RedisModule_ReplicateVerbatim(ctx);
		return REDISMODULE_OK;
	}