peixue-dev/peidu/uniapp/utils/messagePolling.js

/**
 * 消息轮询工具
 * 用于定期检查未读消息数量并更新TabBar角标
 */
import request from './request'

class MessagePolling {
  constructor() {
    this.timer = null
    this.interval = 30000 // 30秒轮询一次
    this.isPolling = false
    this.tabBarIndex = 3 // 消息中心Tab索引（根据实际情况调整）
    this.tabBarRoutes = [
      'pages/index/index',
      'pages/service/list',
      'pages/booking/quick-booking',
      'pages/order/list',
      'pages/user/index'
    ]
  }

  updateTabBarBadge(totalCount) {
    if (typeof uni?.setTabBarBadge !== 'function' || typeof uni?.removeTabBarBadge !== 'function') {
      return
    }

    if (totalCount > 0) {
      uni.setTabBarBadge({
        index: this.tabBarIndex,
        text: totalCount > 99 ? '99+' : totalCount.toString(),
        fail: (err) => {
          if (err?.errMsg && err.errMsg.indexOf('not TabBar page') !== -1) {
            return
          }
          console.warn('[MessagePolling] setTabBarBadge failed:', err)
        }
      })
    } else {
      uni.removeTabBarBadge({
        index: this.tabBarIndex,
        fail: (err) => {
          if (err?.errMsg && err.errMsg.indexOf('not TabBar page') !== -1) {
            return
          }
          console.warn('[MessagePolling] removeTabBarBadge failed:', err)
        }
      })
    }
  }
  
  /**
   * 开始轮询
   */
  start() {
    if (this.isPolling) {
      console.log('[MessagePolling] 轮询已在运行中')
      return
    }
    
    console.log('[MessagePolling] 开始轮询')
    this.isPolling = true
    
    // 立即执行一次
    this.poll()
    
    // 定时轮询
    this.timer = setInterval(() => {
      this.poll()
    }, this.interval)
  }
  
  /**
   * 停止轮询
   */
  stop() {
    if (this.timer) {
      clearInterval(this.timer)
      this.timer = null
    }
    this.isPolling = false
    console.log('[MessagePolling] 停止轮询')
  }
  
  /**
   * 执行一次轮询
   */
  async poll() {
    try {
      const userInfo = uni.getStorageSync('userInfo')
      const userId = userInfo?.id || userInfo?.userId
      const res = await request.get('/api/message/unread-count', { userId })
      
      if (res.code === 200) {
        const totalCount = typeof res.data === 'number'
          ? res.data
          : (res.data?.total || 0)
        
        console.log('[MessagePolling] 未读消息数:', totalCount)
        
        // 更新TabBar角标
        this.updateTabBarBadge(totalCount)
        
        // 触发全局事件，通知其他页面更新
        uni.$emit('unreadCountUpdate', {
          total: totalCount,
          detail: res.data
        })
      }
    } catch (e) {
      console.error('[MessagePolling] 轮询失败:', e)
      // 轮询失败不影响主流程，只记录日志
    }
  }

  isTabBarPage() {
    try {
      const pages = getCurrentPages()
      const currentPage = pages && pages.length ? pages[pages.length - 1] : null
      const route = currentPage && currentPage.route ? currentPage.route : ''
      return !!route && this.tabBarRoutes.includes(route)
    } catch (e) {
      return false
    }
  }
  
  /**
   * 设置TabBar索引
   */
  setTabBarIndex(index) {
    this.tabBarIndex = index
  }
  
  /**
   * 设置轮询间隔（毫秒）
   */
  setInterval(interval) {
    this.interval = interval
    
    // 如果正在轮询，重启以应用新间隔
    if (this.isPolling) {
      this.stop()
      this.start()
    }
  }
}

// 导出单例
const messagePolling = new MessagePolling()

export function startMessagePolling() {
  messagePolling.start()
}

export function stopMessagePolling() {
  messagePolling.stop()
}

export default messagePolling