check_pier.go

package rpcx

import (
	"context"
	"fmt"

	"github.com/meshplus/bitxhub-model/pb"
)

func (cli *ChainClient) CheckMasterPier(address string) (*pb.Response, error) {
	ctx, cancel := context.WithTimeout(context.Background(), CheckPierTimeout)
	defer cancel()

	ctx, err := cli.SetCtxMetadata(ctx)
	if err != nil {
		return nil, fmt.Errorf("set ctx metadata err: %v", err)
	}

	grpcClient, err := cli.pool.getClient()
	if err != nil {
		return nil, err
	}
	defer func() {
		if err := grpcClient.conn.Close(); err != nil {
			cli.logger.Errorf("close conn err: %s", err)
		}
	}()
	request := &pb.Address{
		Address: address,
	}
	return grpcClient.broker.CheckMasterPier(ctx, request)
}

func (cli *ChainClient) SetMasterPier(address string, index string, timeout int64) (*pb.Response, error) {
	ctx, cancel := context.WithTimeout(context.Background(), CheckPierTimeout)
	defer cancel()

	ctx, err := cli.SetCtxMetadata(ctx)
	if err != nil {
		return nil, fmt.Errorf("set ctx metadata err: %v", err)
	}

	grpcClient, err := cli.pool.getClient()
	if err != nil {
		return nil, err
	}
	defer func() {
		if err := grpcClient.conn.Close(); err != nil {
			cli.logger.Errorf("close conn err: %s", err)
		}
	}()
	request := &pb.PierInfo{
		Address: address,
		Index:   index,
		Timeout: timeout,
	}
	return grpcClient.broker.SetMasterPier(ctx, request)
}

func (cli *ChainClient) HeartBeat(address string, index string) (*pb.Response, error) {
	ctx, cancel := context.WithTimeout(context.Background(), CheckPierTimeout)
	defer cancel()

	ctx, err := cli.SetCtxMetadata(ctx)
	if err != nil {
		return nil, fmt.Errorf("set ctx metadata err: %v", err)
	}

	grpcClient, err := cli.pool.getClient()
	if err != nil {
		return nil, err
	}
	defer func() {
		if err := grpcClient.conn.Close(); err != nil {
			cli.logger.Errorf("close conn err: %s", err)
		}
	}()
	request := &pb.PierInfo{
		Address: address,
		Index:   index,
	}
	return grpcClient.broker.HeartBeat(ctx, request)
}