status-go/protocol/peersyncing/peersyncing.go

package peersyncing

import "github.com/status-im/status-go/eth-node/types"

type PeerSyncing struct {
	persistence SyncMessagePersistence
	config      Config
}

func New(config Config) *PeerSyncing {
	syncMessagePersistence := config.SyncMessagePersistence
	if syncMessagePersistence == nil {
		syncMessagePersistence = NewSyncMessageSQLitePersistence(config.Database)
	}

	return &PeerSyncing{
		config:      config,
		persistence: syncMessagePersistence,
	}
}

func (p *PeerSyncing) Add(message SyncMessage) error {
	return p.persistence.Add(message)
}

func (p *PeerSyncing) AvailableMessages() ([]SyncMessage, error) {
	return p.persistence.All()
}

func (p *PeerSyncing) AvailableMessagesMapByChatIDs(groupIDs [][]byte, limit int) (map[string][][]byte, error) {
	availableMessages, err := p.persistence.ByChatIDs(groupIDs, limit)
	if err != nil {
		return nil, err
	}
	availableMessagesMap := make(map[string][][]byte)
	for _, m := range availableMessages {
		chatID := types.Bytes2Hex(m.ChatID)
		availableMessagesMap[chatID] = append(availableMessagesMap[chatID], m.ID)
	}
	return availableMessagesMap, err
}

func (p *PeerSyncing) MessagesByIDs(messageIDs [][]byte) ([]SyncMessage, error) {
	return p.persistence.ByMessageIDs(messageIDs)
}

func (p *PeerSyncing) OnOffer(messages []SyncMessage) ([]SyncMessage, error) {
	return p.persistence.Complement(messages)
}