Go로 PostgreSQL 프록시 만들기 (1) - PG Wire Protocol 이해

[4바이트 길이][프로토콜 버전][파라미터들...]

[1바이트 타입][4바이트 길이][페이로드...]

'Q' | 길이(4바이트) | "SELECT 1\0"

Client                          Server
  │                               │
  │──── StartupMessage ──────────▶│  (버전 3.0 + user/database)
  │                               │
  │◀──── AuthenticationOk ────────│  (R + 0x00000000)
  │◀──── ParameterStatus ────────│  (server_version 등)
  │◀──── BackendKeyData ─────────│  (PID + secret key)
  │◀──── ReadyForQuery ──────────│  (Z + 'I')
  │                               │
  │──── Query("SELECT 1") ──────▶│
  │                               │
  │◀──── RowDescription ─────────│
  │◀──── DataRow ────────────────│
  │◀──── CommandComplete ────────│
  │◀──── ReadyForQuery ──────────│

func ReadMessage(r io.Reader) (*Message, error) {
    // 1바이트 타입 읽기
    var typeBuf [1]byte
    if _, err := io.ReadFull(r, typeBuf[:]); err != nil {
        return nil, err
    }

    // 4바이트 길이 읽기 (Big Endian)
    var length int32
    if err := binary.Read(r, binary.BigEndian, &length); err != nil {
        return nil, err
    }

    // 페이로드 읽기 (길이 - 4, 길이 필드 자체 제외)
    payload := make([]byte, length-4)
    if _, err := io.ReadFull(r, payload); err != nil {
        return nil, err
    }

    return &Message{Type: typeBuf[0], Payload: payload}, nil
}

func ReadStartupMessage(r io.Reader) (*Message, error) {
    var length int32
    binary.Read(r, binary.BigEndian, &length)

    payload := make([]byte, length-4)
    io.ReadFull(r, payload)

    return &Message{Type: 0, Payload: payload}, nil
}

if code == SSLRequestCode {  // 80877103
    clientConn.Write([]byte{'N'})  // SSL 미지원
    // 이후 실제 startup 메시지를 다시 읽음
    startup, _ = ReadStartupMessage(clientConn)
}

// 1. 클라이언트 → 프록시: startup 메시지 수신
// 2. 프록시 → 백엔드: startup 메시지 전달
// 3. 백엔드 → 프록시 → 클라이언트: ReadyForQuery까지 릴레이

func relayAuth(clientConn, backendConn net.Conn) error {
    for {
        msg, _ := protocol.ReadMessage(backendConn)
        protocol.WriteMessage(clientConn, msg.Type, msg.Payload)

        if msg.Type == MsgReadyForQuery {
            return nil  // 인증 완료!
        }
    }
}

func relayQueries(clientConn, backendConn net.Conn) {
    for {
        msg, _ := protocol.ReadMessage(clientConn)

        if msg.Type == MsgTerminate {
            return
        }

        // 여기서 쿼리 텍스트 추출 → 라우팅/캐싱 가능
        if msg.Type == MsgQuery {
            query := ExtractQueryText(msg.Payload)
            slog.Debug("query", "sql", query)
        }

        // 백엔드로 전달
        protocol.WriteMessage(backendConn, msg.Type, msg.Payload)

        // ReadyForQuery까지 응답 릴레이
        relayUntilReady(clientConn, backendConn)
    }
}