message_repo.go

package repo

import (
	"time"

	"github.com/ipfs/go-cid"

	"github.com/filecoin-project/go-address"
	venustypes "github.com/filecoin-project/venus/venus-shared/types"

	"github.com/filecoin-project/go-state-types/abi"
	types "github.com/filecoin-project/venus/venus-shared/types/messager"
)

type MsgQueryParams = types.MsgQueryParams

type MessageRepo interface {
	ExpireMessage(msg []*types.Message) error
	BatchSaveMessage(msg []*types.Message) error
	CreateMessage(msg *types.Message) error
	UpdateMessage(msg *types.Message) error
	UpdateMessageByState(msg *types.Message, state types.MessageState) error

	GetMessageByFromAndNonce(from address.Address, nonce uint64) (*types.Message, error)
	GetMessageByFromNonceAndState(from address.Address, nonce uint64, state types.MessageState) (*types.Message, error)
	GetMessageByUid(id string) (*types.Message, error)
	HasMessageByUid(id string) (bool, error)
	GetMessageState(id string) (types.MessageState, error)
	GetMessageByCid(unsignedCid cid.Cid) (*types.Message, error)
	GetMessageBySignedCid(signedCid cid.Cid) (*types.Message, error)
	GetSignedMessageByTime(start time.Time) ([]*types.Message, error)
	GetSignedMessageByHeight(height abi.ChainEpoch) ([]*types.Message, error)
	GetSignedMessageFromFailedMsg(addr address.Address) ([]*types.Message, error)

	ListMessage() ([]*types.Message, error)
	ListMessageByFromState(from address.Address, state types.MessageState, isAsc bool, pageIndex, pageSize int, d time.Duration) ([]*types.Message, error)
	ListMessageByAddress(addr address.Address) ([]*types.Message, error)
	ListFailedMessage(*MsgQueryParams) ([]*types.Message, error)
	// ListBlockedMessage returns filled messages and unfill messages
	ListBlockedMessage(p *MsgQueryParams, d time.Duration) ([]*types.Message, error)
	ListUnChainMessageByAddress(addr address.Address, topN int) ([]*types.Message, error)
	ListFilledMessageByAddress(addr address.Address) ([]*types.Message, error)
	ListChainMessageByHeight(height abi.ChainEpoch) ([]*types.Message, error)
	ListUnFilledMessage(addr address.Address) ([]*types.Message, error)
	ListSignedMsgs() ([]*types.Message, error)
	ListFilledMessageBelowNonce(addr address.Address, nonce uint64) ([]*types.Message, error)
	ListMessageByParams(*MsgQueryParams) ([]*types.Message, error)

	UpdateMessageInfoByCid(unsignedCid string, receipt *venustypes.MessageReceipt, height abi.ChainEpoch, state types.MessageState, tsKey venustypes.TipSetKey) error
	UpdateMessageStateByCid(unsignedCid string, state types.MessageState) error
	UpdateMessageStateByID(id string, state types.MessageState) error
	MarkBadMessage(id string) error
	UpdateErrMsg(id string, errMsg string) error
}