source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.archiva.reporting;
   2
   3 /*
   4  * Copyright 2005-2006 The Apache Software Foundation.
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 import org.apache.maven.archiva.reporting.model.io.xpp3.ReportingXpp3Reader;
  20 import org.apache.maven.archiva.reporting.model.io.xpp3.ReportingXpp3Writer;
  21 import org.apache.maven.artifact.repository.ArtifactRepository;
  22 import org.codehaus.plexus.logging.AbstractLogEnabled;
  23 import org.codehaus.plexus.util.IOUtil;
  24 import org.codehaus.plexus.util.xml.pull.XmlPullParserException;
  25
  26 import java.io.File;
  27 import java.io.FileNotFoundException;
  28 import java.io.FileReader;
  29 import java.io.FileWriter;
  30 import java.io.IOException;
  31 import java.util.HashMap;
  32 import java.util.Map;
  33
  34 /**
  35  * Load and store the reports. No synchronization is used, but it is unnecessary as the old object
  36  * can continue to be used.
  37  *
  38  * @author <a href="mailto:brett@apache.org">Brett Porter</a>
  39  * @todo would be great for plexus to do this for us - so the configuration would be a component itself rather than this store
  40  * @todo support other implementations than XML file
  41  * @plexus.component
  42  */
  43 public class DefaultReportingStore
  44     extends AbstractLogEnabled
  45     implements ReportingStore
  46 {
  47     /**
  48      * The cached reports for given repositories.
  49      */
  50     private Map/*<ArtifactRepository,ReportingDatabase>*/ reports = new HashMap();
  51
  52     public ReportingDatabase getReportsFromStore( ArtifactRepository repository )
  53         throws ReportingStoreException
  54     {
  55         ReportingDatabase database = (ReportingDatabase) reports.get( repository );
  56
  57         if ( database == null )
  58         {
  59             ReportingXpp3Reader reader = new ReportingXpp3Reader();
  60
  61             File file = new File( repository.getBasedir(), "report-database.xml" );
  62
  63             FileReader fileReader = null;
  64             try
  65             {
  66                 fileReader = new FileReader( file );
  67             }
  68             catch ( FileNotFoundException e )
  69             {
  70                 database = new ReportingDatabase( repository );
  71             }
  72
  73             if ( database == null )
  74             {
  75                 getLogger().info( "Reading report database from " + file );
  76                 try
  77                 {
  78                     database = new ReportingDatabase( reader.read( fileReader, false ), repository );
  79                 }
  80                 catch ( IOException e )
  81                 {
  82                     throw new ReportingStoreException( e.getMessage(), e );
  83                 }
  84                 catch ( XmlPullParserException e )
  85                 {
  86                     throw new ReportingStoreException( e.getMessage(), e );
  87                 }
  88                 finally
  89                 {
  90                     IOUtil.close( fileReader );
  91                 }
  92             }
  93
  94             reports.put( repository, database );
  95         }
  96         return database;
  97     }
  98
  99     public void storeReports( ReportingDatabase database, ArtifactRepository repository )
 100         throws ReportingStoreException
 101     {
 102         database.updateTimings();
 103
 104         ReportingXpp3Writer writer = new ReportingXpp3Writer();
 105
 106         File file = new File( repository.getBasedir(), "report-database.xml" );
 107         getLogger().info( "Writing reports to " + file );
 108         FileWriter fileWriter = null;
 109         try
 110         {
 111             file.getParentFile().mkdirs();
 112
 113             fileWriter = new FileWriter( file );
 114             writer.write( fileWriter, database.getReporting() );
 115         }
 116         catch ( IOException e )
 117         {
 118             throw new ReportingStoreException( e.getMessage(), e );
 119         }
 120         finally
 121         {
 122             IOUtil.close( fileWriter );
 123         }
 124     }
 125 }