diff --git a/src/java/com/samskivert/jdbc/depot/clause/SelectClause.java b/src/java/com/samskivert/jdbc/depot/clause/SelectClause.java index 8bca190..66b4cdd 100644 --- a/src/java/com/samskivert/jdbc/depot/clause/SelectClause.java +++ b/src/java/com/samskivert/jdbc/depot/clause/SelectClause.java @@ -28,6 +28,7 @@ import java.util.Map; import com.samskivert.jdbc.depot.PersistentRecord; import com.samskivert.jdbc.depot.WhereClause; +import com.samskivert.jdbc.depot.annotation.Computed; import com.samskivert.jdbc.depot.expression.ExpressionVisitor; /** @@ -160,6 +161,16 @@ public class SelectClause extends QueryClause public void addClasses (Collection> classSet) { classSet.add(_pClass); + + // TODO: This should not have to do a getAnnotation(). + Computed computed = _pClass.getAnnotation(Computed.class); + if (computed != null) { + Class shadowClass = computed.shadowOf(); + if (shadowClass != null) { + classSet.add(shadowClass); + } + } + if (_fromOverride != null) { _fromOverride.addClasses(classSet); }