backend_dormitory/src/main/java/com/template/controller/WelcomeBuildController.java

package com.template.controller;


import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.template.api.WelcomeBuildControllerAPI;
import com.template.common.utils.ExcelUtils;
import com.template.model.enumModel.eFileType;
import com.template.model.pojo.*;
import com.template.model.result.CommonResult;
import com.template.model.result.PageUtils;
import com.template.model.vo.StudentPageVo;
import com.template.model.vo.WelcomeBuildVo;
import com.template.services.WelcomeBuildService;
import com.template.services.WelcomeSchoolService;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.DataFormatter;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * <p>
 * 前端控制器
 * </p>
 *
 * @author ceshi
 * @since 2025-06-13
 */
@RestController
public class WelcomeBuildController implements WelcomeBuildControllerAPI {

    @Autowired
    private WelcomeBuildService welcomeBuildService;

    @Autowired
    WelcomeSchoolService welcomeSchoolService;

    @Override
    public CommonResult saveBuild(WelcomeBuild welcomeBuild) {
        String build = welcomeBuild.getBuild();
        LambdaQueryWrapper<WelcomeBuild> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(WelcomeBuild::getBuild, build)
                .eq(WelcomeBuild::getSchool,welcomeBuild.getSchool());
        WelcomeBuild wb = welcomeBuildService.getOne(wrapper);

        if (ObjectUtils.isNotEmpty(wb)) {
            return CommonResult.fail("已存在该楼栋");
        }

        welcomeBuildService.save(welcomeBuild);

        return CommonResult.ok();
    }

    @Override
    public CommonResult deleteBuild(Integer buildId) {

        boolean result = welcomeBuildService.removeById(buildId);

        return result ? CommonResult.ok() : CommonResult.fail();
    }

    @Override
    public CommonResult updateBuild(WelcomeBuild welcomeBuild) {

        //先查询是否有当前楼栋信息
        LambdaQueryWrapper<WelcomeBuild> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(WelcomeBuild::getBuild, welcomeBuild.getBuild())
                .eq(WelcomeBuild::getSchool,welcomeBuild.getSchool());
        WelcomeBuild one = welcomeBuildService.getOne(wrapper);

        if (ObjectUtils.isNotEmpty(one)) {
            Integer id = one.getId();
            if (!id.equals(welcomeBuild.getId())) {
                return CommonResult.fail("修改的楼栋已存在");
            }
        }

        welcomeBuildService.updateById(welcomeBuild);

        return CommonResult.ok();
    }

    @Override
    public CommonResult listBuild(int currentPage, int pageCount, Integer schoolId,String buildId, String sex) {

        PageUtils<WelcomeBuildVo> pageUtils =welcomeBuildService.listBuild(currentPage,pageCount,schoolId,buildId,sex);

        return  CommonResult.ok(pageUtils);
    }

    @Override
    public CommonResult schoolGroup() {
        List<WelcomeSchool> schoolGroup = welcomeSchoolService.schoolGroup();

        return CommonResult.ok(schoolGroup);
    }

    @Override
    public CommonResult buildGroup(Integer schoolId) {
        List<WelcomeBuild> schoolGroup = welcomeBuildService.buildGroup(schoolId);

        return CommonResult.ok(schoolGroup);
    }

    @Override
    public CommonResult importBuildExcel(MultipartFile file) throws IOException, ParseException {
        System.out.println("导入楼栋信息");
        if (file.isEmpty() || file.getSize() == 0) {
            return CommonResult.fail("导入文件不能为空");
        }
        String ContentType = file.getContentType();
        InputStream inputStream = file.getInputStream();
        List<WelcomeBuild> result = new ArrayList<>();

        //xls格式文件
        if (ContentType.equals(eFileType.Xls.getValue())) {

            CommonResult<List<WelcomeBuild>> resultData = readXls(inputStream);
            if (!resultData.isSuccess()) {
                return resultData;
            }
            result = resultData.getData();

        } else if (ContentType.equals(eFileType.Xlsx.getValue())) {
            CommonResult<List<WelcomeBuild>> resultData = readXlsx(inputStream);
            if (!resultData.isSuccess()) {
                return resultData;
            }
            result = resultData.getData();

        } else {
            return CommonResult.fail("楼栋信息数据导入只支持Xls或Xlsx格式文件");
        }

        boolean resultBool = welcomeBuildService.saveOrUpdateBatch(result);

        System.out.println("导入楼栋1");
        return resultBool ? CommonResult.ok("导入成功") : CommonResult.fail("导入失败");
    }

    /**
     * xls文件读取方法
     *
     * @param inputStream
     * @return
     * @throws IOException
     * @throws ParseException
     */
    private CommonResult<List<WelcomeBuild>> readXls(InputStream inputStream) throws IOException, ParseException {
        List<WelcomeBuild> result = new ArrayList<>();
        HSSFWorkbook sheets = new HSSFWorkbook(inputStream);

        List<WelcomeBuild> wb = welcomeBuildService.list(null);

        //读取第一张sheet
        HSSFSheet sheetAt = sheets.getSheetAt(0);
        DataFormatter dataFormatter = new DataFormatter();

        try {
            //rowNum = 3 从第三行开始获取值
            //sheetAt.getLastRowNum()：从0开始统计数量 所以得+1
            for (int rowNum = 0; rowNum < sheetAt.getLastRowNum() + 1; rowNum++) {

                HSSFRow row = sheetAt.getRow(rowNum);

                if (row != null) {
                    //使用了getStringCellValue()方法来获取值，POI会判断单元格的类型，如果非字符串类型就会抛出上面的异常。
                    //所以先使用setCellType()方法先将该单元格的类型设置为STRING
                    //然后poi会根据字符串读取它
                    //第一行数据获取月份
                    if (rowNum == 0) {
                        String number = dataFormatter.formatCellValue(row.getCell(0));//序号
                        if (!number.equals("序号")) {
                            return CommonResult.fail("导入数据第一列为序号");
                        }
                        String school = dataFormatter.formatCellValue(row.getCell(1));//校区名称
                        if (!school.equals("校区名称")) {
                            return CommonResult.fail("导入数据第二列为校区名称");
                        }
                        String build = dataFormatter.formatCellValue(row.getCell(2));//楼栋名称
                        if (!build.equals("楼栋名称")) {
                            return CommonResult.fail("导入数据第三列为楼栋名称");
                        }
                        String sex = dataFormatter.formatCellValue(row.getCell(3));//楼栋性别
                        if (!sex.equals("楼栋性别")) {
                            return CommonResult.fail("导入数据第四列为楼栋性别");
                        }
                        String floors = dataFormatter.formatCellValue(row.getCell(4));//楼栋层数
                        if (!floors.equals("楼栋层数")) {
                            return CommonResult.fail("导入数据第五列为楼栋层数");
                        }
                        String startFloor = dataFormatter.formatCellValue(row.getCell(5));//起始层数
                        if (!startFloor.equals("起始层数")) {
                            return CommonResult.fail("导入数据第六列为起始层数");
                        }
                        String remark = dataFormatter.formatCellValue(row.getCell(6));//备注
                        if (!remark.equals("备注")) {
                            return CommonResult.fail("导入数据第七列为备注");
                        }
                    } else {
                        WelcomeBuild buildData = new WelcomeBuild();

                        String school = dataFormatter.formatCellValue(row.getCell(1));//校区名称
                        String build = dataFormatter.formatCellValue(row.getCell(2));//楼栋名称
                        Optional<WelcomeBuild> ob = wb.stream().filter(e -> e.getSchool().equals(school) && e.getBuild().equals(build)).findFirst();
                        if(ob != null && ob.isPresent()){
                            buildData.setId(ob.get().getId());
                        }

                        buildData.setSchool(school);
                        buildData.setBuild(build);
                        String sex = dataFormatter.formatCellValue(row.getCell(3));//楼栋性别
                        buildData.setSex(sex);
                        String floors = dataFormatter.formatCellValue(row.getCell(4));//楼栋层数
                        buildData.setFloors(floors);
                        String startFloor = dataFormatter.formatCellValue(row.getCell(5));//起始层数
                        buildData.setStartFloor(Integer.valueOf(startFloor));
                        String remark = dataFormatter.formatCellValue(row.getCell(6));//备注
                        buildData.setRemark(remark);

                        result.add(buildData);
                    }
                }
            }
        } catch (Exception e) {
            return CommonResult.fail("请按模板格式导入数据");
        }

        return CommonResult.ok(result);
    }

    /**
     * xls文件读取方法
     *
     * @param inputStream
     * @return
     * @throws IOException
     * @throws ParseException
     */
    private CommonResult<List<WelcomeBuild>> readXlsx(InputStream inputStream) throws IOException, ParseException {
        List<WelcomeBuild> result = new ArrayList<>();
        XSSFWorkbook sheets = new XSSFWorkbook(inputStream);

        List<WelcomeBuild> wb = welcomeBuildService.list(null);

        //读取第一张sheet
        XSSFSheet sheetAt = sheets.getSheetAt(0);
        DataFormatter dataFormatter = new DataFormatter();

        try {
            //rowNum = 3 从第三行开始获取值
            //sheetAt.getLastRowNum()：从0开始统计数量 所以得+1
            for (int rowNum = 0; rowNum < sheetAt.getLastRowNum() + 1; rowNum++) {

                XSSFRow row = sheetAt.getRow(rowNum);

                if (row != null) {
                    //使用了getStringCellValue()方法来获取值，POI会判断单元格的类型，如果非字符串类型就会抛出上面的异常。
                    //所以先使用setCellType()方法先将该单元格的类型设置为STRING
                    //然后poi会根据字符串读取它
                    //第一行数据获取月份
                    if (rowNum == 0) {
                        String number = dataFormatter.formatCellValue(row.getCell(0));//序号
                        if (!number.equals("序号")) {
                            return CommonResult.fail("导入数据第一列为序号");
                        }
                        String school = dataFormatter.formatCellValue(row.getCell(1));//校区名称
                        if (!school.equals("校区名称")) {
                            return CommonResult.fail("导入数据第二列为校区名称");
                        }
                        String build = dataFormatter.formatCellValue(row.getCell(2));//楼栋名称
                        if (!build.equals("楼栋名称")) {
                            return CommonResult.fail("导入数据第三列为楼栋名称");
                        }
                        String sex = dataFormatter.formatCellValue(row.getCell(3));//楼栋性别
                        if (!sex.equals("楼栋性别")) {
                            return CommonResult.fail("导入数据第四列为楼栋性别");
                        }
                        String floors = dataFormatter.formatCellValue(row.getCell(4));//楼栋层数
                        if (!floors.equals("楼栋层数")) {
                            return CommonResult.fail("导入数据第五列为楼栋层数");
                        }
                        String startFloor = dataFormatter.formatCellValue(row.getCell(5));//起始层数
                        if (!startFloor.equals("起始层数")) {
                            return CommonResult.fail("导入数据第六列为起始层数");
                        }
                        String remark = dataFormatter.formatCellValue(row.getCell(6));//备注
                        if (!remark.equals("备注")) {
                            return CommonResult.fail("导入数据第七列为备注");
                        }
                    } else {
                        WelcomeBuild buildData = new WelcomeBuild();

                        String school = dataFormatter.formatCellValue(row.getCell(1));//校区名称
                        String build = dataFormatter.formatCellValue(row.getCell(2));//楼栋名称
                        Optional<WelcomeBuild> ob = wb.stream().filter(e -> e.getSchool().equals(school) && e.getBuild().equals(build)).findFirst();
                        if(ob != null && ob.isPresent()){
                            buildData.setId(ob.get().getId());
                        }

                        buildData.setSchool(school);
                        buildData.setBuild(build);
                        String sex = dataFormatter.formatCellValue(row.getCell(3));//楼栋性别
                        buildData.setSex(sex);
                        String floors = dataFormatter.formatCellValue(row.getCell(4));//楼栋层数
                        buildData.setFloors(floors);
                        String startFloor = dataFormatter.formatCellValue(row.getCell(5));//起始层数
                        buildData.setStartFloor(Integer.valueOf(startFloor));
                        String remark = dataFormatter.formatCellValue(row.getCell(6));//备注
                        buildData.setRemark(remark);

                        result.add(buildData);
                    }
                }
            }
        } catch (Exception e) {
            return CommonResult.fail("请按模板格式导入数据");
        }

        return CommonResult.ok(result);
    }

    @Override
    public CommonResult downloadBuildExcel() {
        return CommonResult.ok("200", "操作成功", "https://chtech.ncjti.edu.cn/alumnus/homeimage/学生信息管理.xlsx");
    }

    @Override
    public void welcomeBuildExport(HttpServletResponse response, String schoolName, String buildName, String sex) {
        List<WelcomeBuild> result = welcomeBuildService.listBuild(schoolName,buildName,sex);
        //导出
        Workbook workbook = new XSSFWorkbook();
        Sheet sheet = workbook.createSheet("楼栋信息");

        Row headerRow = sheet.createRow(0);
        headerRow.createCell(0).setCellValue("序号");
        headerRow.createCell(1).setCellValue("校区名称");
        headerRow.createCell(2).setCellValue("楼栋名称");
        headerRow.createCell(3).setCellValue("楼栋性别");
        headerRow.createCell(4).setCellValue("楼栋层数");
        headerRow.createCell(5).setCellValue("起始层数");
        headerRow.createCell(6).setCellValue("备注");
        for (int i = 0; i < result.size(); i++) {
            WelcomeBuild build = result.get(i);
            Row dataRow = sheet.createRow(i + 1);
            dataRow.createCell(0).setCellValue(i + 1);
            dataRow.createCell(1).setCellValue(build.getSchool());
            dataRow.createCell(2).setCellValue(build.getBuild());
            dataRow.createCell(3).setCellValue(build.getSex());
            dataRow.createCell(4).setCellValue(build.getFloors());
            dataRow.createCell(5).setCellValue(build.getStartFloor());
            dataRow.createCell(6).setCellValue(build.getRemark());
        }
        // 将工作簿写入文件
        ExcelUtils.excelDownload(workbook, "楼栋信息.xlsx", response);
    }

}