使用CakePHP以块的形式导入CSV文件

时间:2013-11-27 10:36:35

标签: php mysql cakephp csv

我正在尝试将大约23.000行的大型CSV文件导入我的MySQL数据库。我不能一次导入所有规则,这是行不通的。所以我想知道或者有可能以块的形式读取文件,即使在我使用cakePHP事务时也是如此。到目前为止,这是我的代码:

// Get data source for transactions
$dataSource = $this->FeedImport->Product->getDataSource();

try{
    //Start Transactions
    $dataSource->begin();

    // Create an empty array for the CSV data
    $data = array();
    $i = 0;

    // read each data row in the file
    while (($row = fgetcsv($handle)) !== false) {
        // for each header field
        foreach ($header as $k=>$head) {
            // Remove any special characters from $head
            $head = preg_replace('/[^A-Za-z0-9\-]/', '', $head);
            if(array_key_exists($head, $this->fields)){
                //Check the row contains an image, if so, download
                if(preg_match('/\.(?:jpe?g|png|gif)$/i', $row[$k])){
                    foreach($this->fields[$head] as $table => $field){
                        $imageFileName = uniqid($supplier.'_');
                        $data[$i][$table][][$field] = $imageFileName.'.'.end(explode('.', $row[$k]));
                        $this->__importImg($row[$k]);
                    }
                }else{
                    foreach($this->fields[$head] as $table => $field){
                        if($table == 'Term'){
                            if(isset($row[$k]) && !$this->FeedImport->Product->Term->find('first', array('conditions' => array('Term.name' => $row[$k])))){
                                if(!$this->FeedImport->Product->Term->save(
                                    array(
                                        'name' => $row[$k]
                                    )
                                ));
                            }
                            if(isset($row[$k])) $term = $this->FeedImport->Product->Term->find('first', array('conditions' => array('Term.name' => $row[$k])));
                            $data[$i][$table][$table][$field] = (isset($term['Term']['term_id'])) ? $term['Term']['term_id'] : '';
                        }else{
                            $data[$i][$table][$field] = (isset($row[$k])) ? $row[$k] : '';
                        }
                    }
                }
            }
        }

        $data[$i]['Product']['product_id_supplier'] = $data[$i]['Product']['slug'];
        $data[$i]['Product']['supplier_id'] = $supplier;
        $data[$i]['Product']['feedimport_id'] = 1;

        $i++;
    }

    // save the row
    if (!$this->FeedImport->Product->saveAll($data)) {
        throw new Exception();
    }

} catch(Exception $e) {
    $dataSource->rollback($e);
}
$dataSource->commit();

我把上面的代码放在一个单独的函数中,所以我可以为while循环提供一个起始行和结束行。但是我遇到了困难,我不知道如何使用fgetcsv设置开始和结束规则。有人可以帮助我吗?

我尝试过使用fseek等等,但我无法完成它...有人可以帮助我吗?

我考虑使用LOAD DATA INFILE导入那些大型产品,但我认为这不会很好用,因为我使用多个连接表和一些例外将数据导入到几个表中...所以这太糟糕了

2 个答案:

答案 0 :(得分:1)

使用PHP5.5,您可以利用新的Generator功能。 见http://mark-story.com/posts/view/php-generators-a-useful-example

答案 1 :(得分:1)

可能的解决方法如下

while (($row = fgetcsv($handle)) !== false) {
    if ($i === 5000){
        try {
            if (!$this->FeedImport->Product->saveAll($data))
                throw new Exception();
        } catch (Exception $e){
            $dataSource->rollback($e);
        }

        $i = 0;
        $data = [];
    }

    // Code
}

对于每5000条记录,它将提交数据,重置计数器和数据数组并继续。