Skip to content

Update CsvImport.php - Check For Header In The File #10

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open

Update CsvImport.php - Check For Header In The File #10

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
8fb172f
Update CsvImport.php
tiag00liveira Oct 6, 2016
6c1e303
Update CsvImport.php
tiag00liveira Oct 6, 2016
0cadd13
Update CsvImport.php
tiag00liveira Oct 7, 2016
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 30 additions & 3 deletions src/RdnCsv/Controller/Plugin/CsvImport.php
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ class CsvImport extends AbstractPlugin implements \Countable, \Iterator
*
* @return CsvImport
*/
public function __invoke($filepath, $useFirstRecordAsHeader = true, $delimiter = ',', $enclosure = '"', $escape = '\\')
public function __invoke($filepath, $useFirstRecordAsHeader = true, $delimiter = ',', $enclosure = '"', $escape = '\\', $checkHeader=false)
{
$this->file = new SplFileObject($filepath);
$this->file->setFlags(SplFileObject::READ_CSV | SplFileObject::READ_AHEAD | SplFileObject::SKIP_EMPTY | SplFileObject::DROP_NEW_LINE);
Expand All @@ -45,7 +45,10 @@ public function __invoke($filepath, $useFirstRecordAsHeader = true, $delimiter =

$this->useFirstRecordAsHeader = $useFirstRecordAsHeader;

$this->checkHeader = $checkHeader;

return $this;

}

/**
Expand All @@ -66,15 +69,18 @@ public function getFile()
*/
public function rewind()
{
$this->checkHeader();
$this->file->rewind();
if ($this->useFirstRecordAsHeader)
{

if (!$this->file->valid())
{
throw new \RuntimeException('Expected first row to be header, but reached EOF instead');
}
$this->header = $this->file->current();
$this->file->next();

}
}

Expand Down Expand Up @@ -108,6 +114,7 @@ public function current()
return array_combine($header, $line);
}
return $line;

}

/**
Expand All @@ -131,9 +138,9 @@ public function valid()
}

/**
* Count number of records in file (excluding header if first record is used as header).
* Count elements of an object
*
* @return int
* @return int The custom count as an integer.
*/
public function count()
{
Expand All @@ -148,4 +155,24 @@ public function count()
}
return $total;
}
/**
* check for the header in the file
*
* @throws \RuntimeException
*/
public function checkHeader()
{
$checkHeader = $this->checkHeader;
if ($checkHeader) {
foreach ($this->file as $line){
$data[] = $line;
}
$headerFile = $data[0];
$check = array_values(array_diff($headerFile, $checkHeader));
if ($check) {
throw new \RuntimeException(sprintf('Not Found Header %s',$check[0]));
}
}
}

}