feat: add nft of erc1155

0xcregis · Sep 14, 2023 · 811d767 · 811d767
1 parent 21a5001
commit 811d767
Show file tree

Hide file tree

Showing 7 changed files with 63 additions and 28 deletions.
diff --git a/collect/service/cmd/chain/ether/service.go b/collect/service/cmd/chain/ether/service.go
@@ -20,14 +20,16 @@ import (
 )
 
 type Service struct {
-	log                *xlog.XLog
-	chain              *config.Chain
-	nodeId             string
-	transferTopic      string
-	store              collect.StoreTaskInterface
-	txChainClient      blockchain.API
-	blockChainClient   blockchain.API
-	receiptChainClient blockchain.API
+	log                    *xlog.XLog
+	chain                  *config.Chain
+	nodeId                 string
+	transferTopic          string
+	nftTransferSingleTopic string
+	nftTransferBatchTopic  string
+	store                  collect.StoreTaskInterface
+	txChainClient          blockchain.API
+	blockChainClient       blockchain.API
+	receiptChainClient     blockchain.API
 }
 
 func (s *Service) GetMultiBlockByNumber(blockNumber string, log *logrus.Entry, flag bool) ([]*collect.BlockInterface, []*collect.TxInterface) {
@@ -364,7 +366,17 @@ func (s *Service) buildContract(receipt *collect.Receipt) (*collect.Receipt, err
 		}
 
 		//erc1155
-		//todo save contract of erc-1155 to receipt.log
+		// save contract of erc-1155 to receipt.log
+		if len(g.Topics) == 4 && g.Topics[0] == s.nftTransferSingleTopic {
+			//处理 普通资产和 1155 协议 资产转移
+			mp := make(map[string]interface{}, 3)
+			token, _ := s.getToken(int64(s.chain.BlockChainCode), receipt.From, g.Address, "1155")
+			mp["token"] = token
+			mp["eip"] = 1155
+			mp["data"] = g.Data
+			bs, _ := json.Marshal(mp)
+			g.Data = string(bs)
+		}
 
 	}
 
@@ -480,7 +492,7 @@ func (s *Service) CheckAddress(tx []byte, addrList map[string]int64) bool {
 	return has
 }
 
-func NewService(c *config.Chain, x *xlog.XLog, store collect.StoreTaskInterface, nodeId string, transferTopic string) collect.BlockChainInterface {
+func NewService(c *config.Chain, x *xlog.XLog, store collect.StoreTaskInterface, nodeId string, transferTopic, nftTransferSingleTopic string) collect.BlockChainInterface {
 
 	var blockClient blockchain.API
 	if c.BlockTask != nil {
@@ -526,13 +538,14 @@ func NewService(c *config.Chain, x *xlog.XLog, store collect.StoreTaskInterface,
 	}
 
 	return &Service{
-		log:                x,
-		chain:              c,
-		store:              store,
-		txChainClient:      txClient,
-		blockChainClient:   blockClient,
-		receiptChainClient: receiptClient,
-		nodeId:             nodeId,
-		transferTopic:      transferTopic,
+		log:                    x,
+		chain:                  c,
+		store:                  store,
+		txChainClient:          txClient,
+		blockChainClient:       blockClient,
+		receiptChainClient:     receiptClient,
+		nodeId:                 nodeId,
+		transferTopic:          transferTopic,
+		nftTransferSingleTopic: nftTransferSingleTopic,
 	}
 }
diff --git a/collect/service/cmd/chain/gw.go b/collect/service/cmd/chain/gw.go
@@ -19,7 +19,7 @@ func GetBlockchain(blockchain int, c *config.Chain, store collect.StoreTaskInter
 	x := xlog.NewXLogger().BuildOutType(xlog.FILE).BuildFormatter(xlog.FORMAT_JSON).BuildFile(fmt.Sprintf("%v/chain_info", logConfig.Path), 24*time.Hour)
 	var srv collect.BlockChainInterface
 	if blockchain == 200 {
-		srv = ether.NewService(c, x, store, nodeId, collect.EthTopic)
+		srv = ether.NewService(c, x, store, nodeId, collect.EthTopic, collect.EthNftTransferSingleTopic)
 	} else if blockchain == 205 {
 		srv = tron2.NewService(c, x, store, nodeId, collect.TronTopic)
 	} else if blockchain == 201 {

diff --git a/collect/type.go b/collect/type.go
@@ -3,9 +3,10 @@ package collect
 import "time"
 
 const (
-	TronTopic    = "ddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
-	EthTopic     = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
-	PolygonTopic = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+	TronTopic                 = "ddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+	EthTopic                  = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+	EthNftTransferSingleTopic = "0xc3d58168c5ae7397731d063d5bbf3d657854427343f4c083240f7aacaa2d0f62"
+	PolygonTopic              = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
 )
 
 /**

diff --git a/store/chain/ether/kafka_util.go b/store/chain/ether/kafka_util.go
@@ -56,7 +56,7 @@ func GetTxType(body []byte) (uint64, error) {
 		return 2, nil
 	}
 }
-func ParseTx(body []byte, transferTopic string, blockchain int64) (*store.SubTx, error) {
+func ParseTx(body []byte, transferTopic, nftTransferSingleTopic string, blockchain int64) (*store.SubTx, error) {
 	var r store.SubTx
 	root := gjson.ParseBytes(body)
 	r.BlockChain = uint64(blockchain)
@@ -193,6 +193,26 @@ func ParseTx(body []byte, transferTopic string, blockchain int64) (*store.SubTx,
 				}
 			}
 
+			if eip == 1155 {
+				tps := v.Get("topics").Array()
+				v := r.Get("data").String()
+				var from, to string
+				if len(tps) == 4 && tps[0].String() == nftTransferSingleTopic {
+					//method = tps[0].String()
+					from = tps[1].String()
+					to = tps[2].String()
+					var m store.ContractTx
+					m.Contract = contract
+					m.Value = v
+					m.From, _ = util.Hex2Address(from)
+					m.To, _ = util.Hex2Address(to)
+					m.Method = "Transfer"
+					m.EIP = eip
+					m.Token = token
+					contractTx = append(contractTx, &m)
+				}
+			}
+
 		}
 		//else {
 		// 合约数据不完整，直接放弃了

diff --git a/store/chain/ether/kafka_util_test.go b/store/chain/ether/kafka_util_test.go
@@ -33,7 +33,7 @@ func TestParseTx(t *testing.T) {
 }
 
 `
-	tx, err := ParseTx([]byte(str), store.EthTopic, 200)
+	tx, err := ParseTx([]byte(str), store.EthTopic, store.EthTransferSingleTopic, 200)
 	if err != nil {
 		t.Error(err)
 	} else {

diff --git a/store/chain/gw.go b/store/chain/gw.go
@@ -62,7 +62,7 @@ func GetTxFromKafka(value []byte, blockChain int64) (*store.Tx, error) {
 
 func ParseTx(blockchain int64, msg *kafka.Message) (*store.SubTx, error) {
 	if blockchain == 200 {
-		return ether.ParseTx(msg.Value, store.EthTopic, blockchain)
+		return ether.ParseTx(msg.Value, store.EthTopic, store.EthTransferSingleTopic, blockchain)
 	}
 	if blockchain == 205 {
 		return tron.ParseTx(msg.Value, store.TronTopic, blockchain)

diff --git a/store/type.go b/store/type.go
@@ -1,9 +1,10 @@
 package store
 
 const (
-	TronTopic    = "ddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
-	EthTopic     = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
-	PolygonTopic = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+	TronTopic              = "ddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+	EthTopic               = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
+	EthTransferSingleTopic = "0xc3d58168c5ae7397731d063d5bbf3d657854427343f4c083240f7aacaa2d0f62"
+	PolygonTopic           = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
 )
 
 type MonitorAddress struct {