X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=trunk%2Fcrawler%2Fkiss%2Fsrc%2Forg%2Fwamblee%2Fcrawler%2Fkiss%2Fmain%2FProgramActionExecutor.java;h=44132b5f5e88d42f93ee2d82f9f64f4bf2331aa8;hb=5fef4c532fd806ed24f6cefdef502c6f4055d128;hp=3ef511cffe4ce9c8b36c445d163e12c57dd7886e;hpb=d1e70d1aa60fc975f6b4e65c1a683996d32b9d40;p=utils diff --git a/trunk/crawler/kiss/src/org/wamblee/crawler/kiss/main/ProgramActionExecutor.java b/trunk/crawler/kiss/src/org/wamblee/crawler/kiss/main/ProgramActionExecutor.java index 3ef511cf..44132b5f 100644 --- a/trunk/crawler/kiss/src/org/wamblee/crawler/kiss/main/ProgramActionExecutor.java +++ b/trunk/crawler/kiss/src/org/wamblee/crawler/kiss/main/ProgramActionExecutor.java @@ -12,7 +12,7 @@ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. - */ + */ package org.wamblee.crawler.kiss.main; @@ -22,156 +22,130 @@ import java.util.Set; import java.util.TreeMap; import java.util.TreeSet; +import org.apache.commons.logging.Log; +import org.apache.commons.logging.LogFactory; import org.dom4j.DocumentFactory; import org.dom4j.Element; import org.wamblee.crawler.kiss.guide.Program; import org.wamblee.crawler.kiss.guide.Program.RecordingResult; /** - * Provides execution of actions for programs. Actions use - * this class to tell the executor what to do. The executor then decide - * on exactly what to do and in what order and makes decisions in case - * of conflicts. + * Provides execution of actions for programs. Actions use this class to tell + * the executor what to do. The executor then decides on exactly what to do and + * in what order and makes decisions in case of conflicts. */ public class ProgramActionExecutor { + private static final Log LOG = LogFactory.getLog(ProgramActionExecutor.class); + /** - * A map of category name to a set of program. Useful for displaying the output of - * possibly interesting programs on a per category basis. + * A map of category name to a set of program. Useful for displaying the + * output of possibly interesting programs on a per category basis. */ private Map> _interestingShows; - + /** - * Set of programs to record. + * Set of programs to record. */ private Set _showsToRecord; - + /** - * Map or recording result to a set of programs. + * Map or recording result to a set of programs. */ private EnumMap> _recordings; - + /** - * Constructs the program action executor. - * + * Constructs the program action executor. + * */ - public ProgramActionExecutor() { - _interestingShows = new TreeMap>(); + public ProgramActionExecutor() { + _interestingShows = new TreeMap>(); _showsToRecord = new TreeSet(new Program.TimeSorter()); _recordings = new EnumMap>( RecordingResult.class); for (RecordingResult result : RecordingResult.values()) { - _recordings.put(result, new TreeSet(new Program.TimeSorter())); + _recordings.put(result, new TreeSet( + new Program.TimeSorter())); } } - + /** * Called by an action to indicate the desire to record a program. - * @param aPriority Priority of the program. Used to resolve conflicts. - * @param aProgram Program to record. + * + * @param aPriority + * Priority of the program. Used to resolve conflicts. + * @param aProgram + * Program to record. */ - public void recordProgram(int aPriority, Program aProgram) { + public void recordProgram(int aPriority, Program aProgram) { + LOG.info("priority = " + aPriority + ", program: " + aProgram); _showsToRecord.add(aProgram); } - + /** - * Called by an action to indicate that a program is interesting. - * @param aCategory Category of the program. - * @param aProgram Program. + * Called by an action to indicate that a program is interesting. + * + * @param aCategory + * Category of the program. + * @param aProgram + * Program. */ - public void interestingProgram(String aCategory, Program aProgram) { - Set programs = _interestingShows.get(aCategory); - if ( programs == null ) { + public void interestingProgram(String aCategory, Program aProgram) { + LOG.info("category = '" + aCategory + "', program: " + aProgram); + Set programs = _interestingShows.get(aCategory); + if (programs == null) { programs = new TreeSet(new Program.TimeSorter()); _interestingShows.put(aCategory, programs); } programs.add(aProgram); } - + /** * Makes sure that the actions are performed. - * + * */ - public void commit() { - for (Program program: _showsToRecord) { - RecordingResult result = program.record(); + public void commit() { + for (Program program : _showsToRecord) { + RecordingResult result = program.record(); _recordings.get(result).add(program); } } - + /** - * Gets the report describing what was done. - * @return Report. + * Get report as XML. + * + * @return XML report */ - public String getReport() { - StringBuffer msg = new StringBuffer("Summary of KiSS crawler: \n\n\n"); - - boolean printed = false; + public Element getReport() { + DocumentFactory factory = DocumentFactory.getInstance(); + Element report = factory.createElement("report"); for (RecordingResult result : RecordingResult.values()) { if (_recordings.get(result).size() > 0) { - msg.append(result.getDescription() + "\n\n"); - for (Program program : _recordings.get(result)) { - msg.append(program + "\n\n"); - printed = true; - } - } - } - - if ( _interestingShows.size() > 0 ) { - msg.append("Possibly interesting shows:\n\n"); - for (String category: _interestingShows.keySet()) { - if ( category.length() > 0 ) { - msg.append("Category: " + category + "\n\n"); - } - for (Program program: _interestingShows.get(category)) { - msg.append(program + "\n\n"); - printed = true; - } - } - - } - if (!printed) { - msg.append("No suitable programs found"); - } + Element recordingResult = report.addElement("recorded") + .addAttribute("result", result.toString()); - return msg.toString(); - } - - /** - * Get report as XML. - * @return XML report - */ - public Element getXmlReport() { - DocumentFactory factory = DocumentFactory.getInstance(); - Element report = factory.createElement("report"); - - for (RecordingResult result : RecordingResult.values()) { - if (_recordings.get(result).size() > 0) { - Element recordingResult = report.addElement("recorded").addAttribute("result", result.toString()); - for (Program program : _recordings.get(result)) { - recordingResult.add(program.asXml()); + recordingResult.add(program.asXml()); } } } - - - if ( _interestingShows.size() > 0 ) { + + if (_interestingShows.size() > 0) { Element interesting = report.addElement("interesting"); - for (String category: _interestingShows.keySet()) { - Element categoryElem = interesting; - if ( category.length() > 0 ) { + for (String category : _interestingShows.keySet()) { + Element categoryElem = interesting; + if (category.length() > 0) { categoryElem = interesting.addElement("category"); categoryElem.addAttribute("name", category); } - for (Program program: _interestingShows.get(category)) { - categoryElem.add(program.asXml()); + for (Program program : _interestingShows.get(category)) { + categoryElem.add(program.asXml()); } } - + } - - return report; + + return report; } }