[BABEL] Functions find_ENR_queryEnv, get_toast_parent_oid moved from header file to C file. (#728)

Functions moved to queryenv C file

Co-authored-by: Ayush-061 <ayushhx@amazon.com>

Author: Ayush-061

src/backend/utils/misc/queryenvironment.c

@@ -54,6 +54,10 @@
 #include "utils/rel.h"
 
 #define NUM_ENR_CATALOGS 11
+#define BBF_TEMP_TOAST_PREFIX "#pg_toast_"
+#define BBF_TABLEVAR_TOAST_PREFIX "@pg_toast_"
+#define BBF_TOAST_PREFIX_LEN 10
+#define BASE_10 10
 
 find_object_in_enr_hook_type find_object_in_enr_hook = NULL;
 is_enr_to_sys_object_dependency_hook_type is_enr_to_sys_object_dependency_hook = NULL;
@@ -1965,4 +1969,43 @@ bool has_existing_enr_relations()
 	}
 
 	return false;
+}
+
+/*
+ * get_toast_parent_oid
+ *
+ * Extract parent table OID from temp toast table name.
+ * Toast table names follow the pattern "#pg_toast_<parent_rel_oid>" for temp tables
+ * or "@pg_toast_<parent_rel_oid>" for table variables.
+ * Returns the parent OID, or InvalidOid if the name doesn't match either pattern.
+ */
+Oid
+get_toast_parent_oid(const char *relname)
+{
+	if (strncmp(relname, BBF_TEMP_TOAST_PREFIX, BBF_TOAST_PREFIX_LEN) == 0 ||
+		strncmp(relname, BBF_TABLEVAR_TOAST_PREFIX, BBF_TOAST_PREFIX_LEN) == 0)
+		return (Oid) strtoul(relname + BBF_TOAST_PREFIX_LEN, NULL, BASE_10);
+	return InvalidOid;
+}
+
+/*
+ * find_ENR_queryEnv
+ *
+ * Find the query environment where an ENR with the given OID is registered.
+ * Searches from currentQueryEnv up through parent environments.
+ * Returns the queryEnv where the ENR is found, or NULL if not found.
+ */
+QueryEnvironment *
+find_ENR_queryEnv(Oid relid)
+{
+	QueryEnvironment *qe = currentQueryEnv;
+
+	while (qe)
+	{
+		if (get_ENR_withoid(qe, relid, ENR_TSQL_TEMP, false))
+			return qe;
+		qe = qe->parentEnv;
+	}
+
+	return NULL;
 }

src/include/utils/queryenvironment.h

@@ -157,6 +157,8 @@ extern bool ENRGetSystableScan(Relation rel, Oid indexoid, int nkeys, ScanKey ke
 extern bool ENRAddTuple(Relation rel, HeapTuple tup);
 extern bool ENRDropTuple(Relation rel, HeapTuple tup);
 extern bool ENRUpdateTuple(Relation rel, HeapTuple tup);
+extern Oid get_toast_parent_oid(const char *relname);
+extern QueryEnvironment *find_ENR_queryEnv(Oid relid);
 
 extern void ENRDropEntry(Oid id);
 extern PGDLLEXPORT void ENRDropTempTables(QueryEnvironment *queryEnv);
@@ -188,51 +190,4 @@ extern PGDLLIMPORT find_object_in_enr_hook_type find_object_in_enr_hook;
 typedef bool (*is_enr_to_sys_object_dependency_hook_type) (const ObjectAddress *depender, const ObjectAddress *referenced);
 extern PGDLLIMPORT is_enr_to_sys_object_dependency_hook_type is_enr_to_sys_object_dependency_hook;
 
-/*
- * Babelfish toast table name prefixes and length.
- * Toast tables for temp tables are named "#pg_toast_<parent_rel_oid>".
- * Toast tables for table variables are named "@pg_toast_<parent_rel_oid>".
- * Both prefixes have the same length (10 characters).
- */
-#define BBF_TEMP_TOAST_PREFIX		"#pg_toast_"
-#define BBF_TABLEVAR_TOAST_PREFIX	"@pg_toast_"
-#define BBF_TOAST_PREFIX_LEN		10
-
-/*
- * Extract parent table OID from temp toast table name.
- * Toast table names follow the pattern "#pg_toast_<parent_rel_oid>" for temp tables
- * or "@pg_toast_<parent_rel_oid>" for table variables.
- * Returns the parent OID, or InvalidOid if the name doesn't match either pattern.
- */
-static inline Oid
-get_toast_parent_oid(const char *relname)
-{
-	if (strncmp(relname, BBF_TEMP_TOAST_PREFIX, BBF_TOAST_PREFIX_LEN) == 0 ||
-		strncmp(relname, BBF_TABLEVAR_TOAST_PREFIX, BBF_TOAST_PREFIX_LEN) == 0)
-		return (Oid) strtoul(relname + BBF_TOAST_PREFIX_LEN, NULL, 10);
-	return InvalidOid;
-}
-
-/*
- * find_ENR_queryEnv
- *
- * Find the query environment where an ENR with the given OID is registered.
- * Searches from currentQueryEnv up through parent environments.
- * Returns the queryEnv where the ENR is found, or NULL if not found.
- */
-static inline QueryEnvironment *
-find_ENR_queryEnv(Oid relid)
-{
-	QueryEnvironment *qe = currentQueryEnv;
-
-	while (qe)
-	{
-		if (get_ENR_withoid(qe, relid, ENR_TSQL_TEMP, false))
-			return qe;
-		qe = qe->parentEnv;
-	}
-
-	return NULL;
-}
-
-#endif							/* QUERYENVIRONMENT_H */
+#endif							/* QUERYENVIRONMENT_H */