tld-consumer-9560/src/main/java/com/tld/controller/UserMaterialController.java

package com.tld.controller;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.tld.excel.ExcelUtils;
import com.tld.model.UserMaterial;
import com.tld.service.UserMaterialService;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 用户绑定物料
 */
@RestController
@RequestMapping("userMaterial")
@RequiredArgsConstructor
public class UserMaterialController {

    private final UserMaterialService userMaterialService;

    /**
     * 查询用户绑定物料信息
     * @param userMaterial 参数
     * @return 返回结果
     */
    @GetMapping("getUserMaterial")
    public Map<String, Object> getUserMaterial(UserMaterial userMaterial){
        return userMaterialService.getUserMaterial(userMaterial);
    }

    /**
     * 新增用户绑定物料信息
     * @param userMaterial 参数
     * @return 返回结果
     */
    @PostMapping("addUserMaterial")
    public Map<String, Object> addUserMaterial(@RequestBody UserMaterial userMaterial, HttpServletRequest request){
        return userMaterialService.addUserMaterial(userMaterial, request);
    }

    /**
     * 修改用户绑定物料信息
     * @param userMaterial 参数
     * @return 返回结果
     */
    @PostMapping("updateUserMaterial")
    public Map<String, Object> updateUserMaterial(@RequestBody UserMaterial userMaterial, HttpServletRequest request){
        return userMaterialService.updateUserMaterial(userMaterial, request);
    }

    /**
     * 删除用户绑定物料信息
     * @param id 参数
     * @return 返回结果
     */
    @DeleteMapping("delUserMaterial")
    public Map<String, Object> delUserMaterial(String id){
        return userMaterialService.delUserMaterial(id);
    }

    /**
     * 查询所有用户
     * @return 参数
     */
    @GetMapping("getUser")
    public Map<String, Object> getUser(){
        return userMaterialService.getUser();
    }


    /**
     * 查询用户绑定物料信息导出
     * @param response 参数
     */
    @GetMapping("Export")
    public void Export(UserMaterial userMaterial, HttpServletResponse response){
        userMaterialService.Export(userMaterial, response);
    }

    /**
     * 物料用户绑定导入
     * @param file 数据
     * @param request 用户id
     * @throws Exception
     */
    @PostMapping("/import")
    public Map<String, Object> importExcel(@RequestPart("file") MultipartFile file, HttpServletRequest request) throws Exception{
        Map<String, Object> map = new HashMap<>();
        JSONArray array = ExcelUtils.readMultipartFile(file);
        for(int i = 0; i < array.size(); i++){
            JSONObject jsonObject = array.getJSONObject(i);
            UserMaterial userMaterial = new UserMaterial()
                    .setMaterialId(jsonObject.get("物料编号").toString())
                    .setUserName(jsonObject.get("工号").toString())
                    .setModifyUser(request.getHeader("userId"));
            userMaterialService.replaceUser(userMaterial);
        }
        map.put("msg", "200");
        map.put("data", array.size());
        return map;
    }

}