From ae19d149516b347bc9af90fa68827a26229457f5 Mon Sep 17 00:00:00 2001
From: Nybkox <nybkox@gmail.com>
Date: Thu, 5 Dec 2024 15:31:50 +0100
Subject: [PATCH] feat: add return type to pssql funcs

---
 packages/types/dbinfo.d.ts                             | 4 +++-
 plugins/dbgate-plugin-postgres/src/backend/Analyser.js | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/types/dbinfo.d.ts b/packages/types/dbinfo.d.ts
index 609669fa2..45511e271 100644
--- a/packages/types/dbinfo.d.ts
+++ b/packages/types/dbinfo.d.ts
@@ -132,7 +132,9 @@ export interface CallableObjectInfo extends SqlObjectInfo {
 
 export interface ProcedureInfo extends CallableObjectInfo {}
 
-export interface FunctionInfo extends CallableObjectInfo {}
+export interface FunctionInfo extends CallableObjectInfo {
+  returnType?: string;
+}
 
 export interface TriggerInfo extends SqlObjectInfo {}
 
diff --git a/plugins/dbgate-plugin-postgres/src/backend/Analyser.js b/plugins/dbgate-plugin-postgres/src/backend/Analyser.js
index 90bb94ab8..bae89b4eb 100644
--- a/plugins/dbgate-plugin-postgres/src/backend/Analyser.js
+++ b/plugins/dbgate-plugin-postgres/src/backend/Analyser.js
@@ -346,6 +346,7 @@ class Analyser extends DatabaseAnalyser {
           schemaName: func.schema_name,
           contentHash: func.hash_code,
           parameters: functionNameToParameters[`${func.schema_name}.${func.pure_name}`],
+          returnType: func.data_type,
         })),
     };