scan.go 5.8 KB

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