Method functions

database/postgres/database.go:601–653

functions fetches user-defined functions from the database

()

// functions fetches user-defined functions from the database
func (d *PostgresDatabase) functions() ([]string, error) {
	// Query to get user-defined functions (excluding system functions and extension functions)
	// We use pg_get_functiondef to get the complete function definition
	// pg_get_function_identity_arguments gives us the function signature for comments
	rows, err := d.db.Query(`
		SELECT n.nspname AS func_schema,
		       p.proname AS func_name,
		       pg_get_functiondef(p.oid) AS func_def,
		       pg_get_function_identity_arguments(p.oid) AS func_args,
		       obj_description(p.oid, 'pg_proc') AS func_comment
		FROM pg_catalog.pg_proc p
		INNER JOIN pg_catalog.pg_namespace n ON p.pronamespace = n.oid
		WHERE n.nspname NOT IN ('pg_catalog', 'information_schema', 'pg_toast', 'sys')
		  AND p.prokind = 'f'
		  AND NOT EXISTS (SELECT 1 FROM pg_depend d WHERE d.objid = p.oid AND d.classid = (SELECT oid FROM pg_catalog.pg_class WHERE relname = 'pg_proc') AND d.deptype = 'e')
		ORDER BY n.nspname, p.proname;
	`)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var ddls []string
	for rows.Next() {
		var funcSchema, funcName, funcDef, funcArgs string
		var funcComment sql.NullString
		if err := rows.Scan(&funcSchema, &funcName, &funcDef, &funcArgs, &funcComment); err != nil {
			return nil, err
		}
		if d.config.TargetSchema != nil && !slices.Contains(d.config.TargetSchema, funcSchema) {
			continue
		}
		// pg_get_functiondef returns the complete CREATE FUNCTION statement
		// We just need to ensure it ends with a semicolon
		funcDef = strings.TrimSpace(funcDef)
		if !strings.HasSuffix(funcDef, ";") {
			funcDef += ";"
		}
		ddls = append(ddls, funcDef)

		// Add comment if exists
		if funcComment.Valid {
			ddls = append(ddls, fmt.Sprintf(
				"COMMENT ON FUNCTION %s.%s(%s) IS %s;",
				d.quoteIdentifierIfNeeded(funcSchema), d.quoteIdentifierIfNeeded(funcName), funcArgs, schemaLib.StringConstant(funcComment.String),
			))
		}
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return ddls, nil
}

// triggers fetches user-defined triggers from the database
func (d *PostgresDatabase) triggers() ([]string, error) {