WebSocket

API Definition

api.ts

export type API = {
  add: (a: number, b: number, callback?: (sum: number) => void) => Promise<number>
}

export const apiMethods: API = {
  add: async (a, b, callback) => {
    callback?.(a + b)
    return a + b
  }
}

Server

server.ts

import { expose } from "kkrpc"
import { webSocketTransport } from "kkrpc/ws"
import { WebSocketServer } from "ws"
import { apiMethods } from "./api"

const wss = new WebSocketServer({ port: 3000 })

wss.on("connection", (ws) => {
  expose(apiMethods, webSocketTransport(ws))
})

Client

client.ts

import { wrap } from "kkrpc"
import { webSocketClientTransport } from "kkrpc/ws"
import type { API } from "./api"

const api = wrap<API>(webSocketClientTransport("ws://localhost:3000"))

const sum = await api.add(5, 3)
await api.add(10, 20, (value) => console.log(value))