scan.go 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252
  1. package blockchain
  2. import (
  3. "context"
  4. "encoding/hex"
  5. "fmt"
  6. "ktogame/contractgo/Collect"
  7. "ktogame/controller"
  8. "ktogame/dbUtil"
  9. "ktogame/models"
  10. "os"
  11. "strings"
  12. "time"
  13. "github.com/astaxie/beego"
  14. "github.com/astaxie/beego/logs"
  15. "github.com/ethereum/go-ethereum/accounts/abi"
  16. utilAddress "github.com/korthochain/korthochain/pkg/address"
  17. "github.com/korthochain/korthochain/pkg/block"
  18. pb "github.com/korthochain/korthochain/pkg/server/grpcserver/message"
  19. "github.com/korthochain/korthochain/pkg/transaction"
  20. "google.golang.org/grpc"
  21. )
  22. var ktoClient pb.GreeterClient
  23. var ktoRpc = beego.AppConfig.String("miner_node")
  24. var currentBlock uint64
  25. var TopsMap map[string]string
  26. func init() {
  27. kc, err := grpc.Dial(ktoRpc, grpc.WithInsecure(), grpc.WithBlock())
  28. if err != nil {
  29. logs.Error(err)
  30. os.Exit(1)
  31. }
  32. ktoClient = pb.NewGreeterClient(kc)
  33. utilAddress.SetNetWork("mainnet")
  34. tm := make(map[string]string, 0)
  35. tm[TOPIC_PARTICIPATE] = COLLECT_PARTICIPATE
  36. tm[TOPIC_CLAIM] = COLLECT_CLAIMREWARDS
  37. TopsMap = tm
  38. go scanBlock()
  39. go confirmClaimedTxs()
  40. }
  41. func scanBlock() {
  42. logs.Info("scanBlock...")
  43. var bi models.BlockInfo
  44. ok, err := dbUtil.Engine.Id(1).Get(&bi)
  45. if err != nil {
  46. fmt.Println("获取最新快高错误=", err)
  47. return
  48. }
  49. if !ok {
  50. fmt.Println("获取最新快高失败!")
  51. return
  52. }
  53. currentBlock = uint64(bi.BlockNumber) + 1
  54. //currentBlock =62203394
  55. for {
  56. time.Sleep(time.Second * 3)
  57. res, err := ktoClient.GetMaxBlockHeight(context.Background(), &pb.ReqMaxBlockHeight{})
  58. if err != nil {
  59. logs.Error("获取交易block number错误=", err)
  60. continue
  61. }
  62. logs.Info("currentBlock:", currentBlock, "chainblock:", res.MaxHeight)
  63. if res.MaxHeight < currentBlock {
  64. currentBlock = res.MaxHeight
  65. }
  66. bl, err := ktoClient.GetBlockByNum(context.Background(), &pb.ReqBlockByNumber{Height: currentBlock})
  67. if err != nil || bl.Code != 0 {
  68. logs.Error("获取交易block错误=", err)
  69. continue
  70. }
  71. blc, errs := block.Deserialize(bl.Data)
  72. if errs != nil {
  73. logs.Error("解析blcock错误=", errs)
  74. continue
  75. }
  76. var ERR error
  77. for _, v := range blc.Transactions {
  78. if len(v.Input) == 0 {
  79. continue
  80. }
  81. evm, err := transaction.DecodeEvmData(v.Input)
  82. if err != nil {
  83. logs.Error(err)
  84. ERR = err
  85. break
  86. }
  87. if len(evm.Logs) == 0 {
  88. continue
  89. }
  90. for _, l := range evm.Logs {
  91. th := l.Topics[0].Hex()
  92. method := TopsMap[th]
  93. contract := l.Address.String()
  94. if contract == COLLECTCONTRACT {
  95. if method == COLLECT_PARTICIPATE {
  96. var ev EventParticipate
  97. abi, err := abi.JSON(strings.NewReader(Collect.CollectMetaData.ABI))
  98. if err != nil {
  99. logs.Error(err)
  100. ERR = err
  101. break
  102. }
  103. err = abi.UnpackIntoInterface(&ev, method, l.Data)
  104. if err != nil {
  105. logs.Error(err)
  106. ERR = err
  107. break
  108. }
  109. logs.Info("participate data:", ev.Participant.String(), ev.Inviter.String(), v.HashToString(), float64(ev.Amount.Uint64()/controller.Decimals))
  110. //handle user participate
  111. err = participate(dbUtil.Engine, ev.Participant.String(), ev.Inviter.String(), v.HashToString(), float64(ev.Amount.Uint64()/controller.Decimals))
  112. if err != nil {
  113. logs.Error(err)
  114. ERR = err
  115. break
  116. }
  117. } else if method == COLLECT_CLAIMREWARDS {
  118. var evt EventClaim
  119. abi, err := abi.JSON(strings.NewReader(Collect.CollectMetaData.ABI))
  120. if err != nil {
  121. logs.Error(err)
  122. ERR = err
  123. break
  124. }
  125. err = abi.UnpackIntoInterface(&evt, method, l.Data)
  126. if err != nil {
  127. logs.Error(err)
  128. ERR = err
  129. break
  130. }
  131. fmt.Printf("claim event data=%+v\n ", evt)
  132. var bts []byte
  133. for _, b := range evt.Signature {
  134. bts = append(bts, b)
  135. }
  136. err = checkClaim(dbUtil.Engine, evt.User.String(), v.HashToString(), hex.EncodeToString(bts), float64(evt.Amount.Uint64()/controller.Decimals), int64(v.BlockNum))
  137. if err != nil {
  138. logs.Error(err)
  139. ERR = err
  140. break
  141. }
  142. }
  143. }
  144. }
  145. if ERR != nil {
  146. logs.Error(err)
  147. break
  148. }
  149. }
  150. if ERR != nil {
  151. logs.Error("处理错误=", ERR)
  152. continue
  153. }
  154. bi.BlockNumber = int64(currentBlock)
  155. _, err = dbUtil.Engine.ID(1).Update(&bi)
  156. if err != nil {
  157. logs.Error("更新最新快高错误=", err)
  158. return
  159. }
  160. currentBlock++
  161. }
  162. }
  163. func confirmClaimedTxs() {
  164. var mark int64 = 0
  165. for {
  166. time.Sleep(time.Second * 5)
  167. logs.Info("confirmClaimedTxs mark= %v\n", mark)
  168. var txs []models.ClaimedTxs
  169. err := dbUtil.Engine.Where("state = ?", 0).Where("droped = ?", 0).Where("id > ?", mark).Find(&txs)
  170. if err != nil {
  171. logs.Error(err)
  172. continue
  173. }
  174. if len(txs) > 0 {
  175. mark = txs[len(txs)-1].Id
  176. }
  177. for _, t := range txs {
  178. go func(tx models.ClaimedTxs) {
  179. interval := 0
  180. for {
  181. time.Sleep(time.Second * 5)
  182. if interval > (CONFIRMINTERVAL / 5) {
  183. var ui models.UserInfo
  184. _, err := dbUtil.Engine.Id(tx.Hash).Get(&ui)
  185. if err != nil {
  186. mark = tx.Id - 1
  187. return
  188. }
  189. ui.AvailableClaim += tx.Amount
  190. ui.TotalClaimed -= tx.Amount
  191. _, err = dbUtil.Engine.ID(tx.Addr).Cols("total_claimed,available_claim").Update(&ui)
  192. if err != nil {
  193. mark = tx.Id - 1
  194. return
  195. }
  196. tx.Droped = 1
  197. _, err = dbUtil.Engine.ID(tx.Hash).Cols("droped").Update(&tx)
  198. if err != nil {
  199. mark = tx.Id - 1
  200. return
  201. }
  202. return
  203. }
  204. ptx, err := ktoClient.GetTxByHash(context.Background(), &pb.ReqTxByHash{Hash: tx.Hash})
  205. if err != nil {
  206. interval++
  207. continue
  208. }
  209. if ptx.Code != 0 {
  210. interval++
  211. continue
  212. }
  213. ftx, err := transaction.DeserializeFinishedTransaction(ptx.Data)
  214. if err != nil {
  215. interval++
  216. continue
  217. }
  218. //pending tx
  219. if ftx.BlockNum == 0 {
  220. interval++
  221. continue
  222. }
  223. tx.State = 1
  224. _, err = dbUtil.Engine.ID(tx.Hash).Cols("state").Update(&tx)
  225. if err != nil {
  226. mark = tx.Id - 1
  227. }
  228. return
  229. }
  230. }(t)
  231. }
  232. }
  233. }