调整试验数据返回数据结构

whyczh

2021-03-31 8e560d924cd1d6918b969b6b0978b613da0bdfb1

调整试验数据返回数据结构

 src/main/java/com/whyc/controller/TestDataController.java

@@ -1,8 +1,11 @@
package com.whyc.controller;

import com.google.gson.JsonObject;
import com.whyc.dto.Response;
import com.whyc.dto.TestDataDTO;
import com.whyc.pojo.MotorState;
import com.whyc.pojo.TestPlan;
import com.whyc.service.DeviceInfService;
import com.whyc.service.MotorStateService;
import com.whyc.service.TestPlanService;
import io.swagger.annotations.Api;
@@ -29,6 +32,8 @@
    private TestPlanService testPlanService;
    @Resource
    private MotorStateService motorStateService;
    @Autowired
    private DeviceInfService deviceInfService;

    @GetMapping()
    @ApiOperation("获取历史数据")
@@ -42,11 +47,16 @@
        String startTime = dateFormat.format(testPlan.getStartTime());
        String endTime = dateFormat.format(testPlan.getEndTime());
        //通过设备和开始结束时间来获取对应的数据
        Map<String,List<MotorState>> map = new HashMap<>();
        List<TestDataDTO> list = new ArrayList<TestDataDTO>();
        for (String deviceId: devIds) {
            List<MotorState> historyData =  motorStateService.getHistoryData(deviceId,startTime,endTime);
            map.put(deviceId,historyData);
            String deviceName = deviceInfService.getOneByDeviceId(Integer.valueOf(deviceId)).getDeviceName();
            TestDataDTO testDataDTO = new TestDataDTO();
            testDataDTO.setDeviceId(deviceId);
            testDataDTO.setDeviceName(deviceName);
            testDataDTO.setData(historyData);
            list.add(testDataDTO);
        }
        return new Response().set(1,map);
        return new Response().set(1,list);
    }
}

 src/main/java/com/whyc/dto/TestDataDTO.java

New file
@@ -0,0 +1,36 @@
package com.whyc.dto;

import com.whyc.pojo.MotorState;
import org.apache.ibatis.type.Alias;

import java.util.List;
@Alias("TestDataDTO")
public class TestDataDTO {
    private String deviceId;
    private String deviceName;
    private List<MotorState> data;

    public String getDeviceId() {
        return deviceId;
    }

    public void setDeviceId(String deviceId) {
        this.deviceId = deviceId;
    }

    public String getDeviceName() {
        return deviceName;
    }

    public void setDeviceName(String deviceName) {
        this.deviceName = deviceName;
    }

    public List<MotorState> getData() {
        return data;
    }

    public void setData(List<MotorState> data) {
        this.data = data;
    }
}