Auth - recreate client on getconection if disconnected

app/api/auth/[...nextauth]/options.ts

@@ -1,9 +1,55 @@
 import { RedisClientType, createClient } from "falkordb";
 import CredentialsProvider from "next-auth/providers/credentials"
-import { AuthOptions } from "next-auth"
+import { AuthOptions, User } from "next-auth"
 
 
-export const connections = new Map<number, RedisClientType>();
+const connections = new Map<number, RedisClientType>();
+
+async function newClient(credentials: {host: string, port: string, password: string, username: string}, id: number) {
+    const client = await createClient({
+        socket: {
+            host: credentials.host ?? "localhost",
+            port: credentials.port ? parseInt(credentials.port) : 6379,
+            reconnectStrategy: false
+        },
+        password: credentials.password ?? undefined,
+        username: credentials.username ?? undefined
+    })
+
+    // Save connection in connections map for later use
+    connections.set(id, client as RedisClientType)
+
+    await client.on('error', err => {
+        // Close coonection on error and remove from connections map
+        console.error('FalkorDB Client Error', err)
+        const connection = connections.get(id)
+        if (connection) {
+            connections.delete(id)
+            connection.disconnect()
+            .catch((e) => {
+                console.warn('FalkorDB Client Disconnect Error', e)
+            })
+        }
+    }).connect();
+
+    // Verify connection
+    await client.ping()
+    return client
+}
+
+export async function getConnection(user: User) {
+    let conn = connections.get(user.id)
+    if (!conn) {
+        await newClient({
+            host: user.host,
+            port: user.port.toString() ?? "6379",
+            username: user.username,
+            password: user.password,
+        }, user.id)
+    }
+    return conn  
+}
+
 let userId = 1;
 
 const authOptions: AuthOptions = {
@@ -26,34 +72,7 @@ const authOptions: AuthOptions = {
                     const id = userId;
                     userId += 1;
 
-                    const client = await createClient({
-                        socket: {
-                            host: credentials.host ?? "localhost",
-                            port: credentials.port ? parseInt(credentials.port, 10) : 6379,
-                            reconnectStrategy: false
-                        },
-                        password: credentials.password ?? undefined,
-                        username: credentials.username ?? undefined
-                    })
-
-                    // Save connection in connections map for later use
-                    connections.set(id, client as RedisClientType)
-
-                    await client.on('error', err => {
-                        // Close coonection on error and remove from connections map
-                        console.error('FalkorDB Client Error', err)
-                        const connection = connections.get(id)
-                        if (connection) {
-                            connections.delete(id)
-                            connection.disconnect()
-                            .catch((e) => {
-                                console.warn('FalkorDB Client Disconnect Error', e)
-                            })
-                        }
-                    }).connect();
-
-                    // Verify connection
-                    await client.ping()
+                    await newClient(credentials, id)
 
                     const res = {
                         id,

app/api/graph/[graph]/[node]/route.ts

@@ -1,7 +1,7 @@
 import { NextRequest, NextResponse } from "next/server";
 import { Graph } from 'falkordb';
 import { getServerSession } from "next-auth/next";
-import authOptions, { connections } from "../../../auth/[...nextauth]/options";
+import authOptions, { getConnection } from "../../../auth/[...nextauth]/options";
 
 // eslint-disable-next-line import/prefer-default-export
 export async function GET(request: NextRequest, { params }: { params: { graph: string, node: string } }) {
@@ -12,7 +12,7 @@ export async function GET(request: NextRequest, { params }: { params: { graph: s
         return NextResponse.json({ message: "Not authenticated" }, { status: 401 })
     }
 
-    const client = connections.get(id)
+    const client = await getConnection(session.user)
     if (!client) {
         return NextResponse.json({ message: "Not authenticated" }, { status: 401 })
     }

app/api/graph/route.ts

@@ -1,7 +1,7 @@
 import { NextRequest, NextResponse } from "next/server";
 import { Graph } from 'falkordb';
 import { getServerSession } from "next-auth/next";
-import authOptions, { connections } from "../auth/[...nextauth]/options";
+import authOptions, { getConnection } from "../auth/[...nextauth]/options";
 
 // eslint-disable-next-line import/prefer-default-export
 export async function GET(request: NextRequest) {
@@ -12,7 +12,7 @@ export async function GET(request: NextRequest) {
         return NextResponse.json({ message: "Not authenticated" }, { status: 401 })
     }
 
-    const client = connections.get(id)
+    const client = await getConnection(session.user)
     if(!client) {
         return NextResponse.json({ message: "Not authenticated" }, { status: 401 })
     }