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

   1 package org.apache.maven.archiva.database.constraints;
   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 org.apache.maven.archiva.database.Constraint;
  23 import org.apache.maven.archiva.model.ArchivaArtifactModel;
  24
  25 import java.util.List;
  26
  27 /**
  28  * Obtain a set of unique ArtifactIds for the specified groupId.
  29  *
  30  * @version $Id$
  31  */
  32 public class UniqueArtifactIdConstraint
  33     extends AbstractSimpleConstraint
  34     implements Constraint
  35 {
  36     private StringBuffer sql = new StringBuffer();
  37
  38     private Class<?> resultClass;
  39
  40     /**
  41      * Obtain a set of unique ArtifactIds for the specified groupId.
  42      *
  43      * @param groupId the groupId to search for artifactIds within.
  44      */
  45     public UniqueArtifactIdConstraint( List<String> selectedRepositoryIds, String groupId )
  46     {
  47         appendSelect( sql, false );
  48         sql.append( " WHERE " );
  49         SqlBuilder.appendWhereSelectedRepositories( sql, "repositoryId", selectedRepositoryIds );
  50         sql.append( " && " );
  51         appendWhereSelectedGroupId( sql );
  52         appendGroupBy( sql );
  53
  54         super.params = new Object[] { groupId };
  55     }
  56
  57     /**
  58      * Obtain a set of unique ArtifactIds for the specified groupId.
  59      *
  60      * @param groupId the groupId to search for artifactIds within.
  61      */
  62     public UniqueArtifactIdConstraint( String groupId )
  63     {
  64         appendSelect( sql, false );
  65         sql.append( " WHERE " );
  66         appendWhereSelectedGroupId( sql );
  67         appendGroupBy( sql );
  68
  69         super.params = new Object[] { groupId };
  70     }
  71
  72     /**
  73      * Obtain a set of unique artifactIds with respect to their groups from the specified repository.
  74      *
  75      * @param repoId
  76      * @param isUnique
  77      */
  78     public UniqueArtifactIdConstraint( String repoId, boolean isUnique )
  79     {
  80         appendSelect( sql, isUnique );
  81         sql.append( " WHERE repositoryId == \"" + repoId + "\"" );
  82
  83         resultClass = Object[].class;
  84     }
  85
  86     @SuppressWarnings("unchecked")
  87     public Class getResultClass()
  88     {
  89         if( resultClass != null )
  90         {
  91             return resultClass;
  92         }
  93
  94         return String.class;
  95     }
  96
  97     public String getSelectSql()
  98     {
  99         return sql.toString();
 100     }
 101
 102     private void appendGroupBy( StringBuffer buf )
 103     {
 104         buf.append( " GROUP BY artifactId ORDER BY artifactId ASCENDING" );
 105     }
 106
 107     private void appendSelect( StringBuffer buf, boolean isUnique )
 108     {
 109         if( isUnique )
 110         {
 111             buf.append( "SELECT DISTINCT groupId, artifactId FROM " ).append( ArchivaArtifactModel.class.getName() );
 112         }
 113         else
 114         {
 115             buf.append( "SELECT artifactId FROM " ).append( ArchivaArtifactModel.class.getName() );
 116         }
 117     }
 118
 119     private void appendWhereSelectedGroupId( StringBuffer buf )
 120     {
 121         buf.append( " groupId == selectedGroupId PARAMETERS String selectedGroupId" );
 122     }
 123
 124 }