research-ocp-fhe/demo-frontend/lib/redis.ts

import { Redis } from '@upstash/redis';

// Redis client configuration for Upstash
const redis = new Redis({
  url: process.env.UPSTASH_REDIS_REST_URL,
  token: process.env.UPSTASH_REDIS_REST_TOKEN,
});

// Redis key prefixes for organization
export const KEY_PREFIXES = {
  COMPANY: 'company',
  COMPANY_BY_FOUNDER: 'founder',
  COMPANY_BY_INVESTOR: 'companyinvestor',
  COMPANY_ROUNDS: 'rounds',
  COMPANY_INVESTORS: 'investors',
  COMPANY_SEARCH: 'search',
  ROUND_INVESTMENTS: 'roundinvestments', // For storing round-specific investment security IDs
} as const;

// Utility function to generate Redis keys
export const generateKey = (prefix: string, identifier: string): string =>
  `${prefix}:${identifier}`;

// Health check function for Upstash Redis
export const checkRedisHealth = async (): Promise<boolean> => {
  try {
    await redis.ping();
    return true;
  } catch (error) {
    console.error('Redis health check failed:', error);
    return false;
  }
};

export default redis;