Function command_kill_op

pg_documentdb/src/commands/kill_op.c:91–151

* Implements the kill_op command. * The command takes a killOp wire protocol compatible command spec and makes an attempt to cancel * the operation uniquely identified by the `op` field in the command spec. * The op field is of the format : . * where * shardId is the unique identifier for the PG backend process running the operation. * opId is the microsecond timestamp of when

 * See src/commands/current_op.c for more details
 */
Datum
command_kill_op(PG_FUNCTION_ARGS)
{
	pgbson *commandSpec = PG_GETARG_PGBSON(0);
	ParsedKillOpArgs parsedArgs;
	memset(&parsedArgs, 0, sizeof(ParsedKillOpArgs));

	ValidateAndParseKillOpCommand(commandSpec, &parsedArgs);

	/*
	 * We need a null terminated string for pg_strtoint64
	 */
	char *shardIdString = palloc(parsedArgs.shardIdView.length + 1); /* +1 for null terminator */
	memcpy(shardIdString, parsedArgs.shardIdView.string, parsedArgs.shardIdView.length);
	shardIdString[parsedArgs.shardIdView.length] = '\0';
	int64 shardId = pg_strtoint64(shardIdString);

	if (shardId <= SINGLE_NODE_ID)
	{
		ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_BADVALUE),
						errmsg("Invalid shardid: %s", shardIdString),
						errdetail_log(
							"Invalid shardid: %s, shardId provided doesn't match the"
							" expected format from currentOp", shardIdString)));
	}

	int nargs = 0;
	bool readOnly = false;
	Oid *argTypes = NULL;
	Datum *argValues = NULL;
	char *argNulls;

	const char *killOpQuery = GetOperationCancellationQuery(shardId, &parsedArgs.opIdView,
															&nargs, &argTypes, &argValues,
															&argNulls,
															GetDefaultOperationCancellationQuery);

	/* If hook doesn't provide any query it's no-op success */
	if (killOpQuery != NULL && nargs > 0 && argTypes != NULL &&
		argValues != NULL && argNulls != NULL)
	{
		bool isNull = false;
		ExtensionExecuteQueryWithArgsViaSPI(killOpQuery, nargs, argTypes, argValues,
											argNulls, readOnly, SPI_OK_SELECT, &isNull);
	}

	/* Build success response */
	pgbson_writer writer;
	PgbsonWriterInit(&writer);

	StringInfo shardName = makeStringInfo();
	int processId = shardId % SINGLE_NODE_ID;
	int shardNameId = (int) ((shardId - processId) / SINGLE_NODE_ID);
	appendStringInfo(shardName, "shard%d", shardNameId);

	PgbsonWriterAppendUtf8(&writer, "shard", 5, shardName->data);
	PgbsonWriterAppendInt32OrDouble(&writer, "shardid", 7, processId);
	PgbsonWriterAppendDouble(&writer, "ok", 2, 1.0);