请写一段微信小程序与服务端进行socket通信的代码

以下是一个简单的微信小程序与服务端进行socket通信的示例代码：

小程序端代码：

const socketTask = wx.connectSocket({
  url: 'wss://example.com/ws',
  header: {
    'content-type': 'application/json'
  },
  success: function () {
    console.log('WebSocket连接成功')
  }
})

socketTask.onOpen(function () {
  console.log('WebSocket已打开')
  socketTask.send({
    data: 'Hello, WebSocket!'
  })
})

socketTask.onMessage(function (res) {
  console.log('收到服务器内容：' + res.data)
})

socketTask.onClose(function () {
  console.log('WebSocket已关闭')
})

socketTask.onError(function () {
  console.log('WebSocket连接发生错误')
})

服务端代码：

const WebSocket = require('ws')
const wss = new WebSocket.Server({
  port: 8080
})

wss.on('connection', function (ws) {
  console.log('WebSocket已连接')
  ws.on('message', function (message) {
    console.log('收到客户端内容：' + message)
    ws.send('Hello, WebSocket!')
  })
  ws.on('close', function () {
    console.log('WebSocket已关闭')
  })
})

这段代码中，小程序使用`wx.connectSocket()`方法连接到服务端的WebSocket地址，并监听WebSocket事件，当连接成功、收到消息、关闭WebSocket连接、连接发生错误时进行相应的处理。服务端使用`ws`模块创建WebSocket服务器，并监听WebSocket连接事件，当有客户端连接到WebSocket服务器时，也会进行相应的处理，包括收到客户端消息、发送消息和关闭WebSocket连接等。