121 lines
3.1 KiB
Go
121 lines
3.1 KiB
Go
|
package tari
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/base64"
|
|||
|
|
"encoding/hex"
|
|||
|
|
"fmt"
|
|||
|
|
base_node "pool/internal/gbt/tari/proto"
|
|||
|
|
block "pool/internal/gbt/tari/proto/block"
|
|||
|
|
|
|||
|
|
"google.golang.org/grpc"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// var PowAlgoMap = []string{"randomxM", "sha3x", "randomxT"}
|
|||
|
|
|
|||
|
|
type BaseNodeClient struct {
|
|||
|
|
Conn *grpc.ClientConn
|
|||
|
|
Client base_node.BaseNodeClient
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建BaseNode客户端
|
|||
|
|
func NewBaseNodeClient(address string) (*BaseNodeClient, error) {
|
|||
|
|
conn, err := grpc.Dial(address, grpc.WithInsecure())
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to connect to base node: %v", err)
|
|||
|
|
}
|
|||
|
|
// 移除 defer conn.Close()，让连接保持打开状态
|
|||
|
|
|
|||
|
|
client := base_node.NewBaseNodeClient(conn)
|
|||
|
|
|
|||
|
|
return &BaseNodeClient{
|
|||
|
|
Conn: conn,
|
|||
|
|
Client: client,
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Close 关闭GRPC连接
|
|||
|
|
func (c *BaseNodeClient) Close() error {
|
|||
|
|
if c.Conn != nil {
|
|||
|
|
return c.Conn.Close()
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type TariBlockTemplate struct {
|
|||
|
|
MiningHash string
|
|||
|
|
Header string
|
|||
|
|
Body string
|
|||
|
|
Height uint64
|
|||
|
|
TargetDifficulty uint64
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func base64ToHex(data string) (string, error) {
|
|||
|
|
result, err := base64.StdEncoding.DecodeString(data)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
return hex.EncodeToString(result), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (nc *BaseNodeClient) GetBlockTemplate(ctx context.Context, powIndex uint32) (TariBlockTemplate, error) {
|
|||
|
|
var pow_algo block.PowAlgo_PowAlgos
|
|||
|
|
switch powIndex {
|
|||
|
|
case 0:
|
|||
|
|
pow_algo = block.PowAlgo_POW_ALGOS_RANDOMXM
|
|||
|
|
case 1:
|
|||
|
|
pow_algo = block.PowAlgo_POW_ALGOS_SHA3X
|
|||
|
|
case 2:
|
|||
|
|
pow_algo = block.PowAlgo_POW_ALGOS_RANDOMXT
|
|||
|
|
default:
|
|||
|
|
fmt.Println("不支持", powIndex, "算法")
|
|||
|
|
}
|
|||
|
|
templateReq := &base_node.NewBlockTemplateRequest{
|
|||
|
|
Algo: &block.PowAlgo{
|
|||
|
|
PowAlgo: pow_algo,
|
|||
|
|
},
|
|||
|
|
MaxWeight: 0x00,
|
|||
|
|
}
|
|||
|
|
templateResp, err := nc.Client.GetNewBlockTemplate(ctx, templateReq)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(pow_algo, "获取区块模板失败：", err)
|
|||
|
|
return TariBlockTemplate{}, err
|
|||
|
|
}
|
|||
|
|
// 节点未同步完成
|
|||
|
|
if !templateResp.InitialSyncAchieved {
|
|||
|
|
fmt.Println("节点还未完成同步")
|
|||
|
|
return TariBlockTemplate{}, err
|
|||
|
|
}
|
|||
|
|
blk_template, miner_data := templateResp.NewBlockTemplate, templateResp.MinerData
|
|||
|
|
targetDifficulty := miner_data.TargetDifficulty
|
|||
|
|
height := blk_template.Header.Height
|
|||
|
|
blockBlob, err := nc.Client.GetNewBlockBlob(ctx, blk_template)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("获得block blob失败：", err)
|
|||
|
|
return TariBlockTemplate{}, err
|
|||
|
|
}
|
|||
|
|
mining_hash, header, body := blockBlob.BlockHash, blockBlob.Header, blockBlob.BlockBody
|
|||
|
|
var t_bt = TariBlockTemplate{
|
|||
|
|
MiningHash: hex.EncodeToString(mining_hash),
|
|||
|
|
Header: hex.EncodeToString(header),
|
|||
|
|
Body: hex.EncodeToString(body),
|
|||
|
|
TargetDifficulty: targetDifficulty,
|
|||
|
|
Height: height,
|
|||
|
|
}
|
|||
|
|
return t_bt, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (nc *BaseNodeClient) SubmitBlock(ctx context.Context, header, body []byte) ([]byte, error) {
|
|||
|
|
submitReq := &base_node.BlockBlobRequest{
|
|||
|
|
HeaderBlob: header,
|
|||
|
|
BodyBlob: body,
|
|||
|
|
}
|
|||
|
|
submitResp, err := nc.Client.SubmitBlockBlob(ctx, submitReq)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("提交区块失败：", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
blockHash := submitResp.BlockHash
|
|||
|
|
return blockHash, nil
|
|||
|
|
}
|