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.util.HashSet;
23 import java.util.Set;
24
25 import org.apache.maven.index.context.IndexingContext;
26
27 /**
28 * A special reusable filter, that filters the result set to unique Repository-GroupId-ArtifactId combination, leaving
29 * out Version. There is a switch to make the Indexer-wide unique by ignoring repositories too.
30 *
31 * @author cstamas
32 */
33 public class UniqueArtifactFilterPostprocessor
34 implements ArtifactInfoFilter
35 {
36 public static final String COLLAPSED = "COLLAPSED";
37
38 private final Set<Field> uniqueFields = new HashSet<Field>();
39
40 private final Set<String> gas = new HashSet<String>();
41
42 public UniqueArtifactFilterPostprocessor()
43 {
44 }
45
46 public UniqueArtifactFilterPostprocessor( Set<Field> uniqueFields )
47 {
48 this.uniqueFields.addAll( uniqueFields );
49 }
50
51 public boolean accepts( IndexingContext ctx, ArtifactInfo ai )
52 {
53 StringBuilder sb = new StringBuilder();
54
55 for ( Field field : uniqueFields )
56 {
57 sb.append( ai.getFieldValue( field ) ).append( ":" );
58 }
59
60 String key = sb.toString().substring( 0, sb.length() - 1 );
61
62 if ( gas.contains( key ) )
63 {
64 return false;
65 }
66 else
67 {
68 gas.add( key );
69
70 postprocess( ctx, ai );
71
72 return true;
73 }
74 }
75
76 public void postprocess( IndexingContext ctx, ArtifactInfo ai )
77 {
78 for ( Field field : ai.getFields() )
79 {
80 if ( !uniqueFields.contains( field ) )
81 {
82 ai.setFieldValue( field, COLLAPSED );
83 }
84 }
85 }
86
87 public void addField( Field field )
88 {
89 uniqueFields.add( field );
90 }
91 }