API网关对于WebSocket连接的支持取决于具体的实现和配置。一般来说，API网关可以支持大量的WebSocket连接，但具体的限制取决于服务器和网络资源的可用性。

以下是一个使用Node.js和Express框架实现WebSocket连接的示例代码：

const express = require('express');
const app = express();
const http = require('http').Server(app);
const io = require('socket.io')(http);

io.on('connection', (socket) => {
  console.log('A client connected');

  socket.on('disconnect', () => {
    console.log('A client disconnected');
  });

  socket.on('message', (data) => {
    console.log(`Received message from client: ${data}`);
    // 处理接收到的消息
    // ...
    // 发送回复消息给客户端
    socket.emit('reply', 'Hello from server');
  });
});

http.listen(3000, () => {
  console.log('Server started on port 3000');
});

在上面的示例中，使用Express创建了一个HTTP服务器并使用Socket.IO库处理WebSocket连接。当有客户端连接到服务器时，会触发connection事件。在connection事件处理程序中，可以处理客户端发送的消息，并通过emit方法将回复消息发送回客户端。

请注意，这只是一个简单的示例，实际情况中可能需要根据具体的业务需求进行配置和扩展。在生产环境中，可能需要考虑使用负载均衡和扩展机制来支持大量的WebSocket连接。