console/web/src/pages/System/Cluster/models/cluster.js

import {createClusterConfig, searchClusterConfig, updateClusterConfig,deleteClusterConfig,
   getClusterStatus, tryConnect} from "@/services/cluster";
import {message} from "antd";
import {formatESSearchResult} from '@/lib/elasticsearch/util';

export default {
  namespace: 'clusterConfig',
  state: {
    editMode: '',
    editValue: {},
  },
  effects:{
    *fetchClusterStatus({payload}, {call, put}){
      let res = yield call(getClusterStatus, payload);
      if(res.error){
        message.error(res.error)
        return false;
      }
      yield put({
        type: 'saveData',
        payload: {
          clusterStatus: res
        }
      });
      return res;
    },
    *fetchClusterList({payload}, {call, put, select}){
      let res = yield call(searchClusterConfig, payload);
      if(res.error){
        message.error(res.error)
        return false;
      }
      res = formatESSearchResult(res)
      const {clusterStatus} = yield select(state => state.clusterConfig);
      // for(let item of res.data){
      //   item.status= clusterStatus[item.id]
      // }
      yield put({
        type: 'saveData',
        payload: res
      })
    },
    *addCluster({payload}, {call, put, select}) {
      let res = yield call(createClusterConfig, payload)
      if(res.error){
        message.error(res.error)
        return false;
      }
      let {data, total} = yield select(state => state.clusterConfig);
      if(!data){
        return
      }
      data.unshift({
        ...res._source,
        id: res._id,
      });
      yield put({
        type: 'saveData',
        payload: {
          data,
          total: {
            ...total,
            value: total.value + 1
          },
        }
      })
      yield put({
        type: 'global/addCluster',
        payload: {
          id: res._id,
          name: res._source.name,
        }
      })
      return res;
    },
    *updateCluster({payload}, {call, put, select}) {
      let res = yield call(updateClusterConfig, payload)
      if(res.error){
        message.error(res.error)
        return false;
      }
      let {data} = yield select(state => state.clusterConfig);
      let idx = data.findIndex((item)=>{
        return item.id === res._id;
      });

      let originalEnabled = data[idx].enabled;
      data[idx] = {
        ...data[idx],
        ...res._source
      };
      yield put({
        type: 'saveData',
        payload: {
          data
        }
      })
      //handle global cluster logic
      if(originalEnabled !== res._source.enabled){
        if(res._source.enabled === true) {
          yield put({
            type: 'global/addCluster',
            payload: {
              id: res._id,
              name: res._source.name,
            }
          })
        }else{
          yield put({
            type: 'global/removeCluster',
            payload: {
              id: res._id,
            }
          })
        }
      }else{
        yield put({
          type: 'global/updateCluster',
          payload: {
            id: res._id,
            name: res._source.name,
          }
        })
      }

      return res;
    },
    *deleteCluster({payload}, {call, put, select}) {
      let res = yield call(deleteClusterConfig, payload)
      if(res.error){
        message.error(res.error)
        return false;
      }
      let {data, total} = yield select(state => state.clusterConfig);
      data = data.filter((item)=>{
        return item.id !== payload.id;
      })
      yield put({
        type: 'saveData',
        payload: {
          data,
          total: {
            ...total,
            value: total.value + 1
          }
        }
      })
      yield put({
        type: 'global/removeCluster',
        payload: {
          id: payload.id
        }
      })
      return res;
    },
    *doTryConnect({payload}, {call, put, select}) {
      let res = yield call(tryConnect, payload)
      if(res.error){
        message.error(res.error)
        return false;
      }

      yield put({
        type: 'saveData',
        payload: {
          tempClusterInfo: res,
        }
      })
      return res;
    }
  },
  reducers:{
    saveData(state, {payload}){
      return {
        ...state,
        ...payload,
      }
    }
  }
}