public function setPrimaryKey($primary_key) { $this->primary_key = $primary_key; if ($this->dataSource) { $this->dataSource->setPrimaryKey($primary_key); } return $this; }
private function assertLimit(IDataSource $source) { $all = $source->fetchAll(); Assert::count(self::LIMIT, $all); Assert::count(self::FULL_USER_COUNT, $source->fetchFullData()); Assert::count(self::FULL_USER_COUNT, $source->fetchAllForExport()); Assert::same(self::FULL_USER_COUNT, $source->getTotalCount()); Assert::same(self::LIMIT, $source->count()); return $all; }