From f99c025692099c839a9fa50369b378f7ede5036d Mon Sep 17 00:00:00 2001
From: "zhiqiang.wang" <xv12@qq.com>
Date: Mon, 23 Sep 2024 22:03:20 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=8C=BA=E5=9D=97=E6=89=AB=E6=8F=8F?=
 =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E8=B5=84=E6=BA=90=E5=9B=9E=E6=94=B6=E5=92=8C?=
 =?UTF-8?q?=E4=BB=A3=E7=90=86=E8=AF=86=E5=88=AB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/monitor/tron.go | 62 ++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 59 insertions(+), 3 deletions(-)

diff --git a/app/monitor/tron.go b/app/monitor/tron.go
index 3ac840b..b009e93 100644
--- a/app/monitor/tron.go
+++ b/app/monitor/tron.go
@@ -30,6 +30,15 @@ var usdtTrc20ContractAddress = []byte{0x41, 0xa6, 0x14, 0xf8, 0x03, 0xb6, 0xfd,
 
 var currentBlockHeight int64
 
+type resource struct {
+	ID          string
+	Type        core.ResourceCode
+	Balance     int64
+	FromAddress string
+	RecvAddress string
+	Timestamp   time.Time
+}
+
 type transfer struct {
 	ID          string
 	Amount      float64
@@ -102,6 +111,8 @@ func BlockScanStart() {
 func parseBlockTrans(block *api.BlockExtention, nowHeight int64) {
 	currentBlockHeight = nowHeight
 
+	var unDelegateResources = make([]resource, 0)
+	var delegateResources = make([]resource, 0)
 	var trxTransfer = make([]transfer, 0)
 	var usdtTrc20Transfer = make([]transfer, 0)
 	var timestamp = time.UnixMilli(block.GetBlockHeader().GetRawData().GetTimestamp())
@@ -112,8 +123,47 @@ func parseBlockTrans(block *api.BlockExtention, nowHeight int64) {
 		}
 
 		var itm = v.GetTransaction()
+		var id = hex.EncodeToString(v.Txid)
 		for _, contract := range itm.GetRawData().GetContract() {
-			// 转账交易
+			// 资源代理 DelegateResourceContract
+			if contract.GetType() == core.Transaction_Contract_DelegateResourceContract {
+				var foo = &core.DelegateResourceContract{}
+				err := contract.GetParameter().UnmarshalTo(foo)
+				if err != nil {
+
+					continue
+				}
+
+				delegateResources = append(delegateResources, resource{
+					ID:          id,
+					Type:        foo.Resource,
+					Balance:     foo.Balance,
+					FromAddress: base58CheckEncode(foo.OwnerAddress),
+					RecvAddress: base58CheckEncode(foo.ReceiverAddress),
+					Timestamp:   timestamp,
+				})
+			}
+
+			// 资源回收 UnDelegateResourceContract
+			if contract.GetType() == core.Transaction_Contract_UnDelegateResourceContract {
+				var foo = &core.UnDelegateResourceContract{}
+				err := contract.GetParameter().UnmarshalTo(foo)
+				if err != nil {
+
+					continue
+				}
+
+				unDelegateResources = append(unDelegateResources, resource{
+					ID:          id,
+					Type:        foo.Resource,
+					Balance:     foo.Balance,
+					FromAddress: base58CheckEncode(foo.OwnerAddress),
+					RecvAddress: base58CheckEncode(foo.ReceiverAddress),
+					Timestamp:   timestamp,
+				})
+			}
+
+			// TRX转账交易
 			if contract.GetType() == core.Transaction_Contract_TransferContract {
 				var foo = &core.TransferContract{}
 				err := contract.GetParameter().UnmarshalTo(foo)
@@ -123,7 +173,7 @@ func parseBlockTrans(block *api.BlockExtention, nowHeight int64) {
 				}
 
 				trxTransfer = append(trxTransfer, transfer{
-					ID:          hex.EncodeToString(v.Txid),
+					ID:          id,
 					Amount:      float64(foo.Amount),
 					FromAddress: base58CheckEncode(foo.OwnerAddress),
 					RecvAddress: base58CheckEncode(foo.ToAddress),
@@ -142,7 +192,7 @@ func parseBlockTrans(block *api.BlockExtention, nowHeight int64) {
 					continue
 				}
 
-				var transItem = transfer{Timestamp: timestamp, ID: hex.EncodeToString(v.Txid), FromAddress: base58CheckEncode(foo.OwnerAddress)}
+				var transItem = transfer{Timestamp: timestamp, ID: id, FromAddress: base58CheckEncode(foo.OwnerAddress)}
 				var reader = bytes.NewReader(foo.GetData())
 				if !bytes.Equal(foo.GetContractAddress(), usdtTrc20ContractAddress) { // usdt trc20 contract
 
@@ -171,6 +221,12 @@ func parseBlockTrans(block *api.BlockExtention, nowHeight int64) {
 
 	if len(trxTransfer) > 0 {
 
+	}
+	if len(unDelegateResources) > 0 {
+
+	}
+	if len(delegateResources) > 0 {
+
 	}
 
 	log.Info("区块扫描完成：", nowHeight)