miniprogram-setup/src/useAutoUpdate.ts

小程序支持 setup 选项

import { getUpdateData, resetUpdateData, setUpdateData } from './updateData'

export const refresh = (updateData: (() => void) & { isRefreshing?: boolean }) => {
  if (updateData.isRefreshing)
    return

  updateData.isRefreshing = true

  // 先清空 updateData 再调用是因为要
  // 防止 nextTick 再次触发一次更新。
  resetUpdateData()
  nextTick(() => {
    updateData()
    updateData.isRefreshing = false
  })
  setUpdateData(updateData)
}

export const useAutoUpdate = (handler?: ((...params: any) => any) | null) => {
  if (handler == null)
    return

  const updateData = getUpdateData()
  return function (this: any, ...args: any) {
    if (updateData != null) {
      refresh(updateData)
      // 当使用 await 的时候，其之后的代码
      // 运行的时候 updateData 的值可能已经
      // 被改变，所以在此重新设置回相应的值。
      setUpdateData(updateData)
    }

    // 确保 this 指向不变
    return handler.call(this, ...args)
  }
}

export async function nextTick(handler?: () => void) {
  return await Promise.resolve().then(handler)
}