Function ParseRoleInheritanceResult

pg_documentdb/src/commands/roles.c:1627–1683

* ParseRoleInheritanceResult parses a BSON document from the role inheritance query. * Extracts the child_role and parent_roles fields. */

 * Extracts the child_role and parent_roles fields.
 */
static void
ParseRoleInheritanceResult(pgbson *rowBson, const char **childRole, List **parentRoles)
{
	bson_iter_t iter;
	PgbsonInitIterator(rowBson, &iter);

	*childRole = NULL;
	*parentRoles = NIL;

	while (bson_iter_next(&iter))
	{
		const char *key = bson_iter_key(&iter);

		if (strcmp(key, "child_role") == 0)
		{
			if (bson_iter_type(&iter) != BSON_TYPE_UTF8)
			{
				ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_INTERNALERROR),
								errmsg(
									"Invalid type for 'child_role' in role inheritance query result.")));
			}

			*childRole = bson_iter_utf8(&iter, NULL);
		}
		else if (strcmp(key, "parent_roles") == 0)
		{
			if (bson_iter_type(&iter) != BSON_TYPE_ARRAY)
			{
				ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_INTERNALERROR),
								errmsg(
									"Invalid type for 'parent_roles' in role inheritance query result.")));
			}

			bson_iter_t arrayIter;
			bson_iter_recurse(&iter, &arrayIter);
			while (bson_iter_next(&arrayIter))
			{
				if (bson_iter_type(&arrayIter) != BSON_TYPE_UTF8)
				{
					ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_INTERNALERROR),
									errmsg(
										"Invalid type for element in 'parent_roles' array in role inheritance query result.")));
				}

				const char *parentRole = bson_iter_utf8(&arrayIter, NULL);
				*parentRoles = lappend(*parentRoles, pstrdup(parentRole));
			}
		}
		else
		{
			ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_INTERNALERROR),
							errmsg(
								"Unknown field '%s' in role inheritance query result.",
								key)));
		}
	}
}