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);
    }
  },
});

// WebSocket 연결 설정
const wsProtocol = {
  'graphql-transport-ws': true,
};

const socket = new WebSocket(
  'wss://development.vpms.io/graphql',
  'graphql-transport-ws'
);

socket.onopen = () => {
  // 연결 초기화
  socket.send(JSON.stringify({
    type: 'connection_init',
    payload: {
      authorization: `Bearer ${accessToken}`,
    },
  }));
};

socket.onmessage = (event) => {
  const message = JSON.parse(event.data);

  if (message.type === 'connection_ack') {
    // 연결 성공, 구독 시작
    socket.send(JSON.stringify({
      id: '1',
      type: 'subscribe',
      payload: {
        query: `
          subscription OnAccommodationNotification($accommodationId: ID!) {
            accommodationSubscription(accommodationId: $accommodationId) {
              id
              type
              priority
              dotType
              description
              template
              createdAt
            }
          }
        `,
        variables: {
          accommodationId: '01HQKS9V8X',
        },
      },
    }));
  } else if (message.type === 'next') {
    // 새 알림 수신
    const notification = message.payload.data.accommodationSubscription;
    console.log('새 알림:', notification);
    displayNotification(notification);
  } else if (message.type === 'error') {
    console.error('구독 에러:', message.payload);
  }
};

socket.onerror = (error) => {
  console.error('WebSocket 에러:', error);
};

socket.onclose = () => {
  console.log('WebSocket 연결 종료');
  // 재연결 로직
  setTimeout(reconnect, 5000);
};