aboutsummaryrefslogtreecommitdiffstats
path: root/vdslib/src/main/java/com/yahoo/vdslib/DocumentSummary.java
blob: ed879c4dda228dfba3e856c652d8f26f62be8402 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vdslib;

import com.yahoo.vespa.objects.BufferSerializer;
import com.yahoo.vespa.objects.Deserializer;

import java.nio.ByteOrder;

import static java.nio.charset.StandardCharsets.UTF_8;

public class DocumentSummary {

    private final Summary[] summaries;

    public DocumentSummary(Deserializer buf) {
        BufferSerializer bser = (BufferSerializer) buf; // This is a trick. This should be done in a different way.
        bser.order(ByteOrder.BIG_ENDIAN);
        buf.getInt(null); // legacy - ignored
        int numSummaries = buf.getInt(null);
        summaries = new Summary[numSummaries];
        if (numSummaries > 0) {
            int summaryBufferSize = buf.getInt(null);

            byte[] cArr = bser.getBuf().array();
            int start = bser.getBuf().arrayOffset() + bser.position();
            bser.position(bser.position() + summaryBufferSize);
            for(int i=0; i < numSummaries; i++) {
                int summarySize = buf.getInt(null);
                int end = start;
                while (cArr[end++] != 0);
                byte [] sb = new byte [summarySize];
                System.arraycopy(cArr, end, sb, 0, summarySize);
                summaries[i] = new Summary(new String(cArr, start, end-start-1, UTF_8), sb);
                start = end + summarySize;
            }
        }
    }

    final public int getSummaryCount()         { return summaries.length; }
    final public Summary getSummary(int hitNo) { return summaries[hitNo]; }

    public static class Summary implements Comparable<Summary> {

        private final String  docId;
        private byte[] summary;

        private Summary(String docId) {
            this.docId = docId;
        }

        public Summary(String docId, byte [] summary) {
            this(docId);
            this.summary = summary;
        }

        final public String getDocId()           { return docId; }
        final public byte [] getSummary()        { return summary; }

        public int compareTo(Summary s) {
            return getDocId().compareTo(s.getDocId());
        }

    }

}