feat: add support for mongodb+srv protocol

kapetacom · Mar 1, 2024 · 7ec1c90 · 7ec1c90
1 parent 219a2fd
commit 7ec1c90
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/utils.ts b/src/utils.ts
@@ -8,13 +8,15 @@ export const RESOURCE_TYPE = 'kapeta/resource-type-mongodb';
 export const PORT_TYPE = 'mongodb';
 export async function createDBURI(provider:ConfigProvider, resourceName: string) {
     const dbInfo = await provider.getResourceInfo(RESOURCE_TYPE, PORT_TYPE, resourceName);
+    console.log("resourceName", resourceName, dbInfo);
     if (!dbInfo) {
         throw new Error(`Resource ${resourceName} not found`);
     }
     const dbName =
         dbInfo.options && dbInfo.options.dbName
             ? dbInfo.options.dbName
             : resourceName;
+    const protocol = dbInfo.options && dbInfo.options.protocol ? dbInfo.options.protocol : 'mongodb';
 
     let credentials = ''
     if (dbInfo.credentials?.username) {
@@ -25,5 +27,11 @@ export async function createDBURI(provider:ConfigProvider, resourceName: string)
         }
     }
 
-    return `mongodb://${credentials}@${dbInfo.host}:${dbInfo.port}/${encodeURIComponent(dbName)}?authSource=admin&directConnection=true`;
+    if (protocol == 'mongodb+srv') {
+        const result = `mongodb+srv://${credentials}@${dbInfo.host}/${encodeURIComponent(dbName)}?authSource=admin`;
+        console.log("mongo db url", result);
+        return result;
+    } else {
+        return `mongodb://${credentials}@${dbInfo.host}:${dbInfo.port}/${encodeURIComponent(dbName)}?authSource=admin&directConnection=true`;
+    }
 }