waku-org/go-waku

package store

import (
    "github.com/waku-org/go-waku/waku/v2/protocol"
    wpb "github.com/waku-org/go-waku/waku/v2/protocol/pb"
    "github.com/waku-org/go-waku/waku/v2/protocol/store/pb"
    "google.golang.org/protobuf/proto"
)

type Criteria interface {
    PopulateStoreRequest(request *pb.StoreQueryRequest)
}

type FilterCriteria struct {
    protocol.ContentFilter
    TimeStart *int64
    TimeEnd   *int64
}

func (f FilterCriteria) PopulateStoreRequest(request *pb.StoreQueryRequest) {
    request.ContentTopics = f.ContentTopicsList()
    request.PubsubTopic = proto.String(f.PubsubTopic)
    request.TimeStart = f.TimeStart
    request.TimeEnd = f.TimeEnd
}

type MessageHashCriteria struct {
    MessageHashes []wpb.MessageHash
}

func (m MessageHashCriteria) PopulateStoreRequest(request *pb.StoreQueryRequest) {
    request.MessageHashes = make([][]byte, len(m.MessageHashes))
    for i := range m.MessageHashes {
        request.MessageHashes[i] = m.MessageHashes[i][:]
    }
}