微信小程序websocket实现实时聊天代码

小程序中使用 WebSocket 实现实时聊天的步骤如下：

1. 在小程序 app.json 文件中添加网络权限：

{
  "permission": {
    "scope.userLocation": {
      "desc": "获取用户地理位置信息"
    },
    "scope.record": {
      "desc": "录音功能，用于发送语音消息"
    },
    "scope.writePhotosAlbum": {
      "desc": "保存到相册功能，用于保存聊天图片"
    },
    "scope.userInfo": {
      "desc": "获取用户信息，用于显示聊天用户头像和昵称"
    },
    "scope.websocket": {
      "desc": "使用 WebSocket，用于实现实时聊天"
    }
  }
}

2. 在小程序页面中创建 WebSocket 连接：

// 建立连接
let socketTask = wx.connectSocket({
  url: 'wss://your.websocket.url',
  success() {
    console.log('WebSocket 连接成功')
  },
  fail() {
    console.log('WebSocket 连接失败')
  }
})

// 监听 WebSocket 连接状态变化
socketTask.onOpen(() => {
  console.log('WebSocket 已连接')
})
socketTask.onError(() => {
  console.log('WebSocket 连接错误')
})
socketTask.onClose(() => {
  console.log('WebSocket 已关闭')
})

3. 发送和接收消息：

// 发送消息
socketTask.send({
  data: 'Hello, WebSocket!',
  success() {
    console.log('消息发送成功')
  },
  fail() {
    console.log('消息发送失败')
  }
})

// 接收消息
socketTask.onMessage((res) => {
  console.log('接收到消息：', res.data)
})

完整的实时聊天代码示例：

Page({
  data: {
    messages: [] // 聊天消息列表
  },

  onLoad: function() {
    // 建立连接
    let socketTask = wx.connectSocket({
      url: 'wss://your.websocket.url',
      success() {
        console.log('WebSocket 连接成功')
      },
      fail() {
        console.log('WebSocket 连接失败')
      }
    })

    // 监听 WebSocket 连接状态变化
    socketTask.onOpen(() => {
      console.log('WebSocket 已连接')
    })
    socketTask.onError(() => {
      console.log('WebSocket 连接错误')
    })
    socketTask.onClose(() => {
      console.log('WebSocket 已关闭')
    })

    // 接收消息
    socketTask.onMessage((res) => {
      console.log('接收到消息：', res.data)
      let message = JSON.parse(res.data)
      let messages = this.data.messages
      messages.push(message)
      this.setData({
        messages: messages
      })
    })
  },

  // 发送消息
  sendMessage: function(event) {
    let message = event.detail.value
    if (message) {
      let socketTask = wx.connectSocket({
        url: 'wss://your.websocket.url'
      })
      socketTask.send({
        data: message,
        success() {
          console.log('消息发送成功')
        },
        fail() {
          console.log('消息发送失败')
        }
      })
      this.setData({
        inputMessage: ''
      })
    }
  }
})