add sbp reward calc

Author: viteshan

Diff for: gvite-tools/go.mod

@@ -9,6 +9,7 @@ require (
 	github.com/go-stack/stack v1.8.1 // indirect
 	github.com/kr/text v0.2.0 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect
+	github.com/pkg/errors v0.9.1
 	github.com/shirou/gopsutil v3.21.8+incompatible // indirect
 	github.com/tklauser/go-sysconf v0.3.9 // indirect
 	github.com/tyler-smith/go-bip39 v1.1.0

Diff for: gvite-tools/go.sum

@@ -16,7 +16,6 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/deckarep/golang-set v1.7.1 h1:SCQV0S6gTtp6itiFrTqI+pfmJ4LN85S1YzhDf9rTHJQ=
 github.com/deckarep/golang-set v1.7.1/go.mod h1:93vsz/8Wt4joVM7c2AVqh+YRMiUSc14yDtF28KmMOgQ=
-github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
 github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4=
 github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
@@ -58,7 +57,6 @@ github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
 github.com/hashicorp/golang-lru v0.5.5-0.20210104140557-80c98217689d h1:dg1dEPuWpEqDnvIw251EVy4zlP8gWbsGj4BsUKCRpYs=
 github.com/hashicorp/golang-lru v0.5.5-0.20210104140557-80c98217689d/go.mod h1:iADmTwqILo4mZ8BN3D2Q6+9jd8WM5uGBxy+E8yxSoD4=
-github.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
 github.com/json-iterator/go v1.1.7/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
@@ -123,7 +121,6 @@ github.com/urfave/cli/v2 v2.3.0 h1:qph92Y649prgesehzOrQjdWyxFOp/QVM+6imKHad91M=
 github.com/urfave/cli/v2 v2.3.0/go.mod h1:LJmUH05zAU44vOAcrfzZQKsZbVcdbOG8rtL3/XcUArI=
 github.com/vitelabs/go-vite/v2 v2.11.2 h1:6V1iU5Q24BfSs5VV0XPl32oytpQymt8xw/1SZQoE5s8=
 github.com/vitelabs/go-vite/v2 v2.11.2/go.mod h1:mdldLHfQp5yl9G5m4YeOwESuqVnKNImZhww2iW7s1DY=
-github.com/viteshan/gvite-tools v0.0.1 h1:s/wpEum+WmgJzX2Q/9H1jV7yFfY08Eh1Ehzca0fclG0=
 github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
 go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
 go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
@@ -190,15 +187,13 @@ google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miE
 google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=
 google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
 google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
-google.golang.org/protobuf v1.26.0 h1:bxAC2xTBsZGibn2RTntX0oH50xLsqy1OxA9tTL3p/lk=
 google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
 google.golang.org/protobuf v1.27.1 h1:SnqbnDw1V7RiZcXPx5MEeqPv2s79L9i7BJUlG/+RurQ=
 google.golang.org/protobuf v1.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
-gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=
 gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
 gopkg.in/go-playground/assert.v1 v1.2.1/go.mod h1:9RXL0bg/zibRAgZUYszZSwO/z8Y/a8bDuhia5mkpMnE=
 gopkg.in/go-playground/validator.v9 v9.29.1/go.mod h1:+c9/zcJMFNgbLvly1L1V+PpxWdVbfP1avr/N00E2vyQ=
@@ -210,7 +205,6 @@ gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkep
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
 gopkg.in/urfave/cli.v1 v1.20.0/go.mod h1:vuBzUtMdQeixQj8LVd+/98pzhxNGQoyuPBlsXHOQNO0=
 gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
-gopkg.in/yaml.v2 v2.2.3 h1:fvjTMHxHEw/mxHbtzPi3JCcKXQRAnQTBRo6YCJSVHKI=
 gopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.3.0 h1:clyUAQHOM3G0M3f5vQj7LuJrETvjVot3Z5el9nffUtU=

Diff for: gvite-tools/sbp/reward_sbp.go

@@ -0,0 +1,125 @@
+package sbp
+
+import (
+	"math/big"
+
+	"github.com/pkg/errors"
+	"github.com/vitelabs/go-vite/v2/client"
+	"github.com/vitelabs/go-vite/v2/common/types"
+	"github.com/vitelabs/go-vite/v2/rpcapi/api"
+)
+
+var OnePercentVite, _ = big.NewInt(0).SetString("10000000000000000", 10)
+var OneVite, _ = big.NewInt(0).SetString("1000000000000000000", 10)
+var OneViteFloat, _ = big.NewFloat(0).SetString("1000000000000000000")
+
+type SBPRewardVote struct {
+	SbpReward   *api.Reward
+	VoteTotal   *big.Int
+	VoteDetails map[types.Address]*big.Int
+	Idx         uint64
+}
+
+type SBPRewardDropDetails struct {
+	ToAddr types.Address
+	Amount *big.Int
+}
+
+func CalReward(startIdx uint64, endIdx uint64, sbp string, sbpRewardAddr types.Address, rpc client.RpcClient) (map[uint64]*SBPRewardVote, error) {
+	result := make(map[uint64]*SBPRewardVote)
+	for i := startIdx; i <= endIdx; i++ {
+		tmp := &SBPRewardVote{}
+		result[i] = tmp
+		reward, err := rpc.GetRewardByIndex(i)
+		if err != nil {
+			return nil, err
+		}
+
+		tmp.SbpReward = reward.RewardMap[sbp]
+		details, err := rpc.GetVoteDetailsByIndex(i)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, v := range details {
+			if v.Name == sbp {
+				tmp.VoteTotal, tmp.VoteDetails = reCalReward(sbpRewardAddr, v.Balance, v.Addr)
+			}
+		}
+	}
+	return result, nil
+}
+
+func reCalReward(sbpRewardAddr types.Address, totalBalance *big.Int, voteDetails map[types.Address]*big.Int) (*big.Int, map[types.Address]*big.Int) {
+	sbpFixVote := big.NewInt(1000000)
+	sbpFixVote.Mul(sbpFixVote, OneVite)
+
+	reTotalBalance := big.NewInt(0)
+	reTotalBalance.Add(reTotalBalance, totalBalance)
+	reTotalBalance.Add(reTotalBalance, sbpFixVote)
+
+	reSbp, ok := voteDetails[sbpRewardAddr]
+	if !ok {
+		voteDetails[sbpRewardAddr] = sbpFixVote
+	} else {
+		reSbp.Add(reSbp, sbpFixVote)
+	}
+	return reTotalBalance, voteDetails
+}
+
+func MergeReward(rewardVoteMap map[uint64]*SBPRewardVote) (*big.Int, error) {
+	totalReward := big.NewInt(0)
+	for _, v := range rewardVoteMap {
+		total, flag := big.NewInt(0).SetString(v.SbpReward.TotalReward, 10)
+		if !flag {
+			return nil, errors.Errorf("can't parse [%s] to bigInt.", v.SbpReward.TotalReward)
+		}
+		totalReward.Add(totalReward, total)
+	}
+	return totalReward, nil
+}
+
+func CalRewardDropDetails(rewardVoteMap map[uint64]*SBPRewardVote) (map[uint64]map[types.Address]*SBPRewardDropDetails, error) {
+	result := make(map[uint64]map[types.Address]*SBPRewardDropDetails)
+
+	for k, v := range rewardVoteMap {
+		if v.VoteTotal.Sign() <= 0 {
+			continue
+		}
+		m := make(map[types.Address]*SBPRewardDropDetails)
+		total := big.NewInt(0)
+		for kk, vv := range v.VoteDetails {
+			details := &SBPRewardDropDetails{ToAddr: kk}
+			total.Add(total, vv)
+			totalReward, flag := big.NewInt(0).SetString(v.SbpReward.TotalReward, 10)
+			if !flag {
+				return nil, errors.Errorf("bigInt fail %s", v.SbpReward.TotalReward)
+			}
+			totalReward.Mul(totalReward, vv)
+			totalReward.Div(totalReward, v.VoteTotal)
+
+			details.Amount = totalReward
+			m[kk] = details
+		}
+		if total.Cmp(v.VoteTotal) != 0 {
+			return nil, errors.Errorf("total vote not equal")
+		}
+		result[k] = m
+	}
+	return result, nil
+}
+
+func MergeRewardDrop(details map[uint64]map[types.Address]*SBPRewardDropDetails) (map[types.Address]*SBPRewardDropDetails, error) {
+	result := make(map[types.Address]*SBPRewardDropDetails)
+	for _, v := range details {
+		for kk, vv := range v {
+			dropDetails, ok := result[kk]
+			if !ok {
+				dropDetails = &SBPRewardDropDetails{ToAddr: kk, Amount: big.NewInt(0)}
+				result[kk] = dropDetails
+			}
+			dropDetails.Amount.Add(dropDetails.Amount, vv.Amount)
+		}
+	}
+	return result, nil
+}

Diff for: gvite-tools/sbp/reward_sbp_test.go

@@ -0,0 +1,170 @@
+package sbp
+
+import (
+	"fmt"
+	"math/big"
+	"math/rand"
+	"testing"
+
+	"github.com/vitelabs/go-vite/v2/client"
+	core "github.com/vitelabs/go-vite/v2/interfaces/core"
+	comm "github.com/viteshan/vite_tools/gvite-tools/common"
+
+	"github.com/vitelabs/go-vite/v2/common/types"
+)
+
+// var RawUrl = "http://116.63.158.55:48132"
+var RawUrl = "https://node.vite.net/gvite"
+
+// 发送奖励地址
+var selfAddr = types.HexToAddressPanic("vite_xxx")
+
+// 发送奖励助记词
+var mnemonic = "xxxxx"
+
+// sbp注册100w的算份额
+var sbpFixAddr = types.HexToAddressPanic("vite_xxx")
+
+var sbpName = "V.Morgen"
+
+func TestCalReward(t *testing.T) {
+	cli := comm.GetCli(RawUrl)
+	startIdx := uint64(668)
+	endIdx := uint64(683)
+	for i := startIdx; i <= endIdx; i++ {
+		fmt.Printf("start:%d, end: %d\n", startIdx, startIdx+6)
+		oneP(startIdx, startIdx+6, cli, t)
+		startIdx = startIdx + 7
+	}
+}
+func oneP(startIdx, endIdx uint64, cli client.RpcClient, t *testing.T) {
+
+	// startIdx := uint64(143)
+	// endIdx := uint64(149)
+
+	if endIdx < startIdx {
+		t.Fatal("error for idx")
+	}
+
+	//todo reward address
+	votes, err := CalReward(startIdx, endIdx, sbpName, sbpFixAddr, cli)
+	if err != nil {
+		t.Error(err)
+		t.FailNow()
+	}
+
+	//t.Log(votes)
+
+	rewardTotal, err := MergeReward(votes)
+	if err != nil {
+		t.Error(err)
+		t.FailNow()
+	}
+	avg := big.NewInt(0).Set(rewardTotal)
+
+	t.Log("total reward", big.NewInt(0).Div(rewardTotal, OnePercentVite).String(), avg.Div(avg, big.NewInt(int64(endIdx-startIdx+1))).String(), int64(endIdx-startIdx+1))
+
+	details, err := CalRewardDropDetails(votes)
+
+	if err != nil {
+		t.Error(err)
+		t.FailNow()
+	}
+	// t.Log(details)
+
+	{ // print to csv
+		var logs = ""
+		for i := startIdx; i <= endIdx; i++ {
+			details := details[i]
+			for kk, vv := range details {
+				amount := big.NewInt(0).Set(vv.Amount)
+				amountStr := amount.String()
+				logs += fmt.Sprintf("%d,%s,%f,%s\n", i, kk, float64(amount.Div(vv.Amount, OnePercentVite).Int64())/100.0, amountStr)
+			}
+		}
+		fmt.Println(logs)
+	}
+	return
+
+	mergedDetails, err := MergeRewardDrop(details)
+
+	{ // 打印一共发送了多少钱
+		sendTotal := big.NewInt(0)
+		for k, v := range mergedDetails {
+			amount := big.NewInt(0).Set(v.Amount)
+			t.Log(k, amount.Div(amount, OnePercentVite))
+			sendTotal.Add(sendTotal, amount)
+		}
+
+		//t.Log("all send", sendTotal.Div(sendTotal, oneVite))
+		t.Log("all send", sendTotal)
+	}
+	{ // 打印每万份投票每天大概是多少收益
+		fixAvgVote := big.NewInt(0)
+		allAvgVote := big.NewInt(0)
+		{
+			for k, v := range votes {
+				s := big.NewInt(0).Div(v.VoteDetails[sbpFixAddr], OnePercentVite)
+				t.Log(k, s.String())
+				fixAvgVote.Add(fixAvgVote, s)
+				allAvgVote.Add(allAvgVote, v.VoteTotal)
+			}
+			fixAvgVote.Div(fixAvgVote, big.NewInt(int64(len(votes))))
+		}
+		avgReward := big.NewInt(0).Set(mergedDetails[sbpFixAddr].Amount)
+		avgReward.Div(avgReward, big.NewInt(int64(len(votes))))
+		avgReward.Div(avgReward, OnePercentVite)
+		fixAvgVote.Div(fixAvgVote, big.NewInt(100))
+
+		allAvgVote.Div(allAvgVote, big.NewInt(int64(len(votes))))
+		t.Log("sbpFixAddr", allAvgVote.Div(allAvgVote, OneVite), avgReward.Div(avgReward, fixAvgVote.Div(fixAvgVote, big.NewInt(10000))))
+	}
+
+	txId := fmt.Sprintf("%d", rand.Intn(10000000))
+
+	var finalLog []*comm.SimpleRequestTx
+
+	var tokenId = core.ViteTokenId
+	{
+		for _, v := range mergedDetails {
+			amount := big.NewInt(0).Set(v.Amount)
+			if amount.Cmp(OnePercentVite) < 0 {
+				t.Log("ignore amount", v.ToAddr, v.Amount)
+				continue
+			}
+			finalLog = append(finalLog, &comm.SimpleRequestTx{ToAddr: v.ToAddr, Amount: amount, TokenId: tokenId, Data: []byte(txId)})
+		}
+
+		for k, v := range finalLog {
+			t.Log(k, v.ToAddr, v.Amount.String())
+		}
+	}
+
+	return
+
+	// key, _, err := comm.DerivationKey(mnemonic, selfAddr)
+	// if err != nil {
+	// 	t.Fatal(err)
+	// }
+
+	// if err != nil {
+	// 	t.Error(err)
+	// 	t.FailNow()
+	// }
+
+	// sender, err := comm.NewSender(cli, selfAddr, key)
+	// if err != nil {
+	// 	t.Error(err)
+	// 	t.FailNow()
+	// }
+
+	// all, err := sender.BatchSend(finalLog, nil)
+	// if err != nil {
+	// 	t.Error(err)
+	// 	t.FailNow()
+	// }
+
+	// for _, v := range all {
+	// 	t.Log(selfAddr, v.Height, v.Hash)
+	// }
+}