iot-device-management-frontend/apps/web-antd/src/views/system/notify/my/index.vue

<script lang="ts" setup>
// TODO @puhui999：需要看看，vben 哪里改下哈。一个是 vben 右上角的站内信、一个是点击查看所有消息，应该跳转到这里。
import type {
  OnActionClickParams,
  VxeTableGridOptions,
} from '#/adapter/vxe-table';
import type { SystemNotifyMessageApi } from '#/api/system/notify/message';

import { Page, useVbenModal } from '@vben/common-ui';
import { MdiCheckboxMarkedCircleOutline } from '@vben/icons';
import { Button, message } from 'ant-design-vue';
import Detail from './modules/detail.vue';
import { DocAlert } from '#/components/doc-alert';

import { useVbenVxeGrid } from '#/adapter/vxe-table';
import {
  getMyNotifyMessagePage,
  updateAllNotifyMessageRead,
  updateNotifyMessageRead,
} from '#/api/system/notify/message';

import { useGridColumns, useGridFormSchema } from './data';

const [DetailModal, detailModalApi] = useVbenModal({
  connectedComponent: Detail,
  destroyOnClose: true,
});

/** 刷新表格 */
function onRefresh() {
  gridApi.query();
}

/** 查看站内信详情 */
function onDetail(row: SystemNotifyMessageApi.SystemNotifyMessage) {
  detailModalApi.setData(row).open();
}

/** 标记一条站内信已读 */
async function onRead(row: SystemNotifyMessageApi.SystemNotifyMessage) {
  message.loading({
    content: '正在标记已读...',
    duration: 0,
    key: 'action_process_msg',
  });
  // 执行标记已读操作
  await updateNotifyMessageRead([row.id]);
  // 提示成功
  message.success({
    content: '标记已读成功',
    key: 'action_process_msg',
  });
  onRefresh();

  // 打开详情
  onDetail(row);
}

/** 标记选中的站内信为已读 */
async function onMarkRead() {
  const rows = gridApi.grid.getCheckboxRecords();
  if (!rows || rows.length === 0) {
    message.warning({
      content: '请选择需要标记的站内信',
      key: 'action_process_msg',
    });
    return;
  }

  const ids = rows.map((row: SystemNotifyMessageApi.SystemNotifyMessage) => row.id);
  message.loading({
    content: '正在标记已读...',
    duration: 0,
    key: 'action_process_msg',
  });
  // 执行标记已读操作
  await updateNotifyMessageRead(ids);
  // 提示成功
  message.success({
    content: '标记已读成功',
    key: 'action_process_msg',
  });
  await gridApi.grid.setAllCheckboxRow(false);
  onRefresh();
}

/** 标记所有站内信为已读 */
async function onMarkAllRead() {
  message.loading({
    content: '正在标记全部已读...',
    duration: 0,
    key: 'action_process_msg',
  });
  // 执行标记已读操作
  await updateAllNotifyMessageRead();
  // 提示成功
  message.success({
    content: '全部标记已读成功',
    key: 'action_process_msg',
  });
  await gridApi.grid.setAllCheckboxRow(false);
  onRefresh();
}

/** 表格操作按钮的回调函数 */
function onActionClick({
  code,
  row,
}: OnActionClickParams<SystemNotifyMessageApi.SystemNotifyMessage>) {
  switch (code) {
    case 'detail': {
      onDetail(row);
      break;
    }
    case 'read': {
      onRead(row);
      break;
    }
  }
}

const [Grid, gridApi] = useVbenVxeGrid({
  formOptions: {
    schema: useGridFormSchema(),
  },
  gridOptions: {
    columns: useGridColumns(onActionClick),
    height: 'auto',
    keepSource: true,
    proxyConfig: {
      ajax: {
        query: async ({ page }, formValues) => {
          return await getMyNotifyMessagePage({
            pageNo: page.currentPage,
            pageSize: page.pageSize,
            ...formValues,
          });
        },
      },
    },
    rowConfig: {
      keyField: 'id',
    },
    toolbarConfig: {
      refresh: { code: 'query' },
      search: true,
    },
    checkboxConfig: {
      checkMethod: (params: { row: SystemNotifyMessageApi.SystemNotifyMessage }) => !params.row.readStatus,
      highlight: true,
    },
  } as VxeTableGridOptions<SystemNotifyMessageApi.SystemNotifyMessage>,
});
</script>
<template>
  <Page auto-content-height>
    <DocAlert title="站内信配置" url="https://doc.iocoder.cn/notify/" />

    <DetailModal @success="onRefresh" />
    <Grid table-title="我的站内信">
      <template #toolbar-tools>
        <Button type="primary" @click="onMarkRead">
          <MdiCheckboxMarkedCircleOutline />
          标记已读
        </Button>
        <Button type="primary" class="ml-2" @click="onMarkAllRead">
          <MdiCheckboxMarkedCircleOutline />
          全部已读
        </Button>
      </template>
    </Grid>
  </Page>
</template>