341 lines
11 KiB
PHP
Executable File
341 lines
11 KiB
PHP
Executable File
<?php
|
||
|
||
/**
|
||
* Niushop商城系统 - 团队十年电商经验汇集巨献!
|
||
* =========================================================
|
||
* Copy right 2019-2029 杭州牛之云科技有限公司, 保留所有权利。
|
||
* ----------------------------------------------
|
||
* 官方网址: https://www.niushop.com
|
||
* =========================================================
|
||
*/
|
||
|
||
namespace app\model\system;
|
||
|
||
use app\model\BaseModel;
|
||
use think\facade\Db;
|
||
|
||
class Export extends BaseModel
|
||
{
|
||
const STATUS_IN_PROCESS = 0;
|
||
const STATUS_SUCCESS = 1;
|
||
const STATUS_FAIL = 2;
|
||
|
||
static public function getStatus($key = null)
|
||
{
|
||
$arr = [
|
||
[
|
||
'id' => self::STATUS_IN_PROCESS,
|
||
'name' => '导出中',
|
||
],
|
||
[
|
||
'id' => self::STATUS_SUCCESS,
|
||
'name' => '导出成功',
|
||
],
|
||
[
|
||
'id' => self::STATUS_FAIL,
|
||
'name' => '导出失败',
|
||
],
|
||
];
|
||
if(isset($arr[0][$key])){
|
||
$arr = array_column($arr, null, $key);
|
||
}
|
||
return $arr;
|
||
}
|
||
|
||
protected $data_num = 0;
|
||
|
||
/**
|
||
* 导出
|
||
* @param $param
|
||
* @return array
|
||
*/
|
||
public function export($param)
|
||
{
|
||
//传参说明
|
||
/*$param = [
|
||
'site_id' => $site_id,
|
||
'store_id' => $export_store_id,
|
||
'from_type' => 'store_account',
|
||
'from_type_name' => '门店账户',
|
||
'condition_desc' => $condition_desc,
|
||
'export_field' => [
|
||
['field' => 'store_name', 'name' => '门店名称'],
|
||
['field' => 'type_name', 'name' => '来源方式'],
|
||
['field' => 'account_data', 'name' => '记录金额'],
|
||
['field' => 'create_time', 'name' => '发生时间'],
|
||
['field' => 'remark', 'name' => '备注'],
|
||
['field' => 'related_no', 'name' => '关联编号'],
|
||
],
|
||
//数据库查询相关参数
|
||
'query' => [
|
||
'table' => 'store_account',
|
||
'alias' => 'sa',
|
||
'join' => [
|
||
['store s', 'sa.store_id = s.store_id', 'inner'],
|
||
],
|
||
'condition' => $condition,
|
||
'field' => 'sa.*,s.store_name',
|
||
'chunk_field' => 'sa.id',
|
||
'chunk_order' => 'asc',
|
||
],
|
||
//数据库查询结果处理 这个其实应该放在query中
|
||
'handle' => function($item_list){
|
||
return $item_list;
|
||
},
|
||
//直接传递导出的数据
|
||
'data' => [],
|
||
];*/
|
||
|
||
set_time_limit(0);
|
||
|
||
$site_id = $param['site_id'];
|
||
$store_id = $param['store_id'] ?? 0;
|
||
$from_type = $param['from_type'];
|
||
$from_type_name = $param['from_type_name'];
|
||
$condition_desc= $param['condition_desc'];
|
||
$export_field = $param['export_field'];
|
||
$query = $param['query'] ?? null;//table alias join condition field order
|
||
$handle = $param['handle'] ?? null;//数据处理
|
||
|
||
try {
|
||
//预先创建导出的记录
|
||
$data = array(
|
||
'condition' => json_encode($condition_desc, JSON_UNESCAPED_UNICODE),
|
||
'create_time' => time(),
|
||
'from_type' => $from_type,
|
||
'from_type_name' => $from_type_name,
|
||
'status' => self::STATUS_IN_PROCESS,
|
||
'site_id' => $site_id,
|
||
'store_id' => $store_id,
|
||
);
|
||
$records_result = $this->addExport($data);
|
||
$export_id = $records_result['data'];
|
||
if (empty($export_id)) return $this->error(null, '创建导出记录失败');
|
||
$this->data_num = 0;
|
||
|
||
//创建目录
|
||
$file_path = 'upload/export/'.$from_type.'/';
|
||
if (!dir_mkdir($file_path)) return $this->error(null, '导出目录创建失败');
|
||
|
||
//创建并打开文件
|
||
$file_name = $from_type_name.date('YmdHis');//csv文件名
|
||
$file_path = $file_path . $file_name . '.csv';
|
||
$fp = fopen($file_path, 'w');
|
||
fwrite($fp, chr(0xEF) . chr(0xBB) . chr(0xBF));
|
||
|
||
//写入第一行表头
|
||
fwrite($fp, join(',', array_column($export_field, 'name'))."\n");
|
||
|
||
//导出数据
|
||
if(isset($param['query'])){
|
||
$export_table = Db::name($query['table'])->where($query['condition']);
|
||
if(!empty($query['alias'])) $export_table = $export_table->alias($query['alias']);
|
||
if(!empty($query['join'])) $export_table = $this->parseJoin($export_table, $query['join']);
|
||
if(!empty($query['group'])) $export_table = $export_table->group($query['group']);
|
||
$export_table->field($query['field'])->chunk(1000, function ($item_list) use ($fp, $export_field, $handle) {
|
||
$item_list = $item_list->toArray();
|
||
if(!empty($handle)){
|
||
$item_list = $handle($item_list);
|
||
}
|
||
$this->data_num += count($item_list);
|
||
$this->itemExport($item_list, $export_field, $fp);
|
||
unset($item_list);
|
||
}, $query['chunk_field'], $query['chunk_order']);
|
||
$export_table->removeOption();
|
||
}else if(isset($param['data'])){
|
||
$this->itemExport($param['data'], $export_field, $fp);
|
||
}else{
|
||
throw new \Exception('query参数和data参数至少要传一个');
|
||
}
|
||
|
||
fclose($fp); //每生成一个文件关闭
|
||
unset($export_table);
|
||
|
||
//更新导出记录
|
||
$this->editExport([
|
||
'path' => $file_path,
|
||
'status' => self::STATUS_SUCCESS,
|
||
'data_num' => $this->data_num,
|
||
], [['export_id', '=', $export_id]]);
|
||
|
||
//返回导出信息
|
||
$export_info = $this->getExportInfo([['export_id', '=', $export_id]])['data'];
|
||
return $this->success($export_info);
|
||
} catch (\Exception $e ) {
|
||
$error = [
|
||
'file' => $e->getFile(),
|
||
'line' => $e->getLine(),
|
||
'message' => $e->getMessage(),
|
||
];
|
||
if(isset($export_id)){
|
||
//更新导出记录
|
||
$this->editExport([
|
||
'status' => self::STATUS_FAIL,
|
||
'fail_log' => json_encode($error, JSON_UNESCAPED_UNICODE),
|
||
], [['export_id', '=', $export_id]]);
|
||
}
|
||
return $this->error($error, '导出失败');
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 给csv写入新的数据
|
||
* @param $item_list
|
||
* @param $field_key
|
||
* @param $temp_line
|
||
* @param $fp
|
||
*/
|
||
protected function itemExport($item_list, $export_field, $fp)
|
||
{
|
||
if(method_exists($item_list, 'toArray')){
|
||
$item_list = $item_list->toArray();
|
||
}
|
||
foreach ($item_list as $item_info) {
|
||
$values = [];
|
||
foreach($export_field as $field_info){
|
||
if(isset($item_info[$field_info['field']])){
|
||
$value = $item_info[$field_info['field']];
|
||
$value = trim($value);
|
||
$value = str_replace(',', '，', $value . "\t");
|
||
$value = str_replace("\n", '', $value);
|
||
$value = str_replace("\r", '', $value);
|
||
$values[] = $value;
|
||
}
|
||
}
|
||
//写入数据
|
||
fwrite($fp, join(',', $values)."\n");
|
||
//销毁变量, 防止内存溢出
|
||
unset($new_line_value);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 解析关联
|
||
* @param $db_obj
|
||
* @param $join
|
||
* @return mixed
|
||
*/
|
||
protected function parseJoin($db_obj, $join)
|
||
{
|
||
foreach ($join as $item) {
|
||
list($table, $on, $type) = $item;
|
||
$type = strtolower($type);
|
||
switch ($type) {
|
||
case 'left':
|
||
$db_obj = $db_obj->leftJoin($table, $on);
|
||
break;
|
||
case 'inner':
|
||
$db_obj = $db_obj->join($table, $on);
|
||
break;
|
||
case 'right':
|
||
$db_obj = $db_obj->rightjoin($table, $on);
|
||
break;
|
||
case 'full':
|
||
$db_obj = $db_obj->fulljoin($table, $on);
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
return $db_obj;
|
||
}
|
||
|
||
/**
|
||
* 添加导出记录
|
||
* @param $data
|
||
* @return array
|
||
*/
|
||
public function addExport($data)
|
||
{
|
||
$res = model('export')->add($data);
|
||
return $this->success($res);
|
||
}
|
||
|
||
/**
|
||
* 更新导出记录
|
||
* @param $data
|
||
* @param $condition
|
||
* @return array
|
||
*/
|
||
public function editExport($data, $condition)
|
||
{
|
||
$res = model('export')->update($data, $condition);
|
||
return $this->success($res);
|
||
}
|
||
|
||
/**
|
||
* 删除导出记录
|
||
* @param $condition
|
||
* @return array
|
||
*/
|
||
public function deleteExport($condition)
|
||
{
|
||
//先查询数据
|
||
$list = model('export')->getList($condition, '*');
|
||
if (!empty($list)) {
|
||
foreach ($list as $k => $v) {
|
||
if (file_exists($v['path'])) {
|
||
//删除物理文件路径
|
||
if (!unlink($v['path'])) {
|
||
//失败
|
||
} else {
|
||
//成功
|
||
}
|
||
}
|
||
}
|
||
$res = model('export')->delete($condition);
|
||
}
|
||
return $this->success($res ?? '');
|
||
}
|
||
|
||
public function getExportInfo($condition, $field = '*')
|
||
{
|
||
$info = model('export')->getInfo($condition, $field);
|
||
$info = $this->handleData($info);
|
||
return $this->success($info);
|
||
}
|
||
|
||
/**
|
||
* 获取导出记录
|
||
* @param $condition
|
||
* @param string $field
|
||
* @param string $order
|
||
* @return array
|
||
*/
|
||
public function getExportList($condition, $field = '*', $order = '')
|
||
{
|
||
$list = model('export')->getList($condition, $field, $order);
|
||
foreach($list as &$val){
|
||
$val = $this->handleData($val);
|
||
}
|
||
return $this->success($list);
|
||
}
|
||
|
||
/**
|
||
* 导出记录
|
||
* @param array $condition
|
||
* @param int $page
|
||
* @param int $page_size
|
||
* @param string $order
|
||
* @param string $field
|
||
* @return array
|
||
*/
|
||
public function getExportPageList($condition = [], $page = 1, $page_size = PAGE_LIST_ROWS, $order = '', $field = '*')
|
||
{
|
||
$list = model('export')->pageList($condition, $field, $order, $page, $page_size);
|
||
foreach($list['list'] as &$val){
|
||
$val = $this->handleData($val);
|
||
}
|
||
return $this->success($list);
|
||
}
|
||
|
||
protected function handleData($data)
|
||
{
|
||
if(isset($data['status'])){
|
||
$status_list = self::getStatus('id');
|
||
$data['status_name'] = $status_list[$data['status']]['name'] ?? '';
|
||
}
|
||
return $data;
|
||
}
|
||
}
|