xufenglei
2018-03-30 9b33e93fccddea356c3d686684fc58ceb176ff39
src/main/java/com/moral/service/impl/DeviceServiceImpl.java
@@ -1,12 +1,21 @@
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;
@@ -17,18 +26,23 @@
import com.moral.mapper.DeviceMapper;
import com.moral.service.AccountService;
import com.moral.service.DeviceService;
import tk.mybatis.mapper.entity.Example;
@Service
public class DeviceServiceImpl implements DeviceService {
   @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);
@@ -55,11 +69,6 @@
   }
   
   @Override
   public List<Map<String, Object>> getSensorsByDevice(String mac) {
      return deviceMapper.getSensorsByDevice(mac);
   }
   @Override
   @Transactional
   public void saveOrUpdateDevice(Device device) {
      ValidateUtil.notNull(device, "param.is.null");
@@ -72,6 +81,7 @@
      if (ObjectUtils.isEmpty(queryDevice)) {
         device.setCreateTime(operateDate);
         device.setState("4");
         device.setIsDelete(Constants.IS_DELETE_FALSE);
         deviceMapper.insertSelective(device);
      }else {
         device.setId(queryDevice.getId());
@@ -81,6 +91,7 @@
   @Override
   public List<Device> getInstallDevicesByOperateUser(Integer uid, Integer pageIndex, Integer pageSize) {
      ValidateUtil.notNull(uid, "param.is.null");
      Device device = new Device();
      device.setOperateUserId(uid);
      PageHelper.startPage(pageIndex, pageSize);
@@ -97,4 +108,214 @@
      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;
   }
}