Skip to content

Conversation

@natan-hoppe-workday
Copy link

Description

Added dedicated class RedisConnector to initialize and connect to Redis while supporting sync and async usages, as well reconnections in case of failures.

Updated the following files with its usage:

  • packages/server/src/CachePool.ts
  • packages/server/src/enterprise/middleware/passport/SessionPersistance.ts
  • packages/server/src/utils/rateLimit.ts

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello @natan-hoppe-workday, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly enhances the stability and maintainability of Redis interactions within the application by consolidating all Redis connection and initialization logic into a dedicated RedisConnector class. This change addresses potential issues with unstable Redis connections by providing a robust, centralized mechanism for managing Redis clients, including automatic reconnection capabilities and standardized configuration parsing, thereby improving overall system reliability.

Highlights

  • New RedisConnector Class: A new RedisConnector class has been introduced in packages/components/src/RedisConnector.ts. This class centralizes the logic for initializing and connecting to a Redis instance, supporting both synchronous and asynchronous usage patterns.
  • Robust Redis Connection Handling: The RedisConnector handles various Redis configuration options from environment variables (e.g., REDIS_URL, REDIS_HOST, REDIS_TLS, REDIS_KEEP_ALIVE, REDIS_CERT, etc.) and includes built-in reconnection logic (reconnectOnError) to improve stability against connection failures.
  • Refactored Redis Client Usage: Existing direct ioredis client instantiations across the codebase have been refactored to utilize the new RedisConnector. This includes updates in CachePool.ts, SessionPersistance.ts, and rateLimit.ts, ensuring a consistent and centralized approach to Redis connectivity.
  • Simplified Redis Integration: The new class abstracts away complex Redis initialization details, providing simple getRedisClient() and getRedisConnection() methods, which trigger lazy initialization and ensure the client is ready for use.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a RedisConnector class to centralize and improve Redis connection handling, which is a great initiative. The refactoring correctly applies this new connector across CachePool, SessionPersistance, and rateLimit. However, the implementation of RedisConnector has a critical flaw in how it provides the Redis client instance synchronously, which will lead to runtime errors. There's also an issue with the reconnection logic's implementation not matching its documented intent. My review focuses on these key areas to ensure the connector is robust and reliable.

Comment on lines +151 to +155
public getRedisClient(): Redis {
// Trigger async init if not yet started
void this.init()
return this.redis
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

critical

This method has a critical race condition. It calls this.init() which starts an asynchronous initialization process, and then immediately returns this.redis. The this.redis property is only assigned inside an async IIFE within init(), so it will be undefined when this method returns. This will cause TypeError exceptions in all places where this method is used, such as in CachePool, SessionPersistance, and RateLimiterManager.

To fix this, the Redis instance must be created synchronously before this method returns. The asynchronous part (like await this.redis.connect()) can remain in the promise.

A possible approach:

  1. Refactor init() and initializeQueueMode() to synchronously create the Redis instance and assign it to this.redis.
  2. The initPromise should then only handle the asynchronous connection logic (i.e., await this.redis.connect()).

With such a change, getRedisClient() can safely return this.redis because it will be available synchronously.

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The race condition is addressed via the ready() method which must be called during async requests as documented.

For example:

const connector = new RedisConnector()
await connector.ready() // fully waits for Redis init
const redis = connector.getRedisClient()

Comment on lines +172 to +176
public getRedisConnection(): Record<string, unknown> {
// Trigger async init if not yet started
void this.init()
return this.connection
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

critical

Similar to getRedisClient(), this method has a critical race condition. It returns this.connection which is assigned asynchronously within init(). This will result in undefined being returned, causing failures in consumers like RateLimiterManager when setting up bullmq.

The fix is the same as for getRedisClient(): ensure this.connection is assigned synchronously during the initialization phase.

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The race condition is addressed via the ready() method which must be called during async requests as documented.

For example:

const connector = new RedisConnector()
await connector.ready() // fully waits for Redis init
const redis = connector.getRedisClient()

Comment on lines +139 to +142
private connectOnError(err: Error): number {
logger.error(`[server]: Redis connection error - ${err.message}`)
return 1
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The comment on line 136 states that this function does not retry the failed command. However, according to the ioredis documentation, returning 1 causes ioredis to retry the failed command. To reconnect without retrying the command, the function should return 2.

This mismatch could lead to unexpected behavior, especially when dealing with non-idempotent commands. Please update the return value to align with the intended behavior described in the comment.

Suggested change
private connectOnError(err: Error): number {
logger.error(`[server]: Redis connection error - ${err.message}`)
return 1
}
private connectOnError(err: Error): number {
logger.error(`[server]: Redis connection error - ${err.message}`)
return 2
}

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wrong feedback.

The source code and documentation confirm the number 1 should be used for reconnection without command retry.

Screenshot 2025-11-19 at 8 31 20 AM

@@ -0,0 +1,179 @@
import { InternalFlowiseError } from '../../server/src/errors/internalFlowiseError'
Copy link
Contributor

@HenryHengZJ HenryHengZJ Nov 22, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

in components, we can't import stuff from server using relative path like this - it will likely cause circular dependency. Relative path imports can only be used if you're importing from the same components folder - this is how it works in a monorepo

search for isPathTraversal you can see how a common function can be created in components, and exposed via index.ts, then later used in files inside server

Since you are trying to create a singleton Redis instance that can be used throughout the application, I recommend to move this to server/src

@HenryHengZJ
Copy link
Contributor

We also need to test if the modified pieces related to Redis (cache, session, rate limit, etc) works on dev server

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants