Added support for joins when doing field selections:
List<Tuple2<Integer,EnumKeyRecord.Type>> jdata =
_repo.from(TestRecord.class).
join(TestRecord.NAME, EnumKeyRecord.NAME).
select(TestRecord.RECORD_ID, EnumKeyRecord.TYPE);
This should eliminate the need for pesky one-off computed records when doing
ad-hoc joins.
This commit is contained in:
@@ -308,6 +308,14 @@ public abstract class FindAllQuery<T extends PersistentRecord,R>
|
|||||||
public ColumnSetQueryMarshaller (PersistenceContext ctx, ColumnSet<T,R> cset) {
|
public ColumnSetQueryMarshaller (PersistenceContext ctx, ColumnSet<T,R> cset) {
|
||||||
_cset = cset;
|
_cset = cset;
|
||||||
_cmarsh = ctx.getMarshaller(cset.ptype);
|
_cmarsh = ctx.getMarshaller(cset.ptype);
|
||||||
|
_marshes = Maps.newHashMap();
|
||||||
|
_marshes.put(cset.ptype, _cmarsh);
|
||||||
|
for (ColumnExp<?> cexp : cset.columns) {
|
||||||
|
Class<? extends PersistentRecord> cclass = cexp.getPersistentClass();
|
||||||
|
if (!_marshes.containsKey(cclass)) {
|
||||||
|
_marshes.put(cclass, ctx.getMarshaller(cexp.getPersistentClass()));
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public String getTableName () {
|
public String getTableName () {
|
||||||
@@ -325,13 +333,16 @@ public abstract class FindAllQuery<T extends PersistentRecord,R>
|
|||||||
public R createObject (ResultSet rs) throws SQLException {
|
public R createObject (ResultSet rs) throws SQLException {
|
||||||
Object[] data = new Object[_cset.columns.length];
|
Object[] data = new Object[_cset.columns.length];
|
||||||
for (int ii = 0; ii < data.length; ii++) {
|
for (int ii = 0; ii < data.length; ii++) {
|
||||||
data[ii] = _cmarsh.getFieldMarshaller(_cset.columns[ii].name).getFromSet(rs);
|
ColumnExp<?> col = _cset.columns[ii];
|
||||||
|
data[ii] = _marshes.get(col.getPersistentClass()).
|
||||||
|
getFieldMarshaller(col.name).getFromSet(rs, ii+1);
|
||||||
}
|
}
|
||||||
return _cset.createObject(data);
|
return _cset.createObject(data);
|
||||||
}
|
}
|
||||||
|
|
||||||
protected ColumnSet<T,R> _cset;
|
protected ColumnSet<T,R> _cset;
|
||||||
protected DepotMarshaller<T> _cmarsh;
|
protected DepotMarshaller<T> _cmarsh;
|
||||||
|
protected Map<Class<? extends PersistentRecord>, DepotMarshaller<?>> _marshes;
|
||||||
}
|
}
|
||||||
|
|
||||||
// from Query
|
// from Query
|
||||||
|
|||||||
@@ -0,0 +1,72 @@
|
|||||||
|
//
|
||||||
|
// $Id$
|
||||||
|
//
|
||||||
|
// Depot library - a Java relational persistence library
|
||||||
|
// Copyright (C) 2006-2010 Michael Bayne and Pär Winzell
|
||||||
|
//
|
||||||
|
// This library is free software; you can redistribute it and/or modify it
|
||||||
|
// under the terms of the GNU Lesser General Public License as published
|
||||||
|
// by the Free Software Foundation; either version 2.1 of the License, or
|
||||||
|
// (at your option) any later version.
|
||||||
|
//
|
||||||
|
// This library is distributed in the hope that it will be useful,
|
||||||
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
// Lesser General Public License for more details.
|
||||||
|
//
|
||||||
|
// You should have received a copy of the GNU Lesser General Public
|
||||||
|
// License along with this library; if not, write to the Free Software
|
||||||
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||||
|
|
||||||
|
package com.samskivert.depot;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import com.google.common.collect.Lists;
|
||||||
|
import com.samskivert.depot.Tuple2;
|
||||||
|
|
||||||
|
import org.junit.Test;
|
||||||
|
import static org.junit.Assert.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Tests individual field selection.
|
||||||
|
*/
|
||||||
|
public class SelectFieldsTest extends TestBase
|
||||||
|
{
|
||||||
|
@Test public void testCreateReadDelete ()
|
||||||
|
{
|
||||||
|
for (int ii = 0; ii < 10; ii++) {
|
||||||
|
_repo.insert(createTestRecord(ii));
|
||||||
|
}
|
||||||
|
_repo.insert(new EnumKeyRecord(EnumKeyRecord.Type.A, "Elvis"));
|
||||||
|
_repo.insert(new EnumKeyRecord(EnumKeyRecord.Type.B, "Moses"));
|
||||||
|
_repo.insert(new EnumKeyRecord(EnumKeyRecord.Type.C, "Abraham"));
|
||||||
|
_repo.insert(new EnumKeyRecord(EnumKeyRecord.Type.D, "Elvis"));
|
||||||
|
|
||||||
|
// test some basic one and two column selects
|
||||||
|
List<Integer> allKeys = _repo.from(TestRecord.class).select(TestRecord.RECORD_ID);
|
||||||
|
assertEquals(allKeys, Lists.newArrayList(0, 1, 2, 3, 4, 5, 6, 7, 8, 9));
|
||||||
|
|
||||||
|
List<Integer> someKeys = _repo.from(TestRecord.class).where(
|
||||||
|
TestRecord.RECORD_ID.greaterThan(5)).select(TestRecord.RECORD_ID);
|
||||||
|
assertEquals(someKeys, Lists.newArrayList(6, 7, 8, 9));
|
||||||
|
|
||||||
|
List<Tuple2<Integer,String>> data = _repo.from(TestRecord.class).where(
|
||||||
|
TestRecord.RECORD_ID.greaterThan(7)).select(TestRecord.RECORD_ID, TestRecord.NAME);
|
||||||
|
assertEquals(data, Lists.newArrayList(Tuple2.newTuple(8, "Elvis"),
|
||||||
|
Tuple2.newTuple(9, "Elvis")));
|
||||||
|
|
||||||
|
// test a basic join
|
||||||
|
List<Tuple2<Integer,EnumKeyRecord.Type>> jdata = _repo.from(TestRecord.class).join(
|
||||||
|
TestRecord.NAME, EnumKeyRecord.NAME).select(TestRecord.RECORD_ID, EnumKeyRecord.TYPE);
|
||||||
|
System.out.println(jdata);
|
||||||
|
|
||||||
|
// finally clean up after ourselves
|
||||||
|
_repo.from(TestRecord.class).where(Exps.trueLiteral()).delete();
|
||||||
|
_repo.from(EnumKeyRecord.class).where(Exps.trueLiteral()).delete();
|
||||||
|
}
|
||||||
|
|
||||||
|
// the HSQL in-memory database persists for the lifetime of the VM, which means we have to
|
||||||
|
// clean up after ourselves in every test; thus we go ahead and share a repository
|
||||||
|
protected TestRepository _repo = createTestRepository();
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user