mirror of
https://github.com/logos-messaging/logos-messaging-go.git
synced 2026-01-07 00:13:13 +00:00
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package publish
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/pb"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/store"
|
|
"github.com/waku-org/go-waku/waku/v2/utils"
|
|
)
|
|
|
|
func NewDefaultStorenodeMessageVerifier(store *store.WakuStore) StorenodeMessageVerifier {
|
|
return &defaultStorenodeMessageVerifier{
|
|
store: store,
|
|
}
|
|
}
|
|
|
|
type defaultStorenodeMessageVerifier struct {
|
|
store *store.WakuStore
|
|
}
|
|
|
|
func (d *defaultStorenodeMessageVerifier) MessageHashesExist(ctx context.Context, requestID []byte, peerInfo peer.AddrInfo, pageSize uint64, messageHashes []pb.MessageHash) ([]pb.MessageHash, error) {
|
|
|
|
addrs := utils.EncapsulatePeerID(peerInfo.ID, peerInfo.Addrs...)
|
|
|
|
var opts []store.RequestOption
|
|
opts = append(opts, store.WithRequestID(requestID))
|
|
opts = append(opts, store.WithPeerAddr(addrs...))
|
|
opts = append(opts, store.WithPaging(false, pageSize))
|
|
opts = append(opts, store.IncludeData(false))
|
|
|
|
response, err := d.store.QueryByHash(ctx, messageHashes, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make([]pb.MessageHash, len(response.Messages()))
|
|
for i, msg := range response.Messages() {
|
|
result[i] = msg.WakuMessageHash()
|
|
}
|
|
|
|
return result, nil
|
|
}
|