package com.whyc.controller;

import com.whyc.dto.Response;
import com.whyc.dto.StandardFeatureCurve;
import com.whyc.service.StandardFeatureCurveService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.sql.ResultSet;
import java.util.List;

@Api(tags = "标准特性曲线")
@RestController
@RequestMapping("standardFeatureCurve")
public class StandardFeatureCurveController {

    @Autowired
    private StandardFeatureCurveService service;

    @ApiOperation("查询指定VA/C的数据")
    @GetMapping("list")
    public Response<List<StandardFeatureCurve>> getList(@RequestParam String va,@RequestParam String c){
        List<StandardFeatureCurve> list = service.getList(va,c);
        if(list == null){
            return new Response().set(1,false);
        }
        return new Response().setII(1,true,list,null);
    }

    @ApiOperation("数据导入")
    @PostMapping("excelImport")
    public Response excelImport(@RequestPart MultipartFile file) throws IOException, InvalidFormatException {
        Response<Object> response = new Response<>();
        String name=file.getOriginalFilename();
        if(!name.endsWith(".xls") && !name.endsWith(".xlsx")){
            response.set(1,false,"文件类型错误");
        }else{
            InputStream is=file.getInputStream();
            service.excelImport(is);
            if(is!=null){
                try {
                    is.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            response.set(1,true,"数据导入完成");
        }
        return response;
    }
    @ApiOperation("测试mybatisDb")
    @GetMapping("testTableDb")
    public Response testTableDb(){
        return service.testTableDb();
    }
}