JAVA中使用Springboot和Mybatis框架导出PDF文件

首先是POJO类，表示学生信息。假设学生有id、姓名(name)和年龄(age)属性，可以创建如下的Student类：

public class Student {
    private Long id;
    private String name;
    private int age;

    // Getters and setters
}

接下来是Mapper接口和Mapper XML文件。Mapper接口定义了对数据库进行操作的方法，Mapper XML文件用于映射这些方法到具体的SQL语句。假设你的数据库表名为students，可以创建如下的Mapper接口和XML文件：

public interface StudentMapper {
    List<Student> getAllStudents();
}

<mapper namespace="com.example.mapper.StudentMapper">
    <select id="getAllStudents" resultType="com.example.pojo.Student">
        SELECT * FROM students
    </select>
</mapper>

然后是Service接口和实现类。Service层定义了业务逻辑，实现类负责实际的业务处理。这里创建一个打印学生表格的功能：

public interface StudentService {
    List<Student> getAllStudents();
    ByteArrayOutputStream generatePDFReport(List<Student> students);
}

@Service
public class StudentServiceImpl implements StudentService {
    @Autowired
    private StudentMapper studentMapper;

    @Override
    public List<Student> getAllStudents() {
        return studentMapper.getAllStudents();
    }

    @Override
    public ByteArrayOutputStream generatePDFReport(List<Student> students) {
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();

        try {
            Document document = new Document();
            PdfWriter.getInstance(document, outputStream);
            document.open();

            PdfPTable table = new PdfPTable(3);
            table.addCell("ID");
            table.addCell("Name");
            table.addCell("Age");

            for (Student student : students) {
                table.addCell(student.getId().toString());
                table.addCell(student.getName());
                table.addCell(String.valueOf(student.getAge()));
            }

            document.add(table);
            document.close();
        } catch (Exception e) {
            e.printStackTrace();
        }

        return outputStream;
    }
}

如果是四个字段及以上需要这样写:

 @Override
    public ByteArrayOutputStream generatePDFReport(List<ShpsOrgRoleResourceRel> shpsOrgRoleResourceRels) {
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();

        try {
            Document document = new Document();
            PdfWriter.getInstance(document, outputStream);
            document.open();

            PdfPTable table = getPdfPTable(shpsOrgRoleResourceRels);

            document.add(table);
            document.close();
        } catch (Exception e) {
            e.printStackTrace();
        }

        return outputStream;
    }

    /**
     * 改模版
     * @param shpsOrgRoleResourceRels
     * @return
     */
    private static PdfPTable getPdfPTable(List<ShpsOrgRoleResourceRel> shpsOrgRoleResourceRels) {
        PdfPTable table = new PdfPTable(4);
        table.addCell("RoleResourceCode");
        table.addCell("RoleCode");
        table.addCell("ResourceCode");
        table.addCell("State");


        for (ShpsOrgRoleResourceRel shpsOrgRoleResourceRel : shpsOrgRoleResourceRels) {
            table.addCell(shpsOrgRoleResourceRel.getRoleResourceCode());
            table.addCell(shpsOrgRoleResourceRel.getRoleCode());
            table.addCell(shpsOrgRoleResourceRel.getResourceCode());
            table.addCell(shpsOrgRoleResourceRel.getState().toString());
        }
        return table;
    }

在这个例子中，generatePDFReport方法接受一个学生列表作为参数，并返回一个包含PDF数据的ByteArrayOutputStream。它使用iText库来生成PDF表格，将学生数据插入到表格中。

接下来是Controller层，处理HTTP请求并调用Service层的方法：

@RestController
public class StudentController {
    @Autowired
    private StudentService studentService;

    @GetMapping("/printStudentsPDF")
    public ResponseEntity<byte[]> printStudentsPDF() {
        List<Student> students = studentService.getAllStudents();
        ByteArrayOutputStream outputStream = studentService.generatePDFReport(students);

        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_PDF);
        headers.setContentDispositionFormData("attachment", "students.pdf");

        return new ResponseEntity<>(outputStream.toByteArray(), headers, HttpStatus.OK);
    }
}

四个字段的Controller并且下载到桌面上的这样写:

/**
     * 导出pdf格式
     * @return
     * @throws IOException
     */
    @RequestMapping("/printShpsOrgRoleResourceRelPDF")
    public ResponseEntity<byte[]> printShpsOrgRoleResourceRelPDF() throws IOException {
        List<ShpsOrgRoleResourceRel> shpsOrgRoleResourceRels = shpsOrgRoleResourceRelService.getShpsOrgRoleResourceRelList();
        ByteArrayOutputStream outputStream = shpsOrgRoleResourceRelService.generatePDFReport(shpsOrgRoleResourceRels);

        byte[] bytes = outputStream.toByteArray();
        ByteArrayResource resource = new ByteArrayResource(bytes);
        // 保存到桌面
        String desktopPath = System.getProperty("user.home") + "/Desktop/shpsOrgRoleResourceRels.pdf";
        FileOutputStream fileOutputStream = new FileOutputStream(desktopPath);
        fileOutputStream.write(bytes);
        fileOutputStream.close();

        //输出到控制台
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_PDF);
        headers.setContentDispositionFormData("attachment", "shpsOrgRoleResourceRels.pdf");

        return new ResponseEntity<>(outputStream.toByteArray(), headers, HttpStatus.OK);
    }

在这个例子中，当访问/printStudentsPDF路径时，将调用Service层的方法获取学生数据并生成PDF文件，然后将PDF文件作为响应返回。

最后是POM文件中的依赖配置，包括Spring Boot、MyBatis以及iText库的依赖。这里只展示关键部分：

<dependencies>
    <!-- Spring Boot -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

    <!-- MyBatis -->
    <dependency>
        <groupId>org.mybatis.spring.boot</groupId>
        <artifactId>mybatis-spring-boot-starter</artifactId>
        <version>2.1.4</version> <!-- 版本号根据实际情况调整 -->
    </dependency>

    <!-- iText PDF -->
    <dependency>
        <groupId>com.itextpdf</groupId>
        <artifactId>itextpdf</artifactId>
        <version>5.5.13</version> <!-- 版本号根据实际情况调整 -->
    </dependency>
</dependencies>