diff -r 11fd79666374 -r 8551d844b4f3 Command/SyncDocumentsCommand.php --- a/Command/SyncDocumentsCommand.php Fri Nov 04 15:59:49 2011 +0100 +++ b/Command/SyncDocumentsCommand.php Sun Nov 06 23:44:37 2011 +0100 @@ -27,6 +27,8 @@ $this ->setName('wikitag:sync-doc') ->setDescription('Synchronize and index document class') + ->addOption('missing', 'm', InputOption::VALUE_NONE, "process missing") + ->addOption('tags', 't', InputOption::VALUE_NONE, "update tags") ->addOption('clear', 'c', InputOption::VALUE_NONE, "clear all docs"); } @@ -34,6 +36,7 @@ { $class = $this->getContainer()->getParameter('wiki_tag.document_class'); $clear = $input->getOption('clear'); + $missing = $input->getOption('missing'); $doctrine = $this->getContainer()->get('doctrine'); @@ -47,15 +50,75 @@ return ; } + if($input->getOption('tags')) + { + if(!$missing) + { + $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc from WikiTagBundle:Document doc"); + $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT COUNT(doc.id) from WikiTagBundle:Document doc"); + } + else + { + $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc from WikiTagBundle:Document doc WHERE doc.tagsStr IS NULL"); + $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT COUNT(doc.id) from WikiTagBundle:Document doc WHERE doc.tagsStr IS NULL"); + } + + + $total = $doccountquery->getSingleScalarResult(); + $done = 0; + $iterable = $docquery->iterate(); + $todetach = array(); + while (($row = $iterable->next()) !== false) { + $done++; + $memory = ((($done%10)==0)?" - mem: ".strval(memory_get_usage(true)):""); + $doc = $row[0]; + $todetach[] = $doc; + $output->writeln("Process doc id ".$doc->getId()." $done/$total ".strval(intval(floatval($done)/floatval($total)*100.0))."%$memory"); + $docrep->updateTagsStr($doc); + if($done%10 == 0) + { + $doctrine->getEntityManager()->flush(); + foreach($todetach as $obj) + { + $doctrine->getEntityManager()->detach($obj); + } + $todetach = array(); + } + } + $doctrine->getEntityManager()->flush(); + + return; + + } + //TODO : check class to implement DocumentInterface //TODO : write progress - $doclist = $rep->findAll(); - $total = count($doclist); + if($missing) + { + $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc FROM $class doc WHERE doc.id not in (SELECT wtdoc FROM WikiTagBundle:Document wtdoc)"); + $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT count(doc.id) FROM $class doc WHERE doc.id not in (SELECT wtdoc FROM WikiTagBundle:Document wtdoc)"); + //$doclist = $doctrine->getEntityManager()->createQuery("SELECT doc FROM $class doc WHERE doc.id not in (SELECT wtdoc FROM WikiTagBundle:Document wtdoc)")->getResult(); + } + else + { + $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc FROM $class doc"); + $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT count(doc.id) FROM $class doc"); + //$doclist = $rep->findAll(); + } + $total = $doccountquery->getSingleScalarResult(); $done = 0; - foreach ($doclist as $doc) { + $iterable = $docquery->iterate(); + while (($row = $iterable->next()) !== false) { $done++; - $output->writeln("Process doc id ".$doc->getId()." $done/$total ".strval(intval(floatval($done)/floatval($total)*100.0))."%"); + $doc = $row[0]; + $memory = ((($done%10)==0)?" - mem: ".strval(memory_get_usage(true)):""); + $output->writeln("Process doc id ".$doc->getId()." $done/$total ".strval(intval(floatval($done)/floatval($total)*100.0))."%$memory"); $docrep->writeDocument($doc, $this->getContainer()->getParameter('wiki_tag.document_id_column'), $this->getContainer()->getParameter('wiki_tag.fields')); + if($done%10 == 0) + { + $doctrine->getEntityManager()->flush(); + $doctrine->getEntityManager()->clear(); + } } $doctrine->getEntityManager()->flush();