ThinkPHP6数据导入与导出：实现数据批量处理

ThinkPHP6数据导入与导出：实现数据批量处理

在实际开发中，经常会遇到需要批量导入和导出数据的需求，例如导入Excel表格中的数据到数据库中，或将数据库中的数据导出为Excel文件。这样的操作可以提高开发效率和减少手动输入数据的工作量。本文将介绍如何使用ThinkPHP6框架实现数据的批量处理，包括数据导入和导出的具体步骤和代码示例。

一、数据导入

1. 准备工作

首先，需要在项目中安装PHPExcel库来处理Excel文件。可以使用Composer来安装PHPExcel，执行以下命令：

composer require phpoffice/phpexcel

安装完成后，将会生成一个vendor目录，其中包含PHPExcel库的相关文件。

2. 导入Excel文件

在ThinkPHP6中，可以使用request()函数获取到用户上传的文件。首先在控制器中创建一个用于处理导入操作的方法：

public function import()
{
    // 获取上传的文件
    $file = request()->file('file');
    
    // 移动到框架应用根目录/uploads/目录下
    $info = $file->validate(['size' => 1048576, 'ext' => 'xls,xlsx'])->move(ROOT_PATH . 'uploads/');

    if ($info) {
        // 获取上传文件的路径
        $filename = $info->getSaveName();
        
        // 处理Excel导入逻辑
        // ...
        
        // 返回成功信息
        return '数据导入成功！';
    } else {
        // 返回错误信息
        return $file->getError();
    }
}

上述代码中，首先通过request()函数获取到上传的文件，并进行合法性校验，限制文件大小为1MB，并只允许上传.xls和.xlsx格式的文件。然后使用move()方法将文件移动到框架的uploads目录下，并将文件名保存到$filename变量中。

接下来，可以在导入逻辑中使用PHPExcel库来读取和处理Excel文件。以下是一个简单的示例：

public function import()
{
    // ...

    // 创建PHPExcel对象
    $excel = new PHPExcel();

    // 读取Excel文件
    $reader = PHPExcel_IOFactory::createReader('Excel2007');
    $PHPExcel = $reader->load(ROOT_PATH . 'uploads/' . $filename);

    // 获取第一个工作表
    $sheet = $PHPExcel->getSheet(0);

    // 获取总行数
    $totalRow = $sheet->getHighestRow();

    // 遍历每一行数据
    for ($i = 2; $i <= $totalRow; $i++) {
        // 获取单元格数据
        $name = $sheet->getCell('A' . $i)->getValue();
        $age = $sheet->getCell('B' . $i)->getValue();

        // 处理数据插入操作
        // ...
    }

    // ...
}

上述代码中，我们使用PHPExcel库创建了一个PHPExcel对象，并使用createReader()方法来读取Excel文件。然后通过getSheet()方法获取第一个工作表的对象，使用getHighestRow()方法获取总行数。

接下来，通过遍历每一行的数据，使用getCell()方法获取指定单元格的值，并将数据插入数据库中，完成导入操作。

二、数据导出

1. 导出数据库数据

首先，在控制器中创建一个用于处理导出操作的方法：

public function export()
{
    // 查询数据库数据
    $data = Db::name('user')->select();
    
    // 处理Excel导出逻辑
    // ...
}

上述代码中，使用ThinkPHP6的查询构造器Db::name('user')->select()查询数据库中的用户数据。

2. 导出为Excel文件

接下来，我们使用PHPExcel库将数据导出为Excel文件：

public function export()
{
    // ...
    
    // 创建PHPExcel对象
    $excel = new PHPExcel();
    
    // 设置工作表标题
    $excel->getActiveSheet()->setTitle('用户数据');

    // 设置表头
    $excel->getActiveSheet()->setCellValue('A1', 'ID');
    $excel->getActiveSheet()->setCellValue('B1', '姓名');
    $excel->getActiveSheet()->setCellValue('C1', '年龄');

    // 设置数据内容
    $row = 2;
    foreach($data as $item) {
        $excel->getActiveSheet()->setCellValue('A' . $row, $item['id']);
        $excel->getActiveSheet()->setCellValue('B' . $row, $item['name']);
        $excel->getActiveSheet()->setCellValue('C' . $row, $item['age']);
        $row++;
    }

    // 导出Excel文件
    header('Content-Type: application/vnd.openxmlformats-officedocument.spreadsheetml.sheet');
    header('Content-Disposition: attachment;filename="user_data.xlsx"');
    header('Cache-Control: max-age=0');
    $writer = PHPExcel_IOFactory::createWriter($excel, 'Excel2007');
    $writer->save('php://output');
}

上述代码中，我们创建了一个PHPExcel对象，并使用setTitle()方法设置工作表的标题。然后使用setCellValue()方法设置表头和数据内容。

最后，通过设置响应头的方式，将导出的Excel文件发送给浏览器进行下载。

总结

本文介绍了如何使用ThinkPHP6框架来实现数据的批量处理，包括数据导入和导出的具体步骤和代码示例。通过使用PHPExcel库，我们可以方便地处理Excel文件，提高开发效率和减少手动输入数据的工作量。希望本文对你有所帮助，能够在实际开发中发挥作用。