xufenglei
2018-03-07 52303c8868a7d63238e5ac579c85721306e51a40
src/main/java/com/moral/service/impl/DeviceServiceImpl.java
@@ -1,23 +1,31 @@
package com.moral.service.impl;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import com.alibaba.fastjson.TypeReference;
import com.github.pagehelper.Page;
import com.moral.common.bean.Constants;
import com.moral.common.bean.PageBean;
import com.moral.common.bean.PageResult;
import com.moral.common.util.ExampleUtil;
import com.moral.common.util.RedisUtils;
import com.moral.mapper.MonitorPointMapper;
import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang3.BooleanUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.ObjectUtils;
import com.github.pagehelper.PageHelper;
import com.moral.common.exception.BusinessException;
import com.moral.common.util.ValidateUtil;
import com.moral.entity.Device;
import com.moral.mapper.DeviceMapper;
import com.moral.service.AccountService;
import com.moral.service.DeviceService;
import tk.mybatis.mapper.entity.Example;
@Service
@@ -25,12 +33,16 @@
   @Resource
   private DeviceMapper deviceMapper;
    @Resource
    private MonitorPointMapper monitorPointMapper;
   @Resource
   private AccountService accountService;
    private Class ENTITY_CLASS = Device.class;
   @Resource
   RedisUtils redisUtils;
   @Override
   public Map<String, Object> getDeviceStatesByAccount(Map<String, Object> parameters) {
      ValidateUtil.notNull(parameters.get("accountId"), "param.is.null");
      Map<String, Object> result = new HashMap<String, Object>();
      accountService.setOrgIdsByAccount(parameters);
      List<Map<String, Object>> list = deviceMapper.getDeviceStatesByAccount(parameters);
@@ -57,38 +69,253 @@
   }
   
   @Override
   public List<Map<String, Object>> getSensorsByDevice(String mac) {
      return deviceMapper.getSensorsByDevice(mac);
   }
   @Override
   @Transactional
   public void saveOrUpdateDevice(Device device) {
      ValidateUtil.notNull(device, "参数不可为空");
      ValidateUtil.notEmpty(device.getMac(), "参数不可为空");
      Example example = new Example(Device.class);
      example.or().andEqualTo("mac",device.getMac());
      List<Device> devices = deviceMapper.selectByExample(example);
      ValidateUtil.notNull(device, "param.is.null");
      ValidateUtil.notEmpty(device.getMac(), "param.is.null");
      Device queryDevice = new Device();
      queryDevice.setMac(device.getMac());
      queryDevice = deviceMapper.selectOne(queryDevice);
      Date operateDate = new Date();
      device.setInstallTime(operateDate);
      if (ObjectUtils.isEmpty(devices)) {
      if (ObjectUtils.isEmpty(queryDevice)) {
         device.setCreateTime(operateDate);
         device.setState("4");
         device.setIsDelete(Constants.IS_DELETE_FALSE);
         deviceMapper.insertSelective(device);
      }else if (devices.size() > 1) {
         throw new BusinessException("设备信息异常,联系管理员!");
      }else {
         device.setId(devices.get(0).getId());
         device.setId(queryDevice.getId());
         deviceMapper.updateByPrimaryKeySelective(device);
      }
   }
   @Override
   public List<Device> getInstallDevicesByOperateUser(Integer uid, Integer pageIndex, Integer pageSize) {
      Example example = new Example(Device.class);
      example.or().andEqualTo("operateUserId",uid);
      ValidateUtil.notNull(uid, "param.is.null");
      Device device = new Device();
      device.setOperateUserId(uid);
      PageHelper.startPage(pageIndex, pageSize);
      List<Device> devices = deviceMapper.selectByExample(example);
      List<Device> devices = deviceMapper.select(device);
      return devices;
   }
   @Override
   public Device getDeviceByMac(String mac) {
      ValidateUtil.notEmpty(mac, "param.is.null");
      Device device = new Device();
      device.setMac(mac);
      device = deviceMapper.selectOne(device);
      return device;
   }
   /**
    *
    * @param map
    * map里 包括 组织id和4个坐标点
    * @return
    */
   @Override
   public List<Device> query(Map<String, Object> map) {
      List <Device> list = deviceMapper.selectByMap(map);
      loadDeviceState(list);
      return list;
   }
   /**
    *  根据组织id和设备名称 分页查询设备
    * @param orgId
    * @param deviceName
    * @param pageSize
    * @param pageNo
    * @return
    */
   @Override
   public PageResult query(Integer orgId, String deviceName, Integer pageSize, Integer pageNo) {
      if(!ObjectUtils.isEmpty(pageSize)&&!ObjectUtils.isEmpty(pageNo)){
         PageHelper.startPage(pageNo,pageSize);
      }
      List<Device> list = deviceMapper.selectByOrgIdAndDevName(orgId,deviceName);
      //从redis里取状态
      loadDeviceState(list);
      if(list instanceof Page){
         return new PageResult(((Page) list).getTotal(),list);
      }
      return new PageResult(null,list);
   }
   /**
    * 根据组织id和监控点id 分页查询设备
    * @param orgId
    * @param mpId
    * @param pageSize
    * @param pageNo
    * @return
    */
   @Override
   public PageResult query(Integer orgId, Integer mpId, Integer pageSize, Integer pageNo) {
      if(!ObjectUtils.isEmpty(pageSize)&&!ObjectUtils.isEmpty(pageNo)){
         PageHelper.startPage(pageNo,pageSize);
      }
      List<Device> list = deviceMapper.selectByOrgIdAndMpId(orgId,mpId);
      //从redis里取状态
      loadDeviceState(list);
      if(list instanceof Page){
         return new PageResult(((Page) list).getTotal(),list);
      }
      return new PageResult(null,list);
   }
   private void loadDeviceState(List<Device> list){
      //从redis里取状态
      list.stream().map( device -> {
         String mac = device.getMac();
         if(!StringUtils.isBlank(mac)){
            String state = getSateFromRedis(device.getMonitorPointId(),mac.toLowerCase());
            device.setState(state);
         }else{
            device.setState(Constants.DEVICE_STATE_OFFLINE);
         }
         return device;
      }).count();
   }
   private String getSateFromRedis(Integer mpId,String mac){
      Map<String,String> stateMap  = getStateMapFromRedis(mpId,mac);
      String state = null;
      if(stateMap != null){
         state  = stateMap.get("state");
      }
      state = state == null ?Constants.DEVICE_STATE_OFFLINE:state;
      return  state;
   }
   public Map<String,String> getStateMapFromRedis(Integer mpId,String mac){
      StringBuilder key = new StringBuilder();
      //拼凑key
      key.append("state_").append(mpId).append("_").append(mac);
      return   redisUtils.get(key.toString(),new TypeReference<Map<String,String>>(){});
   }
   private Device getDeviceWithOrgIdsByMac(String mac) {
      String key = "device_"+mac;
      Device device = redisUtils.get(key,Device.class);
      if(device==null) {
         device = deviceMapper.selectWithOrgIdsByMac(mac);
         if(device!=null){
            redisUtils.set(key,device);
         }
      }
      return device;
   }
   /*
     刷新 redis 设备的信息
    */
   private void refreshDeviceInRedis(Device device){
        if(!StringUtils.isBlank(device.getMac())){
             Device simpleDevice = new Device();
             simpleDevice.setId(device.getId());
             simpleDevice.setDeviceVersion(device.getDeviceVersion());
             simpleDevice.setMac(device.getMac());
             simpleDevice.setMonitorPointId(device.getMonitorPointId());
             if(device.getMonitorPointId()!=null){
                 List<Integer> orgIds = monitorPointMapper.selectOrganizationIds(device.getMonitorPointId());
                 simpleDevice.setOrganizationIds(orgIds);
             }
             String key = "device_"+device.getMac();
             redisUtils.set(key,simpleDevice);
         }
   }
   @Override
   public PageBean queryByPageBean(PageBean pageBean) {
      Example example = ExampleUtil.generateExample(ENTITY_CLASS,pageBean);
      List<Example.Criteria> criteriaList = example.getOredCriteria();
      if(criteriaList!=null&&criteriaList.size()>0){
         for(Example.Criteria cri : criteriaList){
            cri.andNotEqualTo("isDelete", Constants.IS_DELETE_TRUE);
         }
      }else {
         example.or().andNotEqualTo("isDelete",Constants.IS_DELETE_TRUE);
      }
      if(pageBean.getPageSize()>0){
         PageHelper.startPage(pageBean.getPageIndex(),pageBean.getPageSize());
      }
      List<Device> organizationList = deviceMapper.selectWithRelationData(example);
      return new PageBean(organizationList);
   }
   @Override
   public void deleteByIds(Integer[] ids) {
      Device device = new Device();
      device.setIsDelete(Constants.IS_DELETE_TRUE);
      if(ids!=null&&ids.length>0){
         if(ids.length==1){
            device.setId(ids[0]);
            deviceMapper.updateByPrimaryKeySelective(device);
         }else{
            Example example = new Example(ENTITY_CLASS);
            example.or().andIn("id", Arrays.asList(ids));
            deviceMapper.updateByExampleSelective(device,example);
         }
      }
   }
   @Override
   public void addOrModify(Device device){
      try{
         //mac 转小写
         if(StringUtils.isBlank(device.getMac())){
            device.setMac(device.getMac().toLowerCase());
         }
         if(device.getId()==null){
            device.setIsDelete(Constants.IS_DELETE_FALSE);
            deviceMapper.insertSelective(device);
         }else{
            deviceMapper.updateByPrimaryKeySelective(device);
            //刷新redis里设备信息
            refreshDeviceInRedis(device);
         }
      }
      catch (Exception ex){
         throw  ex;
      }
   }
   @Override
   public List<Device> getDevicesByMonitorPointId(Integer monitorPointId) {
      Device device = new Device();
      device.setMonitorPointId(monitorPointId);
      device.setIsDelete(Constants.IS_DELETE_FALSE);
      return deviceMapper.select(device);
   }
   /**
    *  返回map{mac:,state:}
    * @param macList
    * @return
    */
   @Override
   public List<Map<String,String>> queryDevicesState(List<String> macList,Boolean withData) {
      List<Map<String,String>> list = macList.stream().map(mac->{
         Map<String,String> resultMap = new HashMap<>();
         if(!StringUtils.isBlank(mac)){
            mac = mac.toLowerCase();
            Device device = getDeviceWithOrgIdsByMac(mac);
            Map<String,String> stateMap = getStateMapFromRedis(device.getMonitorPointId(),mac);
            if(!MapUtils.isEmpty(stateMap)){
               resultMap.putAll(stateMap);
            }else{
               resultMap.put("state",Constants.DEVICE_STATE_OFFLINE);
               resultMap.put("mac",mac);
            }
            //添加data
            if(BooleanUtils.isTrue(withData)){
               String dataKey = "data_"+mac;
               Map<String,String> dataMap = redisUtils.get(dataKey,new TypeReference<Map<String,String>>(){});
               if(!MapUtils.isEmpty(dataMap)){
                  resultMap.putAll(dataMap);
               }
            }
         }
         return  resultMap;
      }).collect(Collectors.toList());
      return list;
   }
}