risk.js 2.21 KB
import * as services from '@/services/risk';
import { message } from 'antd';
import moment from 'moment';

/* SerachParams */
const staticSearch = {
  current: 1,
  size: 10,
  status: 'ALL',
  riskType: 'ALL',
  fileFormat: 'ALL',
  riskHost: undefined,
  startTime: moment(new Date())
    .add(-7, 'days')
    .startOf('day'),
  endTime: moment(new Date())
    .add(0, 'days')
    .endOf('day'),
};

export default {
  namespace: 'Risk',
  state: {
    dataSearch: {
      ...staticSearch,
    },
    dataRisk: {
      records: [],
      current: 1,
      size: 10,
      total: 0,
      totalPage: 0,
    },
  },
  effects: {
    /* 获取风险列表 */
    *getRiskList({ payload }, { call, put, select }) {
      const {
        dataSearch,
        dataSearch: { status, riskType, fileFormat, startTime, endTime },
      } = yield select(state => state.Risk);
      try {
        const res = yield call(services.getRiskList, {
          current: 1,
          size: 10,
          ...dataSearch,
          ...payload,
          startTime: moment(startTime).format('YYYY-MM-DD HH:mm:ss'),
          endTime: moment(endTime).format('YYYY-MM-DD HH:mm:ss'),
          status: status === 'ALL' ? '' : status,
          riskType: riskType === 'ALL' ? '' : riskType,
          fileFormat: fileFormat === 'ALL' ? '' : fileFormat,
        });
        if (res.code === 0) {
          res.data.records = res.data.records || [];
          yield put({
            type: 'changeState',
            payload: {
              dataRisk: res.data,
            },
          });
        }
      } catch (err) {
        console.error(err);
      }
    },
    /* 风险导入 */
    *importRisk({ payload }, { call, put, select }) {
      try {
        const res = yield call(services.importRisk, payload);
        if (res.code === 0) {
          message.success('导入成功');
          yield put({ type: 'getRiskList' });
        }
      } catch (err) {
        console.error(err);
      }
    },
  },
  reducers: {
    changeState(state, { payload }) {
      return {
        ...state,
        ...payload,
      };
    },
    resetSearch(state, { payload }) {
      return {
        ...state,
        dataSearch: {
          ...staticSearch,
        },
      };
    },
  },
};