서브스크립션과 실시간 기능 구현

// src/common/pubsub.provider.ts (새 파일)
import { Provider } from '@nestjs/common';
import { RedisPubSub } from 'graphql-redis-subscriptions';
import * as Redis from 'ioredis';

export const REDIS_PUB_SUB = 'REDIS_PUB_SUB'; // 주입 토큰

export const pubSubProvider: Provider = {
  provide: REDIS_PUB_SUB,
  useFactory: () => {
    const options = {
      host: process.env.REDIS_HOST || 'localhost',
      port: parseInt(process.env.REDIS_PORT || '6379', 10),
      retryStrategy: times => {
        // 재접속 로직
        return Math.min(times * 50, 2000);
      },
    };
    return new RedisPubSub({
      publisher: new Redis.Redis(options),
      subscriber: new Redis.Redis(options),
    });
  },
};

// src/app.module.ts (imports 및 providers 업데이트)
import { REDIS_PUB_SUB, pubSubProvider } from './common/pubsub.provider'; // 임포트

@Module({
  imports: [
    GraphQLModule.forRoot<ApolloDriverConfig>({
      // ... 기존 설정
      subscriptions: {
        'graphql-ws': true,
        onConnect: (context: any) => {
            // WebSocket 연결 시 인증 로직
            // const token = context.connectionParams.authToken;
            // if (!token) {
            //   throw new Error('Auth token not provided');
            // }
            // const user = verifyToken(token);
            // return { user, usersService: app.get(UsersService) }; // 컨텍스트에 사용자 정보 및 서비스 주입
            return true; // 연결 허용
        },
        // onDisconnect: () => console.log('Disconnected'),
      },
    }),
    // ...
  ],
  controllers: [AppController],
  providers: [
    AppService,
    pubSubProvider, // RedisPubSub 프로바이더 등록
  ],
})
export class AppModule {}

// src/posts/posts.resolver.ts (업데이트)
import { Resolver, Query, Mutation, Args, Int, ResolveField, Parent, Subscription } from '@nestjs/graphql';
import { Post } from './models/post.model';
import { User } from '../users/models/user.model';
import { CreatePostInput } from './dto/create-post.input';
import { UpdatePostInput } from './dto/update-post.input';
import { PubSub } from 'graphql-subscriptions'; // 또는 RedisPubSub
import { UsersLoader } from '../users/loaders/users.loader';
import { GqlContext } from 'src/common/decorators/gql-context.decorator';
import { Inject } from '@nestjs/common';
import { REDIS_PUB_SUB } from 'src/common/pubsub.provider'; // RedisPubSub 주입 토큰 임포트

// const pubSub = new PubSub(); // 이제 이 라인은 제거하거나 RedisPubSub을 사용
@Resolver(() => Post)
export class PostsResolver {
  private readonly posts: Post[] = [ /* ... */ ];
  private nextPostId = 4;

  constructor(
    private readonly usersLoader: UsersLoader,
    @Inject(REDIS_PUB_SUB) private pubSub: PubSub, // PubSub 인스턴스 주입
  ) {}

  // ... (Query, Mutation 메서드)

  @Mutation(() => Post, { description: '새로운 게시물을 생성합니다.' })
  async createPost(@Args('createPostInput') createPostInput: CreatePostInput): Promise<Post> {
    // ...
    await this.pubSub.publish('postAdded', { postAdded: newPost }); // 주입받은 pubSub 사용
    return newPost;
  }

  // ... (다른 Mutation 메서드도 this.pubSub 사용하도록 수정)

  @Subscription(() => Post, { description: '새로운 게시물이 생성될 때 알림을 받습니다.' })
  postAdded() {
    return this.pubSub.asyncIterator('postAdded');
  }

  // ... (다른 Subscription 메서드)
}