Coverage Report

Coverage Report - org.apache.maven.index.DefaultScannerListener

Classes in this File

Line Coverage

Branch Coverage

Complexity

DefaultScannerListener

90 %

92/102

84 %

44/52

5,571

 package org.apache.maven.index;
 
 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership.  The ASF licenses this file
  * to you under the Apache License, Version 2.0 (the
  * "License"); you may not use this file except in compliance
  * with the License.  You may obtain a copy of the License at
  *
  *   http://www.apache.org/licenses/LICENSE-2.0    
  *
  * Unless required by applicable law or agreed to in writing,
  * software distributed under the License is distributed on an
  * "AS IS" BASIS, 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.
  */
 
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
 
 import org.apache.lucene.document.Document;
 import org.apache.lucene.index.CorruptIndexException;
 import org.apache.lucene.index.IndexReader;
 import org.apache.lucene.index.Term;
 import org.apache.lucene.search.TermQuery;
 import org.apache.lucene.search.TopScoreDocCollector;
 import org.apache.maven.index.context.IndexingContext;
 import org.codehaus.plexus.logging.AbstractLogEnabled;
 
 /**
  * A default scanning listener
  * 
  * @author Eugene Kuleshov
  */
 class DefaultScannerListener
     extends AbstractLogEnabled
     implements ArtifactScanningListener
 {
     private final IndexingContext context;
 
     private final IndexerEngine indexerEngine;
 
     private final boolean update;
 
     private final ArtifactScanningListener listener;
 
     private final Set<String> uinfos = new HashSet<String>();
 
     private final Set<String> processedUinfos = new HashSet<String>();
 
     private final Set<String> allGroups = new HashSet<String>();
 
     private final Set<String> groups = new HashSet<String>();
 
     private final List<Exception> exceptions = new ArrayList<Exception>();
 
     private int count = 0;
 
     DefaultScannerListener( IndexingContext context, //
                             IndexerEngine indexerEngine, boolean update, //
                             ArtifactScanningListener listener )
     {
         this.context = context;
         this.indexerEngine = indexerEngine;
         this.update = update;
         this.listener = listener;
     }
 
     public void scanningStarted( IndexingContext ctx )
     {
         try
         {
             if ( update )
             {
                 initialize( ctx );
             }
         }
         catch ( IOException ex )
         {
             exceptions.add( ex );
         }
 
         if ( listener != null )
         {
             listener.scanningStarted( ctx );
         }
     }
 
     public void artifactDiscovered( ArtifactContext ac )
     {
         String uinfo = ac.getArtifactInfo().getUinfo();
 
         // TODO: scattered across commented out changes while I was fixing NEXUS-2712, cstamas
         // These changes should be applied by borks too much the fragile indexer
 
         // if ( VersionUtils.isSnapshot( ac.getArtifactInfo().version ) && processedUinfos.contains( uinfo ) )
         if ( processedUinfos.contains( uinfo ) )
         {
             return; // skip individual snapshots
         }
 
         boolean adding = processedUinfos.add( uinfo );
 
         if ( uinfos.contains( uinfo ) )
         {
             // already indexed
             uinfos.remove( uinfo );
             return;
         }
 
         try
         {
             if ( listener != null )
             {
                 listener.artifactDiscovered( ac );
             }
 
             if ( adding )
             {
                 indexerEngine.index( context, ac );
             }
             else
             {
                 indexerEngine.update( context, ac );
             }
 
             for ( Exception e : ac.getErrors() )
             {
                 artifactError( ac, e );
             }
 
             groups.add( ac.getArtifactInfo().getRootGroup() );
             allGroups.add( ac.getArtifactInfo().groupId );
 
             count++;
         }
         catch ( IOException ex )
         {
             artifactError( ac, ex );
         }
     }
 
     public void scanningFinished( IndexingContext ctx, ScanningResult result )
     {
         result.setTotalFiles( count );
 
         for ( Exception ex : exceptions )
         {
             result.addException( ex );
         }
 
         try
         {
             context.optimize();
 
             context.setRootGroups( groups );
 
             context.setAllGroups( allGroups );
 
             if ( update && !context.isReceivingUpdates() )
             {
                 removeDeletedArtifacts( context, result, result.getRequest().getStartingPath() );
             }
         }
         catch ( IOException ex )
         {
             result.addException( ex );
         }
 
         if ( listener != null )
         {
             listener.scanningFinished( ctx, result );
         }
 
         if ( result.getDeletedFiles() > 0 || result.getTotalFiles() > 0 )
         {
             try
             {
                 context.updateTimestamp( true );
 
                 context.optimize();
             }
             catch ( Exception ex )
             {
                 result.addException( ex );
             }
         }
     }
 
     public void artifactError( ArtifactContext ac, Exception e )
     {
         exceptions.add( e );
 
         if ( listener != null )
         {
             listener.artifactError( ac, e );
         }
     }
 
     private void initialize( IndexingContext ctx )
         throws IOException, CorruptIndexException
     {
         IndexReader r = ctx.getIndexReader();
 
         for ( int i = 0; i < r.maxDoc(); i++ )
         {
             if ( !r.isDeleted( i ) )
             {
                 Document d = r.document( i );
 
                 String uinfo = d.get( ArtifactInfo.UINFO );
 
                 if ( uinfo != null )
                 {
                     uinfos.add( uinfo );
 
                     // add all existing groupIds to the lists, as they will
                     // not be "discovered" and would be missing from the new list..
                     String groupId = uinfo.substring( 0, uinfo.indexOf( '|' ) );
                     int n = groupId.indexOf( '.' );
                     groups.add( n == -1 ? groupId : groupId.substring( 0, n ) );
                     allGroups.add( groupId );
                 }
             }
         }
     }
 
     private void removeDeletedArtifacts( IndexingContext context, ScanningResult result, String contextPath )
         throws IOException
     {
         int deleted = 0;
 
         for ( String uinfo : uinfos )
         {
             TopScoreDocCollector collector = TopScoreDocCollector.create( 1, false );
 
             context.getIndexSearcher().search( new TermQuery( new Term( ArtifactInfo.UINFO, uinfo ) ), collector );
 
             if ( collector.getTotalHits() > 0 )
             {
                 String[] ra = ArtifactInfo.FS_PATTERN.split( uinfo );
 
                 ArtifactInfo ai = new ArtifactInfo();
 
                 ai.repository = context.getRepositoryId();
 
                 ai.groupId = ra[0];
 
                 ai.artifactId = ra[1];
 
                 ai.version = ra[2];
 
                 if ( ra.length > 3 )
                 {
                     ai.classifier = ArtifactInfo.renvl( ra[3] );
                 }
 
                 if ( ra.length > 4 )
                 {
                     ai.packaging = ArtifactInfo.renvl( ra[4] );
                 }
 
                 // minimal ArtifactContext for removal
                 ArtifactContext ac = new ArtifactContext( null, null, null, ai, ai.calculateGav() );
 
                 for ( int i = 0; i < collector.getTotalHits(); i++ )
                 {
                     if ( contextPath == null
                         || context.getGavCalculator().gavToPath( ac.getGav() ).startsWith( contextPath ) )
                     {
                         indexerEngine.remove( context, ac );
                     }
 
                     deleted++;
                 }
             }
         }
 
         if ( deleted > 0 )
         {
             context.commit();
         }
 
         result.setDeletedFiles( deleted );
     }
 
 }

1		package org.apache.maven.index;
2
3		/*
4		* Licensed to the Apache Software Foundation (ASF) under one
5		* or more contributor license agreements. See the NOTICE file
6		* distributed with this work for additional information
7		* regarding copyright ownership. The ASF licenses this file
8		* to you under the Apache License, Version 2.0 (the
9		* "License"); you may not use this file except in compliance
10		* with the License. You may obtain a copy of the License at
11		*
12		* http://www.apache.org/licenses/LICENSE-2.0
13		*
14		* Unless required by applicable law or agreed to in writing,
15		* software distributed under the License is distributed on an
16		* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17		* KIND, either express or implied. See the License for the
18		* specific language governing permissions and limitations
19		* under the License.
20		*/
21
22		import java.io.IOException;
23		import java.util.ArrayList;
24		import java.util.HashSet;
25		import java.util.List;
26		import java.util.Set;
27
28		import org.apache.lucene.document.Document;
29		import org.apache.lucene.index.CorruptIndexException;
30		import org.apache.lucene.index.IndexReader;
31		import org.apache.lucene.index.Term;
32		import org.apache.lucene.search.TermQuery;
33		import org.apache.lucene.search.TopScoreDocCollector;
34		import org.apache.maven.index.context.IndexingContext;
35		import org.codehaus.plexus.logging.AbstractLogEnabled;
36
37		/**
38		* A default scanning listener
39		*
40		* @author Eugene Kuleshov
41		*/
42		class DefaultScannerListener
43		extends AbstractLogEnabled
44		implements ArtifactScanningListener
45		{
46		private final IndexingContext context;
47
48		private final IndexerEngine indexerEngine;
49
50		private final boolean update;
51
52		private final ArtifactScanningListener listener;
53
54	214	private final Set<String> uinfos = new HashSet<String>();
55
56	214	private final Set<String> processedUinfos = new HashSet<String>();
57
58	214	private final Set<String> allGroups = new HashSet<String>();
59
60	214	private final Set<String> groups = new HashSet<String>();
61
62	214	private final List<Exception> exceptions = new ArrayList<Exception>();
63
64	214	private int count = 0;
65
66		DefaultScannerListener( IndexingContext context, //
67		IndexerEngine indexerEngine, boolean update, //
68		ArtifactScanningListener listener )
69	214	{
70	214	this.context = context;
71	214	this.indexerEngine = indexerEngine;
72	214	this.update = update;
73	214	this.listener = listener;
74	214	}
75
76		public void scanningStarted( IndexingContext ctx )
77		{
78		try
79		{
80	214	if ( update )
81		{
82	10	initialize( ctx );
83		}
84		}
85	0	catch ( IOException ex )
86		{
87	0	exceptions.add( ex );
88	214	}
89
90	214	if ( listener != null )
91		{
92	7	listener.scanningStarted( ctx );
93		}
94	214	}
95
96		public void artifactDiscovered( ArtifactContext ac )
97		{
98	9098	String uinfo = ac.getArtifactInfo().getUinfo();
99
100		// TODO: scattered across commented out changes while I was fixing NEXUS-2712, cstamas
101		// These changes should be applied by borks too much the fragile indexer
102
103		// if ( VersionUtils.isSnapshot( ac.getArtifactInfo().version ) && processedUinfos.contains( uinfo ) )
104	9098	if ( processedUinfos.contains( uinfo ) )
105		{
106	3103	return; // skip individual snapshots
107		}
108
109	5995	boolean adding = processedUinfos.add( uinfo );
110
111	5995	if ( uinfos.contains( uinfo ) )
112		{
113		// already indexed
114	6	uinfos.remove( uinfo );
115	6	return;
116		}
117
118		try
119		{
120	5989	if ( listener != null )
121		{
122	343	listener.artifactDiscovered( ac );
123		}
124
125	5989	if ( adding )
126		{
127	5989	indexerEngine.index( context, ac );
128		}
129		else
130		{
131	0	indexerEngine.update( context, ac );
132		}
133
134	5989	for ( Exception e : ac.getErrors() )
135		{
136	45	artifactError( ac, e );
137		}
138
139	5989	groups.add( ac.getArtifactInfo().getRootGroup() );
140	5989	allGroups.add( ac.getArtifactInfo().groupId );
141
142	5989	count++;
143		}
144	0	catch ( IOException ex )
145		{
146	0	artifactError( ac, ex );
147	5989	}
148	5989	}
149
150		public void scanningFinished( IndexingContext ctx, ScanningResult result )
151		{
152	214	result.setTotalFiles( count );
153
154	214	for ( Exception ex : exceptions )
155		{
156	45	result.addException( ex );
157		}
158
159		try
160		{
161	214	context.optimize();
162
163	214	context.setRootGroups( groups );
164
165	214	context.setAllGroups( allGroups );
166
167	214	if ( update && !context.isReceivingUpdates() )
168		{
169	10	removeDeletedArtifacts( context, result, result.getRequest().getStartingPath() );
170		}
171		}
172	0	catch ( IOException ex )
173		{
174	0	result.addException( ex );
175	214	}
176
177	214	if ( listener != null )
178		{
179	7	listener.scanningFinished( ctx, result );
180		}
181
182	214	if ( result.getDeletedFiles() > 0 \|\| result.getTotalFiles() > 0 )
183		{
184		try
185		{
186	204	context.updateTimestamp( true );
187
188	204	context.optimize();
189		}
190	0	catch ( Exception ex )
191		{
192	0	result.addException( ex );
193	204	}
194		}
195	214	}
196
197		public void artifactError( ArtifactContext ac, Exception e )
198		{
199	45	exceptions.add( e );
200
201	45	if ( listener != null )
202		{
203	7	listener.artifactError( ac, e );
204		}
205	45	}
206
207		private void initialize( IndexingContext ctx )
208		throws IOException, CorruptIndexException
209		{
210	10	IndexReader r = ctx.getIndexReader();
211
212	38	for ( int i = 0; i < r.maxDoc(); i++ )
213		{
214	28	if ( !r.isDeleted( i ) )
215		{
216	28	Document d = r.document( i );
217
218	28	String uinfo = d.get( ArtifactInfo.UINFO );
219
220	28	if ( uinfo != null )
221		{
222	12	uinfos.add( uinfo );
223
224		// add all existing groupIds to the lists, as they will
225		// not be "discovered" and would be missing from the new list..
226	12	String groupId = uinfo.substring( 0, uinfo.indexOf( '\|' ) );
227	12	int n = groupId.indexOf( '.' );
228	12	groups.add( n == -1 ? groupId : groupId.substring( 0, n ) );
229	12	allGroups.add( groupId );
230		}
231		}
232		}
233	10	}
234
235		private void removeDeletedArtifacts( IndexingContext context, ScanningResult result, String contextPath )
236		throws IOException
237		{
238	10	int deleted = 0;
239
240	10	for ( String uinfo : uinfos )
241		{
242	6	TopScoreDocCollector collector = TopScoreDocCollector.create( 1, false );
243
244	6	context.getIndexSearcher().search( new TermQuery( new Term( ArtifactInfo.UINFO, uinfo ) ), collector );
245
246	6	if ( collector.getTotalHits() > 0 )
247		{
248	6	String[] ra = ArtifactInfo.FS_PATTERN.split( uinfo );
249
250	6	ArtifactInfo ai = new ArtifactInfo();
251
252	6	ai.repository = context.getRepositoryId();
253
254	6	ai.groupId = ra[0];
255
256	6	ai.artifactId = ra[1];
257
258	6	ai.version = ra[2];
259
260	6	if ( ra.length > 3 )
261		{
262	6	ai.classifier = ArtifactInfo.renvl( ra[3] );
263		}
264
265	6	if ( ra.length > 4 )
266		{
267	3	ai.packaging = ArtifactInfo.renvl( ra[4] );
268		}
269
270		// minimal ArtifactContext for removal
271	6	ArtifactContext ac = new ArtifactContext( null, null, null, ai, ai.calculateGav() );
272
273	12	for ( int i = 0; i < collector.getTotalHits(); i++ )
274		{
275	6	if ( contextPath == null
276		\|\| context.getGavCalculator().gavToPath( ac.getGav() ).startsWith( contextPath ) )
277		{
278	0	indexerEngine.remove( context, ac );
279		}
280
281	6	deleted++;
282		}
283		}
284	6	}
285
286	10	if ( deleted > 0 )
287		{
288	3	context.commit();
289		}
290
291	10	result.setDeletedFiles( deleted );
292	10	}
293
294		}