Examples with QueryResultPacket com.yahoo.fs4.QueryResultPacket used on opensource projects

Example 21 with QueryResultPacket

use of com.yahoo.fs4.QueryResultPacket in project vespa by vespa-engine.

the class PacketWrapperTestCase method testPacketTrimming13.

@Test
public void testPacketTrimming13() {
    CacheKey key = new CacheKey(QueryPacket.create(new Query("/?query=key")));
    PacketWrapper w = createResult(key, 0, 10, 100);
    QueryResultPacket q = createQueryResultPacket(4, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(12, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(16, 10, 100);
    w.addResultPacket(q);
    assertEquals(4, w.getResultPackets().size());
    q = createQueryResultPacket(11, 10, 100);
    w.addResultPacket(q);
    assertEquals(4, w.getResultPackets().size());
    List<?> l = w.getResultPackets();
    assertEquals(0, ((QueryResultPacket) l.get(0)).getOffset());
    assertEquals(4, ((QueryResultPacket) l.get(1)).getOffset());
    assertEquals(12, ((QueryResultPacket) l.get(2)).getOffset());
    assertEquals(16, ((QueryResultPacket) l.get(3)).getOffset());
}

Example 22 with QueryResultPacket

use of com.yahoo.fs4.QueryResultPacket in project vespa by vespa-engine.

the class PacketWrapperTestCase method createQueryResultPacket.

private QueryResultPacket createQueryResultPacket(int offset, int hits, int total) {
    QueryResultPacket r = QueryResultPacket.create();
    r.setDocstamp(1);
    r.setChannel(0);
    r.setTotalDocumentCount(total);
    r.setOffset(offset);
    for (int i = 0; i < hits && i < total; ++i) {
        r.addDocument(new DocumentInfo(DocsumDefinitionTestCase.createGlobalId(offset + i), 1000 - offset - i, 1, 1));
    }
    return r;
}

Example 23 with QueryResultPacket

use of com.yahoo.fs4.QueryResultPacket in project vespa by vespa-engine.

the class PacketWrapperTestCase method testPacketTrimming7.

@Test
public void testPacketTrimming7() {
    final Query query = new Query("/?query=key");
    query.setWindow(50, 10);
    CacheKey key = new CacheKey(QueryPacket.create(query));
    PacketWrapper w = createResult(key, 50, 10, 100);
    QueryResultPacket q = createQueryResultPacket(10, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(40, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(30, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(20, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(0, 10, 100);
    w.addResultPacket(q);
    assertEquals(6, w.getResultPackets().size());
    List<?> l = w.getResultPackets();
    assertEquals(0, ((QueryResultPacket) l.get(0)).getOffset());
    assertEquals(10, ((QueryResultPacket) l.get(1)).getOffset());
    assertEquals(20, ((QueryResultPacket) l.get(2)).getOffset());
    assertEquals(30, ((QueryResultPacket) l.get(3)).getOffset());
    assertEquals(40, ((QueryResultPacket) l.get(4)).getOffset());
    assertEquals(50, ((QueryResultPacket) l.get(5)).getOffset());
}

Example 24 with QueryResultPacket

use of com.yahoo.fs4.QueryResultPacket in project vespa by vespa-engine.

the class PacketWrapperTestCase method testPacketTrimming14.

@Test
public void testPacketTrimming14() {
    CacheKey key = new CacheKey(QueryPacket.create(new Query("/?query=key")));
    PacketWrapper w = createResult(key, 0, 10, 100);
    QueryResultPacket q = createQueryResultPacket(4, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(12, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(16, 10, 100);
    w.addResultPacket(q);
    assertEquals(4, w.getResultPackets().size());
    q = createQueryResultPacket(5, 6, 100);
    w.addResultPacket(q);
    assertEquals(4, w.getResultPackets().size());
    List<?> l = w.getResultPackets();
    assertEquals(0, ((QueryResultPacket) l.get(0)).getOffset());
    assertEquals(4, ((QueryResultPacket) l.get(1)).getOffset());
    assertEquals(12, ((QueryResultPacket) l.get(2)).getOffset());
    assertEquals(16, ((QueryResultPacket) l.get(3)).getOffset());
}

Example 25 with QueryResultPacket

use of com.yahoo.fs4.QueryResultPacket in project vespa by vespa-engine.

the class PacketWrapperTestCase method testPacketTrimming5.

@Test
public void testPacketTrimming5() {
    CacheKey key = new CacheKey(QueryPacket.create(new Query("/?query=key")));
    PacketWrapper w = createResult(key, 0, 10, 100);
    QueryResultPacket q = createQueryResultPacket(5, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(10, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(15, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(15, 85, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(20, 10, 100);
    w.addResultPacket(q);
    q = createQueryResultPacket(25, 10, 100);
    w.addResultPacket(q);
    assertEquals(3, w.getResultPackets().size());
    List<?> l = w.getResultPackets();
    assertEquals(0, ((QueryResultPacket) l.get(0)).getOffset());
    assertEquals(10, ((QueryResultPacket) l.get(1)).getOffset());
    assertEquals(15, ((QueryResultPacket) l.get(2)).getOffset());
}