Subscription Service API

import { ApolloClient, InMemoryCache, split, HttpLink } from '@apollo/client';
import { GraphQLWsLink } from '@apollo/client/link/subscriptions';
import { createClient } from 'graphql-ws';
import { getMainDefinition } from '@apollo/client/utilities';
import { gql } from '@apollo/client';

// HTTP 링크 설정
const httpLink = new HttpLink({
  uri: 'https://development.vpms.io/graphql',
  headers: {
    authorization: `Bearer ${accessToken}`,
  },
});

// WebSocket 링크 설정
const wsLink = new GraphQLWsLink(
  createClient({
    url: 'wss://development.vpms.io/graphql',
    connectionParams: {
      authorization: `Bearer ${accessToken}`,
    },
  })
);

// HTTP와 WebSocket 링크 분기
const splitLink = split(
  ({ query }) => {
    const definition = getMainDefinition(query);
    return (
      definition.kind === 'OperationDefinition' &&
      definition.operation === 'subscription'
    );
  },
  wsLink,
  httpLink
);

const client = new ApolloClient({
  link: splitLink,
  cache: new InMemoryCache(),
});

// 숙박시설 알림 구독
const ACCOMMODATION_SUBSCRIPTION = gql`
  subscription OnAccommodationNotification($accommodationId: ID!) {
    accommodationSubscription(accommodationId: $accommodationId) {
      id
      type
      priority
      dotType
      description
      template
      objective
      relative
      roomTypeId
      createdAt
      data
    }
  }
`;

// 구독 시작
const subscription = client.subscribe({
  query: ACCOMMODATION_SUBSCRIPTION,
  variables: {
    accommodationId: '01HQKS9V8X',
  },
}).subscribe({
  next: ({ data }) => {
    console.log('새 알림:', data.accommodationSubscription);

    // 우선순위에 따른 처리
    const notification = data.accommodationSubscription;
    if (notification.priority >= 3) {
      // 긴급 알림 (빨강)
      showUrgentAlert(notification);
    } else if (notification.priority === 2) {
      // 경고 알림 (노랑)
      showWarning(notification);
    } else {
      // 일반 알림 (초록)
      showInfo(notification);
    }
  },
  error: (error) => {
    console.error('구독 에러:', error);
  },
});

// 데이터 구독 (객실 상태, 인증 요청 등)
const DATA_SUBSCRIPTION = gql`
  subscription OnDataEvent($accommodationId: ID!) {
    dataSubscription(accommodationId: $accommodationId) {
      type
      data
    }
  }
`;

client.subscribe({
  query: DATA_SUBSCRIPTION,
  variables: {
    accommodationId: '01HQKS9V8X',
  },
}).subscribe({
  next: ({ data }) => {
    const event = data.dataSubscription;
    const eventData = JSON.parse(event.data);

    switch (event.type) {
      case 'RoomState':
        updateRoomState(eventData.id, eventData.stateUpdatedAt);
        break;
      case 'AuthRequest':
        handleAuthRequest(eventData);
        break;
      default:
        console.log('알 수 없는 이벤트:', event.type);
    }
  },
});