add query which returns limited number of results

eucalyptus · Nov 30, 2011 · 0b27b39 · 0b27b39
1 parent 7ac5c86
commit 0b27b39
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/clc/modules/msgs/src/main/java/com/eucalyptus/component/Faults.java b/clc/modules/msgs/src/main/java/com/eucalyptus/component/Faults.java
@@ -492,7 +492,7 @@ public static Function<ServiceStatusType, CheckException> transformToExceptions(
   public static Collection<CheckException> lookup( final ServiceConfiguration config ) {
     final EntityTransaction db = Entities.get( CheckException.class );
     try {
-      final List<CheckException> res = Entities.query( new CheckException( config.getName( ) ) );
+      final List<CheckException> res = Entities.query( new CheckException( config.getName( ) ), true, 1 );
       db.commit( );
       return res;
     } catch ( final Exception ex ) {

diff --git a/clc/modules/msgs/src/main/java/com/eucalyptus/entities/Entities.java b/clc/modules/msgs/src/main/java/com/eucalyptus/entities/Entities.java
@@ -233,6 +233,21 @@ public static <T> List<T> query( final T example, final boolean readOnly ) {
     return Lists.newArrayList( Sets.newHashSet( resultList ) );
   }
 
+  @SuppressWarnings( { "unchecked", "cast" } )
+  public static <T> List<T> query( final T example, final boolean readOnly, final int maxResults ) {
+    final Example qbe = Example.create( example ).enableLike( MatchMode.EXACT );
+    final List<T> resultList = ( List<T> ) getTransaction( example ).getTxState( ).getSession( )
+                                                                    .createCriteria( example.getClass( ) )
+                                                                    .setReadOnly( readOnly )
+                                                                    .setResultTransformer( Criteria.DISTINCT_ROOT_ENTITY )
+                                                                    .setCacheable( true )
+                                                                    .add( qbe )
+                                                                    .setMaxResults( maxResults )
+                                                                    .setFetchSize( maxResults )
+                                                                    .list( );
+    return Lists.newArrayList( Sets.newHashSet( resultList ) );
+  }
+
   public static <T> T uniqueResult( final T example ) throws TransactionException, NoSuchElementException {
     try {
       final Object pk = resolvePrimaryKey( example );