Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
zhouop0 committed Jan 24, 2024
1 parent 68e2392 commit e508a8b
Show file tree
Hide file tree
Showing 6 changed files with 22 additions and 26 deletions.
9 changes: 5 additions & 4 deletions internal/handler/checkstatus.go
Original file line number Diff line number Diff line change
@@ -1,18 +1,19 @@
package handler

import (
"time"

"github.com/b2network/b2committer/internal/schema"
"github.com/b2network/b2committer/internal/svc"
"github.com/b2network/b2committer/pkg/log"
"github.com/pkg/errors"
"time"
)

// CheckStatus check proposal vote status
func CheckStatus(ctx *svc.ServiceContext) {
for {
var dbProposal schema.Proposal
err := ctx.DB.Where("status=?", schema.Voting_Status).Order("end_batch_num asc").First(&dbProposal).Error
err := ctx.DB.Where("status=?", schema.VotingStatus).Order("end_batch_num asc").First(&dbProposal).Error
if err != nil {
log.Errorf("[Handler.CheckStatus] find Voting proposal err: %s\n", errors.WithStack(err))
time.Sleep(5 * time.Second)
Expand All @@ -23,7 +24,7 @@ func CheckStatus(ctx *svc.ServiceContext) {
log.Errorf("[Handler.CheckStatus] QueryProposalByID err: %s\n", errors.WithStack(err))
continue
}
if proposal.Status != schema.Voting_Status && proposal.Winner != "" {
if proposal.Status != schema.VotingStatus && proposal.Winner != "" {
dbProposal.Winner = proposal.Winner
dbProposal.Status = proposal.Status
ctx.DB.Save(dbProposal)
Expand All @@ -35,7 +36,7 @@ func CheckStatus(ctx *svc.ServiceContext) {
func CheckStatusTimeOut(ctx *svc.ServiceContext) {
for {
var dbProposal schema.Proposal
err := ctx.DB.Where("status!=?", schema.Succeed_Status).Order("end_batch_num asc").First(&dbProposal).Error
err := ctx.DB.Where("status!=?", schema.SucceedStatus).Order("end_batch_num asc").First(&dbProposal).Error
if err != nil {
log.Errorf("[Handler.CheckStatusTimeOut] find Voting proposal err: %s\n", errors.WithStack(err))
time.Sleep(5 * time.Second)
Expand Down
2 changes: 1 addition & 1 deletion internal/handler/committer.go
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ func committerProposal(ctx *svc.ServiceContext, verifyBatchInfo *VerifyRangBatch
},
EndBatchNum: verifyBatchInfo.endBatchNum,
ProposalID: proposalID,
Status: schema.Voting_Status,
Status: schema.VotingStatus,
StateRootHash: verifyBatchInfo.stateRootHash,
ProofRootHash: verifyBatchInfo.proofRootHash,
StartBatchNum: verifyBatchInfo.startBatchNum,
Expand Down
19 changes: 6 additions & 13 deletions internal/handler/inscribe.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@ package handler

import (
"encoding/json"
"github.com/b2network/b2committer/pkg/inscribe"
"time"

"github.com/b2network/b2committer/pkg/inscribe"

"github.com/b2network/b2committer/pkg/log"

"github.com/b2network/b2committer/internal/schema"
Expand All @@ -17,7 +18,7 @@ import (
func Inscribe(ctx *svc.ServiceContext) {
for {
var dbProposal schema.Proposal
err := ctx.DB.Where("status=?", schema.Pending_Status).Order("end_batch_num asc").First(&dbProposal).Error
err := ctx.DB.Where("status=?", schema.PendingStatus).Order("end_batch_num asc").First(&dbProposal).Error
if err != nil {
log.Errorf("[Handler.Inscribe] Pending and timeout proposal err: %s\n", errors.WithStack(err))
time.Sleep(5 * time.Second)
Expand All @@ -28,13 +29,13 @@ func Inscribe(ctx *svc.ServiceContext) {
log.Errorf("[CheckProposalPending] QueryProposalByID err: %s\n", errors.WithStack(err))
continue
}
if proposal.Status == schema.Succeed_Status {
if proposal.Status == schema.SucceedStatus {
dbProposal.Status = proposal.Status
dbProposal.Winner = proposal.Winner
dbProposal.BitcoinRevealTxHash = proposal.BitcoinTxHash
ctx.DB.Save(dbProposal)
}
if proposal.Status == schema.Pending_Status &&
if proposal.Status == schema.PendingStatus &&
proposal.Winner == ctx.B2NodeConfig.Address && proposal.BitcoinTxHash == "" {
rs, err := inscribe.Inscribe(ctx.BTCConfig.PrivateKey, proposal.StateRootHash, proposal.ProofHash, ctx.BTCConfig.DestinationAddress, ChainParams(ctx.BTCConfig.NetworkName))
if err != nil {
Expand All @@ -58,9 +59,8 @@ func Inscribe(ctx *svc.ServiceContext) {
dbProposal.BitcoinCommitTxHash = rs.CommitTxHash.String()

ctx.DB.Save(dbProposal)

}
if proposal.Status == schema.Pending_Status && proposal.BitcoinTxHash != "" {
if proposal.Status == schema.PendingStatus && proposal.BitcoinTxHash != "" {
_, err = ctx.NodeClient.BitcoinTx(proposal.Id, proposal.Proposer, proposal.BitcoinTxHash)
if err != nil {
log.Errorf("[Handler.Inscribe] BitcoinTx err: %s\n", errors.WithStack(err))
Expand All @@ -70,13 +70,6 @@ func Inscribe(ctx *svc.ServiceContext) {
}
}

// sendBTCTxHashToB2Node step7
func sendBTCTxHashToB2Node(txHash string) {
// TODO send revealTxHash to b2node
// update local proposal btcTxHash
println(txHash)
}

func ChainParams(network string) *chaincfg.Params {
switch network {
case chaincfg.MainNetParams.Name:
Expand Down
8 changes: 4 additions & 4 deletions internal/schema/proposal.go
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
package schema

const (
Voting_Status = 0
Pending_Status = 1
Succeed_Status = 2
Timeout_Status = 3
VotingStatus = 0
PendingStatus = 1
SucceedStatus = 2
TimeoutStatus = 3
)

type Proposal struct {
Expand Down
3 changes: 2 additions & 1 deletion internal/svc/svc.go
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
package svc

import (
"github.com/b2network/b2committer/pkg/b2node"
"time"

"github.com/b2network/b2committer/pkg/b2node"

"github.com/b2network/b2committer/pkg/log"

"github.com/b2network/b2committer/internal/types"
Expand Down
7 changes: 4 additions & 3 deletions pkg/b2node/b2node.go
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ func (n NodeClient) GetAccountInfo(address string) (*eTypes.EthAccount, error) {
}

func (n NodeClient) SubmitProof(id uint64, from string, proofHash string, stateRootHash string,
startIndex uint64, endIndex uint64) (uint64, error) {
startIndex uint64, endIndex uint64,
) (uint64, error) {
msg := committerTypes.NewMsgSubmitProof(id, from, proofHash, stateRootHash, startIndex, endIndex)
msgResponse, err := n.broadcastTx(msg)
if err != nil {
Expand All @@ -72,7 +73,7 @@ func (n NodeClient) SubmitProof(id uint64, from string, proofHash string, stateR
if err != nil {
return 0, fmt.Errorf("[SubmitProof][hex.DecodeString] err: %s", err)
}
var pbMsg = &sdk.TxMsgData{}
pbMsg := &sdk.TxMsgData{}
err = pbMsg.Unmarshal(byteData)
if err != nil {
return 0, fmt.Errorf("[SubmitProof][pbMsg.Unmarshal] err: %s", err)
Expand All @@ -97,7 +98,7 @@ func (n NodeClient) BitcoinTx(proposalID uint64, from string, bitcoinTxHash stri
if err != nil {
return 0, fmt.Errorf("[BitcoinTx][hex.DecodeString] err: %s", err)
}
var pbMsg = &sdk.TxMsgData{}
pbMsg := &sdk.TxMsgData{}
err = pbMsg.Unmarshal(byteData)
if err != nil {
return 0, fmt.Errorf("[BitcoinTx][pbMsg.Unmarshal] err: %s", err)
Expand Down

0 comments on commit e508a8b

Please sign in to comment.