Search in sources :

Example 1 with JPAQuery

use of com.mysema.query.jpa.impl.JPAQuery in project opencast by opencast.

the class AbstractAssetManagerDeleteTestBase method assertSnapshotsTotal.

void assertSnapshotsTotal(long count) {
    assertEquals("[AssetManager] There should be " + count + " snapshots total", count, q.select(q.snapshot()).run().getSize());
    assertEquals(format("[SQL] There should be %d snapshots total", count), count, runCount(new JPAQuery().from(Q_SNAPSHOT)));
}
Also used : JPAQuery(com.mysema.query.jpa.impl.JPAQuery)

Example 2 with JPAQuery

use of com.mysema.query.jpa.impl.JPAQuery in project opencast by opencast.

the class AssetDtos method baseJoin.

/**
 * Create base join for a {@link AssetDtos} query.
 */
public static JPAQuery baseJoin(EntityManager em) {
    final QAssetDto assetDto = QAssetDto.assetDto;
    final QSnapshotDto snapshotDto = QSnapshotDto.snapshotDto;
    return new JPAQuery(em, Database.TEMPLATES).from(assetDto).leftJoin(snapshotDto).on(snapshotDto.id.eq(assetDto.snapshotId));
}
Also used : JPAQuery(com.mysema.query.jpa.impl.JPAQuery)

Example 3 with JPAQuery

use of com.mysema.query.jpa.impl.JPAQuery in project opencast by opencast.

the class AbstractASelectQuery method run.

private AResult run(JPAQueryFactory f) {
    // run query and map the result to records
    final long startTime = System.nanoTime();
    // resolve AST
    final SelectQueryContribution r = contributeSelect(f);
    final boolean toFetchProperties = r.fetch.exists(Booleans.<Expression<?>>eq(QPropertyDto.propertyDto));
    // # create Querydsl query
    final JPAQuery q = f.query();
    // # from
    {
        // Make sure that the snapshotDto is always contained in the from clause because the media package ID and
        // the ID are always selected.
        // Use a mutable hash set to be able to use the removeAll operation.
        final Set<EntityPath<?>> from = Stream.<EntityPath<?>>mk(Q_SNAPSHOT).append(// all collected from clauses
        r.from).append(// all from clauses from the joins
        r.join.map(Join.getFrom)).toSet(SetB.MH);
        // Now remove everything that will be joined. Adding them in both the from and a join
        // clause is not allowed.
        from.removeAll(r.join.map(Join.getJoin).toSet());
        q.from(JpaFns.toEntityPathArray(from));
    }
    // # join
    if (!r.join.isEmpty()) {
        // Group joins by entity and combine all "on" clauses with "or" expressions.
        // This way there is only one join clause per distinct entity which eliminates the need to alias entities
        // like this `new QPropertyDto("alias")`.
        // Entity aliasing produces many issues which seem to cause a huge rewrite of the query building mechanism
        // so it should be prevented at all costs.
        final Map<EntityPath<?>, BooleanExpression> joins = r.join.foldl(new HashMap<EntityPath<?>, BooleanExpression>(), new Fn2<Map<EntityPath<?>, BooleanExpression>, Join, Map<EntityPath<?>, BooleanExpression>>() {

            @Override
            public Map<EntityPath<?>, BooleanExpression> apply(Map<EntityPath<?>, BooleanExpression> sum, Join join) {
                // get the on expression saved with the join, may be null
                final BooleanExpression existing = sum.get(join.join);
                final BooleanExpression combined;
                // combine the existing and the current expression
                if (existing == null) {
                    combined = join.on;
                } else if (existing.equals(join.on)) {
                    // if both expressions are equal there is no need to combine them
                    combined = existing;
                } else {
                    // if different combine with logical "or"
                    combined = existing.or(join.on);
                }
                sum.put(join.join, combined);
                return sum;
            }
        });
        for (final Map.Entry<EntityPath<?>, BooleanExpression> j : joins.entrySet()) {
            q.leftJoin(j.getKey()).on(j.getValue());
        }
    }
    // # where
    q.where(r.where.orNull());
    // # paging
    for (Integer a : r.offset) {
        q.offset(a);
    }
    for (Integer a : r.limit) {
        q.limit(a);
    }
    // # order
    for (OrderSpecifier<?> a : r.order) {
        q.orderBy(a);
    }
    // # distinct
    if (!toFetchProperties) {
        // if no properties shall be fetched the result set can be distinct
        q.distinct();
    }
    // # fetch
    // create parameters for fetch clause, i.e. Querydsl's list() method
    final List<Expression<?>> fetch;
    {
        // check if the media package ID needs to be selected separately
        if (r.fetch.exists(MandatoryFetch.exists)) {
            fetch = r.fetch.toList();
        } else {
            fetch = r.fetch.append(MandatoryFetch.fetch).toList();
        }
    }
    // Run the query and transform the result into records
    final Stream<ARecordImpl> records;
    {
        // run query
        am.getDb().logQuery(q);
        final List<Tuple> result = q.list(JpaFns.toExpressionArray(fetch));
        logger.debug("Pure query ms " + (System.nanoTime() - startTime) / 1000000);
        // map result based on the fact whether properties have been fetched or not
        if (!toFetchProperties) {
            // No properties have been fetched -> each result row (tuple) is a distinct record (snapshot).
            records = $($(result).map(toARecord(r))).map(new Fn<ARecordImpl, ARecordImpl>() {

                @Override
                public ARecordImpl apply(ARecordImpl record) {
                    Opt<Snapshot> snapshotOpt = record.getSnapshot();
                    Snapshot snapshot = null;
                    if (snapshotOpt.isSome()) {
                        // make sure the delivered media package has valid URIs
                        snapshot = am.getHttpAssetProvider().prepareForDelivery(snapshotOpt.get());
                    }
                    return new ARecordImpl(record.getSnapshotId(), record.getMediaPackageId(), record.getProperties(), snapshot);
                }
            });
        } else {
            logger.trace("Fetched properties");
            // Properties have been fetched -> there may be multiple rows (tuples) per snapshot because of the join with the property table.
            // Extract records and properties and link them together.
            // group properties after their media package ID and make sure that no duplicate properties occur
            final Map<String, Set<Property>> propertiesPerMp = $(result).bind(toProperty).foldl(new HashMap<String, Set<Property>>(), new Fn2<Map<String, Set<Property>>, Property, Map<String, Set<Property>>>() {

                @Override
                public Map<String, Set<Property>> apply(Map<String, Set<Property>> sum, Property p) {
                    final String mpId = p.getId().getMediaPackageId();
                    final Set<Property> props = sum.get(mpId);
                    if (props != null) {
                        props.add(p);
                    } else {
                        sum.put(mpId, SetB.MH.mk(p));
                    }
                    return sum;
                }
            });
            // group records after their media package ID
            final Map<String, List<ARecordImpl>> distinctRecords = $($(result).map(toARecord(r)).toSet()).groupMulti(ARecordImpl.getMediaPackageId);
            records = $(distinctRecords.values()).bind(new Fn<List<ARecordImpl>, Iterable<ARecordImpl>>() {

                @Override
                public Iterable<ARecordImpl> apply(List<ARecordImpl> records) {
                    return $(records).map(new Fn<ARecordImpl, ARecordImpl>() {

                        @Override
                        public ARecordImpl apply(ARecordImpl record) {
                            final Set<Property> properties = propertiesPerMp.get(record.getMediaPackageId());
                            final Stream<Property> p = properties != null ? $(properties) : Stream.<Property>empty();
                            Snapshot snapshot = null;
                            Opt<Snapshot> snapshotOpt = record.getSnapshot();
                            if (snapshotOpt.isSome()) {
                                // make sure the delivered media package has valid URIs
                                snapshot = am.getHttpAssetProvider().prepareForDelivery(snapshotOpt.get());
                            }
                            return new ARecordImpl(record.getSnapshotId(), record.getMediaPackageId(), p, snapshot);
                        }
                    });
                }
            });
        }
    }
    final long searchTime = (System.nanoTime() - startTime) / 1000000;
    logger.debug("Complete query ms " + searchTime);
    return new AResultImpl(AbstractASelectQuery.<ARecord>vary(records), sizeOf(records), r.offset.getOr(0), r.limit.getOr(-1), searchTime);
}
Also used : Set(java.util.Set) HashMap(java.util.HashMap) Fn2(com.entwinemedia.fn.Fn2) BooleanExpression(com.mysema.query.types.expr.BooleanExpression) Opt(com.entwinemedia.fn.data.Opt) List(java.util.List) Property(org.opencastproject.assetmanager.api.Property) EntityPath(com.mysema.query.types.EntityPath) Fn(com.entwinemedia.fn.Fn) JPAQuery(com.mysema.query.jpa.impl.JPAQuery) Snapshot(org.opencastproject.assetmanager.api.Snapshot) BooleanExpression(com.mysema.query.types.expr.BooleanExpression) Expression(com.mysema.query.types.Expression) HashMap(java.util.HashMap) Map(java.util.Map)

Example 4 with JPAQuery

use of com.mysema.query.jpa.impl.JPAQuery in project dwoss by gg-net.

the class ReportEmo method request.

/**
 * This method search a Report where all parameters are equal to one in the Database, if no is existing in the database a new one will be created and
 * returned.
 * <p>
 * @param name       is the name of the Report
 * @param contractor is the contractor of the Report as {@link TradeName}.
 * @param starting   is the Date where the report is starting
 * @param end        is the Date where the report is ending.
 * @return the founded or the new created Report.
 */
public Report request(String name, TradeName contractor, Date starting, Date end) {
    Report singleResult = new JPAQuery(reportEm).from(report).where(report.name.equalsIgnoreCase(name).and(report.startingDate.eq(starting).and(report.endingDate.eq(end).and(report.type.eq(contractor))))).singleResult(report);
    if (singleResult != null)
        return singleResult;
    Report report = new Report(name, contractor, starting, end);
    reportEm.persist(report);
    return report;
}
Also used : Report(eu.ggnet.dwoss.report.ee.entity.Report) JPAQuery(com.mysema.query.jpa.impl.JPAQuery)

Example 5 with JPAQuery

use of com.mysema.query.jpa.impl.JPAQuery in project dwoss by gg-net.

the class CustomerEao method find.

/**
 * A more specific search based on parameters and possible wildcard handlig.
 * All parameters witch are blank will be ignored. The search itself ist like based, so explicit as implicit wildcards are possible
 * <p>
 * @param companyName    the {@link Company#name} of {@link Customer#companies}
 * @param firstName      the {@link Contact#firstName} of {@link Customer#contacts}
 * @param lastName       the {@link Contact#lastName} of {@link Customer#contacts}
 * @param email          the {@link Communication#identifier} with {@link Communication#type} == {@link Type#EMAIL} of {@link Contact#communications} of
 *                       {@link Customer#contacts}
 * @param appendWildcard if true, adds a '%' wildcard sign to all set parameters
 * @return a list of customers matching the paramters.
 */
public List<Customer> find(String companyName, String firstName, String lastName, String email, boolean appendWildcard) {
    WildCardHelper W = new WildCardHelper(appendWildcard);
    JPAQuery query = new JPAQuery(em).from(customer);
    if (!isBlank(companyName)) {
        query.join(customer.companies, company).on(company.name.lower().like(W.trim(companyName)));
    }
    if (!isBlank(firstName) || !isBlank(lastName)) {
        query.join(customer.contacts, contact).on();
        BooleanExpression on = null;
        if (!isBlank(firstName)) {
            on = contact.firstName.lower().like(W.trim(firstName));
        }
        if (!isBlank(lastName)) {
            BooleanExpression second = contact.lastName.lower().like(W.trim(lastName));
            if (on != null)
                on = on.and(second);
            else
                on = second;
        }
        query.on(on);
    }
    if (!isBlank(email)) {
        query.join(customer.contacts, contact).join(contact.communications, communication).on(communication.type.eq(EMAIL).and(communication.identifier.lower().like(W.trim(email))));
    }
    L.debug("calling query");
    List<Customer> list = query.list(customer);
    L.debug("Query successful wiht {}", list);
    return list;
}
Also used : BooleanExpression(com.mysema.query.types.expr.BooleanExpression) JPAQuery(com.mysema.query.jpa.impl.JPAQuery)

Aggregations

JPAQuery (com.mysema.query.jpa.impl.JPAQuery)5 BooleanExpression (com.mysema.query.types.expr.BooleanExpression)2 Fn (com.entwinemedia.fn.Fn)1 Fn2 (com.entwinemedia.fn.Fn2)1 Opt (com.entwinemedia.fn.data.Opt)1 EntityPath (com.mysema.query.types.EntityPath)1 Expression (com.mysema.query.types.Expression)1 Report (eu.ggnet.dwoss.report.ee.entity.Report)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Map (java.util.Map)1 Set (java.util.Set)1 Property (org.opencastproject.assetmanager.api.Property)1 Snapshot (org.opencastproject.assetmanager.api.Snapshot)1