90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
package hapi
|
|
|
|
import (
|
|
"context"
|
|
"github.com/filecoin-project/lotus/api/client"
|
|
cliutil "github.com/filecoin-project/lotus/cli/util"
|
|
"time"
|
|
)
|
|
|
|
func (a *app) watchRpc() {
|
|
ticker := time.NewTicker(watchInterval)
|
|
for {
|
|
err := a.updateRpc(context.TODO())
|
|
if err != nil {
|
|
log.Errorw("updating rpc info", "error", err)
|
|
}
|
|
select {
|
|
case <-ticker.C:
|
|
}
|
|
}
|
|
}
|
|
|
|
type minimalApiInfo struct {
|
|
Apis struct {
|
|
ChainApiInfo []string
|
|
}
|
|
}
|
|
|
|
func (a *app) updateRpc(ctx context.Context) error {
|
|
rpcInfos := map[string]minimalApiInfo{} // config name -> api info
|
|
confNameToAddr := map[string]string{} // config name -> api address
|
|
|
|
err := forEachConfig[minimalApiInfo](a, func(name string, info minimalApiInfo) error {
|
|
if len(info.Apis.ChainApiInfo) == 0 {
|
|
return nil
|
|
}
|
|
|
|
rpcInfos[name] = info
|
|
|
|
for _, addr := range info.Apis.ChainApiInfo {
|
|
ai := cliutil.ParseApiInfo(addr)
|
|
confNameToAddr[name] = ai.Addr
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
apiInfos := map[string][]byte{} // api address -> token
|
|
|
|
// for dedup by address
|
|
for _, info := range rpcInfos {
|
|
ai := cliutil.ParseApiInfo(info.Apis.ChainApiInfo[0])
|
|
apiInfos[ai.Addr] = ai.Token
|
|
}
|
|
|
|
a.rpcInfoLk.Lock()
|
|
|
|
// todo improve this shared rpc logic
|
|
if a.workingApi == nil {
|
|
for addr, token := range apiInfos {
|
|
ai := cliutil.APIInfo{
|
|
Addr: addr,
|
|
Token: token,
|
|
}
|
|
|
|
da, err := ai.DialArgs("v1")
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
ah := ai.AuthHeader()
|
|
|
|
v1api, closer, err := client.NewFullNodeRPCV1(ctx, da, ah)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
_ = closer // todo
|
|
|
|
a.workingApi = v1api
|
|
}
|
|
}
|
|
|
|
a.rpcInfoLk.Unlock()
|
|
|
|
return nil
|
|
}
|