ipld-eth-server/pkg/watch/api.go

147 lines
4.4 KiB
Go
Raw Normal View History

2019-05-17 06:27:02 +00:00
// VulcanizeDB
// Copyright © 2019 Vulcanize
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
2020-06-30 00:16:52 +00:00
package watch
2019-05-17 06:27:02 +00:00
import (
"context"
2019-06-06 03:50:12 +00:00
"github.com/ethereum/go-ethereum/p2p"
2020-02-23 23:14:29 +00:00
"github.com/ethereum/go-ethereum/rlp"
2019-05-17 06:27:02 +00:00
"github.com/ethereum/go-ethereum/rpc"
log "github.com/sirupsen/logrus"
"github.com/vulcanize/ipfs-blockchain-watcher/pkg/btc"
"github.com/vulcanize/ipfs-blockchain-watcher/pkg/eth"
2020-06-30 00:16:52 +00:00
"github.com/vulcanize/ipfs-blockchain-watcher/pkg/node"
"github.com/vulcanize/ipfs-blockchain-watcher/pkg/shared"
v "github.com/vulcanize/ipfs-blockchain-watcher/version"
2019-05-17 06:27:02 +00:00
)
// APIName is the namespace used for the state diffing service API
const APIName = "vdb"
2019-05-17 06:27:02 +00:00
// APIVersion is the version of the state diffing service API
const APIVersion = "0.0.1"
2020-06-30 00:16:52 +00:00
// PublicWatcherAPI is the public api for the watcher
type PublicWatcherAPI struct {
w Watcher
2019-05-17 06:27:02 +00:00
}
2020-06-30 00:16:52 +00:00
// NewPublicWatcherAPI creates a new PublicWatcherAPI with the provided underlying Watcher process
func NewPublicWatcherAPI(w Watcher) *PublicWatcherAPI {
return &PublicWatcherAPI{
w: w,
2019-05-17 06:27:02 +00:00
}
}
2020-06-30 00:16:52 +00:00
// Stream is the public method to setup a subscription that fires off IPLD payloads as they are processed
func (api *PublicWatcherAPI) Stream(ctx context.Context, rlpParams []byte) (*rpc.Subscription, error) {
2020-02-23 23:14:29 +00:00
var params shared.SubscriptionSettings
2020-06-30 00:16:52 +00:00
switch api.w.Chain() {
2020-02-23 23:14:29 +00:00
case shared.Ethereum:
var ethParams eth.SubscriptionSettings
if err := rlp.DecodeBytes(rlpParams, &ethParams); err != nil {
return nil, err
}
params = &ethParams
case shared.Bitcoin:
var btcParams btc.SubscriptionSettings
if err := rlp.DecodeBytes(rlpParams, &btcParams); err != nil {
return nil, err
}
params = &btcParams
default:
2020-08-06 20:00:41 +00:00
panic("ipfs-blockchain-watcher is not configured for a specific chain type")
2020-02-23 23:14:29 +00:00
}
2019-05-17 06:27:02 +00:00
// ensure that the RPC connection supports subscriptions
notifier, supported := rpc.NotifierFromContext(ctx)
if !supported {
return nil, rpc.ErrNotificationsUnsupported
}
// create subscription and start waiting for stream events
2019-05-17 06:27:02 +00:00
rpcSub := notifier.CreateSubscription()
go func() {
2019-06-06 03:50:12 +00:00
// subscribe to events from the SyncPublishScreenAndServe service
payloadChannel := make(chan SubscriptionPayload, PayloadChanBufferSize)
quitChan := make(chan bool, 1)
2020-06-30 00:16:52 +00:00
go api.w.Subscribe(rpcSub.ID, payloadChannel, quitChan, params)
2019-05-17 06:27:02 +00:00
// loop and await payloads and relay them to the subscriber using notifier
2019-05-17 06:27:02 +00:00
for {
select {
case packet := <-payloadChannel:
if err := notifier.Notify(rpcSub.ID, packet); err != nil {
2020-06-30 00:16:52 +00:00
log.Error("Failed to send watcher data packet", "err", err)
api.w.Unsubscribe(rpcSub.ID)
return
2019-05-17 06:27:02 +00:00
}
case <-rpcSub.Err():
2020-06-30 00:16:52 +00:00
api.w.Unsubscribe(rpcSub.ID)
2019-05-17 06:27:02 +00:00
return
case <-quitChan:
2020-06-30 00:16:52 +00:00
// don't need to unsubscribe from the watcher, the service does so before sending the quit signal this way
2019-05-17 06:27:02 +00:00
return
}
}
}()
return rpcSub, nil
2019-06-06 03:50:12 +00:00
}
2020-06-30 00:16:52 +00:00
// Node is a public rpc method to allow transformers to fetch the node info for the watcher
// NOTE: this is the node info for the node that the watcher is syncing from, not the node info for the watcher itself
func (api *PublicWatcherAPI) Node() *node.Node {
return api.w.Node()
}
2020-06-30 00:16:52 +00:00
// Chain returns the chain type that this watcher instance supports
func (api *PublicWatcherAPI) Chain() shared.ChainType {
return api.w.Chain()
}
2020-06-30 00:16:52 +00:00
// Struct for holding watcher meta data
type InfoAPI struct{}
2020-06-30 00:16:52 +00:00
// NewInfoAPI creates a new InfoAPI
func NewInfoAPI() *InfoAPI {
return &InfoAPI{}
}
// Modules returns modules supported by this api
func (iapi *InfoAPI) Modules() map[string]string {
return map[string]string{
"vdb": "Stream",
}
}
2020-06-30 00:16:52 +00:00
// NodeInfo gathers and returns a collection of metadata for the watcher
func (iapi *InfoAPI) NodeInfo() *p2p.NodeInfo {
return &p2p.NodeInfo{
// TODO: formalize this
ID: "vulcanizeDB",
2020-08-06 20:00:41 +00:00
Name: "ipfs-blockchain-watcher",
}
}
2020-06-30 00:16:52 +00:00
// Version returns the version of the watcher
func (iapi *InfoAPI) Version() string {
return v.VersionWithMeta
}