cfx_api.go

package virtualfilter

import (
	"github.com/Conflux-Chain/confura/util"
	rpcutil "github.com/Conflux-Chain/confura/util/rpc"
	sdk "github.com/Conflux-Chain/go-conflux-sdk"
	"github.com/Conflux-Chain/go-conflux-sdk/types"
	w3rpc "github.com/openweb3/go-rpc-provider"
	"github.com/sirupsen/logrus"
)

// core space filter API

type cfxFilterApi struct {
	fs        *cfxFilterSystem   // filter system
	fnClients util.ConcurrentMap // full node clients: node name => sdk client
}

func newCfxFilterApi(sys *cfxFilterSystem) *cfxFilterApi {
	return &cfxFilterApi{fs: sys}
}

func (api *cfxFilterApi) NewBlockFilter(nodeUrl string) (w3rpc.ID, error) {
	client, err := api.loadOrGetFnClient(nodeUrl)
	if err != nil {
		return nilRpcId, err
	}

	return api.fs.newBlockFilter(client)
}

func (api *cfxFilterApi) NewPendingTransactionFilter(nodeUrl string) (w3rpc.ID, error) {
	client, err := api.loadOrGetFnClient(nodeUrl)
	if err != nil {
		return nilRpcId, err
	}

	return api.fs.newPendingTransactionFilter(client)
}

func (api *cfxFilterApi) UninstallFilter(id w3rpc.ID) (bool, error) {
	return api.fs.uninstallFilter(id)
}

func (api *cfxFilterApi) NewFilter(nodeUrl string, crit types.LogFilter) (w3rpc.ID, error) {
	client, err := api.loadOrGetFnClient(nodeUrl)
	if err != nil {
		return nilRpcId, err
	}

	return api.fs.newFilter(client, crit)
}

func (api *cfxFilterApi) GetLogFilter(fid w3rpc.ID) (*types.LogFilter, error) {
	vf, ok := api.fs.getFilter(fid)
	if !ok || vf.ftype() != filterTypeLog {
		return nil, errFilterNotFound
	}

	cfxf := vf.(*cfxLogFilter)
	return &cfxf.crit, nil
}

func (api *cfxFilterApi) GetFilterChanges(id w3rpc.ID) (*types.CfxFilterChanges, error) {
	return api.fs.getFilterChanges(id)
}

func (api *cfxFilterApi) loadOrGetFnClient(nodeUrl string) (*sdk.Client, error) {
	nodeName := rpcutil.Url2NodeName(nodeUrl)
	client, _, err := api.fnClients.LoadOrStoreFnErr(nodeName, func(interface{}) (interface{}, error) {
		client, err := rpcutil.NewCfxClient(nodeUrl, rpcutil.WithClientHookMetrics(true))
		if err != nil {
			logrus.WithField("fnNodeUrl", nodeUrl).
				WithError(err).
				Error("Failed to new cfx client for virtual filter")
			return nil, err
		}

		return client, nil
	})

	if err != nil {
		return nil, err
	}

	return client.(*sdk.Client), nil
}