feat(trino-driver): Add special testConnection for Trino

Author: KSDaemon

packages/cubejs-prestodb-driver/src/PrestoDriver.ts

@@ -60,11 +60,11 @@ export class PrestoDriver extends BaseDriver implements DriverInterface {
     return 2;
   }
 
-  private config: PrestoDriverConfiguration;
+  protected readonly config: PrestoDriverConfiguration;
 
-  private catalog: string | undefined;
+  protected readonly catalog: string | undefined;
 
-  private client: any;
+  protected client: any;
 
   /**
    * Class constructor.

packages/cubejs-schema-compiler/src/adapter/index.ts

@@ -16,9 +16,9 @@ export * from './CubeStoreQuery';
 export * from './MysqlQuery';
 export * from './PostgresQuery';
 export * from './MssqlQuery';
+export * from './PrestodbQuery';
 
 // Candidates to move from this package to drivers packages
-// export * from './PrestodbQuery';
 // export * from './RedshiftQuery';
 // export * from './SnowflakeQuery';
 // export * from './SqliteQuery';

packages/cubejs-trino-driver/package.json

@@ -29,6 +29,7 @@
     "@cubejs-backend/prestodb-driver": "1.3.18",
     "@cubejs-backend/schema-compiler": "1.3.18",
     "@cubejs-backend/shared": "1.3.18",
+    "node-fetch": "^2.6.1",
     "presto-client": "^0.12.2",
     "sqlstring": "^2.3.1"
   },

packages/cubejs-trino-driver/src/TrinoDriver.ts

@@ -1,5 +1,6 @@
+import fetch from 'node-fetch';
 import { PrestoDriver } from '@cubejs-backend/prestodb-driver';
-import { PrestodbQuery } from '@cubejs-backend/schema-compiler/dist/src/adapter/PrestodbQuery';
+import { PrestodbQuery } from '@cubejs-backend/schema-compiler';
 
 export class TrinoDriver extends PrestoDriver {
   public constructor(options: any) {
@@ -9,4 +10,23 @@ export class TrinoDriver extends PrestoDriver {
   public static dialectClass() {
     return PrestodbQuery;
   }
+
+  public override async testConnection(): Promise<void> {
+    const { host, port, basic_auth: basicAuth } = this.config;
+    const url = `https://${host}:${port}/v1/info`;
+    const headers: Record<string, string> = {};
+
+    if (basicAuth) {
+      const { user, password } = basicAuth;
+      const encoded = Buffer.from(`${user}:${password}`).toString('base64');
+      headers.Authorization = `Basic ${encoded}`;
+    }
+
+    const response = await fetch(url, { method: 'GET', headers });
+
+    if (!response.ok) {
+      const text = await response.text();
+      throw new Error(`Connection test failed: ${response.status} ${response.statusText} - ${text}`);
+    }
+  }
 }