MsgIndexRecord.java

浏览该文件的文档.

package com.netease.nimlib.sdk.search.model;

import com.netease.nimlib.sdk.msg.constant.SessionTypeEnum;
import com.netease.nimlib.sdk.msg.model.IMMessage;
import com.netease.nimlib.search.helper.MsgIndexHelper;
import com.netease.nimlib.search.highlight.QueryHit;
import com.netease.nimlib.search.model.NIMIndexRecord;
import com.netease.nimlib.session.MsgDBHelper;

import java.util.ArrayList;
import java.util.List;

public class MsgIndexRecord implements Comparable<MsgIndexRecord> {

    private final NIMIndexRecord record;

    private IMMessage message;

    private final String query;

    private List<RecordHitInfo> hitInfo;

    public MsgIndexRecord(NIMIndexRecord record, String query) {
        this.record = record;
        this.query = query;
    }

    public String getText() {
        return record.content;
    }

    public SessionTypeEnum getSessionType() {
        return MsgIndexHelper.getSessionTypeFromNIMRecordId(record.id);
    }

    public String getSessionId() {
        return MsgIndexHelper.getSessionIdFromNIMRecordId(record.id);
    }

    public long getTime() {
        return record.time;
    }

    public int getCount() {
        return record.count;
    }

    public String getQuery() {
        return query;
    }

    public NIMIndexRecord getRecord() {
        return record;
    }

    public IMMessage getMessage() {
        if (message == null) {
            message = MsgDBHelper.queryMessageBySeqId(record.dataid);
        }

        return message;
    }

    public List<RecordHitInfo> getHitInfo() {
        if (hitInfo == null) {
            ArrayList<QueryHit.HitInfo> hits = QueryHit.getInstance().hit(getText(), query, true, false);
            if (hits == null || hits.isEmpty()) {
                // 特殊情况下，搜索结果在短语高亮失效的情况下换成单个字符匹配高亮
                hits = QueryHit.getInstance().hit(getText(), query, true, true);
            }

            hitInfo = new ArrayList<>(hits.size());
            for (QueryHit.HitInfo h : hits) {
                this.hitInfo.add(new RecordHitInfo(h.start, h.end));
            }
        }

        return hitInfo;
    }

    public List<RecordHitInfo> cloneHitInfo() {
        getHitInfo();

        List<RecordHitInfo> clone = new ArrayList<>(hitInfo.size());
        for (RecordHitInfo h : hitInfo) {
            clone.add(new RecordHitInfo(h.start, h.end));
        }

        return clone;
    }

    @Override
    public int compareTo(MsgIndexRecord msgIndexInfo) {
        return (int) (msgIndexInfo.record.time - record.time);
    }

    @Override
    public String toString() {
        return record.toString();
    }
}