Merge pull request #21 from everFinance/broadcast-txs

fix(): add upload and get function from peers

Author: zyjblockchain

client.go

@@ -11,6 +11,7 @@ import (
 	"net/url"
 	"path"
 	"strconv"
+	"strings"
 
 	"github.com/everFinance/sandy_log/log"
 
@@ -53,6 +54,17 @@ func (c *Client) GetInfo() (info *types.NetworkInfo, err error) {
 	return
 }
 
+func (c *Client) Peers() ([]string, error) {
+	body, _, err := c.httpGet("peers")
+	if err != nil {
+		return nil, ErrBadGateway
+	}
+
+	peers := make([]string, 0)
+	err = json.Unmarshal(body, &peers)
+	return peers, err
+}
+
 // GetTransactionByID status: Pending/Invalid hash/overspend
 func (c *Client) GetTransactionByID(id string) (tx *types.Transaction, err error) {
 	body, statusCode, err := c.httpGet(fmt.Sprintf("tx/%s", id))
@@ -422,6 +434,62 @@ func (c *Client) DownloadChunkData(id string) ([]byte, error) {
 	return data, nil
 }
 
+func (c *Client) GetTxDataFromPeers(txId string) ([]byte, error) {
+	peers, err := c.Peers()
+	if err != nil {
+		return nil, err
+	}
+	for _, peer := range peers {
+		if strings.Contains(peer, "127.0") {
+			continue
+		}
+		arNode := NewClient("http://" + peer)
+		data, err := arNode.GetTransactionData(txId)
+		if err != nil {
+			fmt.Printf("get tx data error:%v, peer: %s\n", err, peer)
+			continue
+		}
+		return data, nil
+	}
+	return nil, errors.New("get tx data from peers failed")
+}
+
+func (c *Client) UploadTxDataToPeers(txId string, data []byte) error {
+	peers, err := c.Peers()
+	if err != nil {
+		return err
+	}
+
+	count := 0
+	for _, peer := range peers {
+		if strings.Contains(peer, "127.0") {
+			continue
+		}
+		fmt.Printf("upload peer: %s, count: %d\n", peer, count)
+		arNode := NewClient("http://" + peer)
+		uploader, err := CreateUploader(arNode, txId, data)
+		if err != nil {
+			continue
+		}
+	Loop:
+		for !uploader.IsComplete() {
+			if err := uploader.UploadChunk(); err != nil {
+				break Loop
+			}
+			if uploader.LastResponseStatus != 200 {
+				break Loop
+			}
+		}
+		if uploader.IsComplete() { // upload success
+			count++
+		}
+		if count > 20 {
+			return nil
+		}
+	}
+	return fmt.Errorf("upload tx data to peers failed, txId: %s", txId)
+}
+
 // push to bundler gateway
 
 // SendItemToBundler send bundle bundleItem to bundler gateway

client_test.go

@@ -192,3 +192,34 @@ func TestClient_GetTransactionDataByGateway(t *testing.T) {
 	assert.NoError(t, err)
 	t.Log(len(data))
 }
+
+func TestClient_GetPeers(t *testing.T) {
+	arNode := "https://arweave.net"
+	cli := NewClient(arNode)
+	peers, err := cli.Peers()
+	assert.NoError(t, err)
+	t.Log(len(peers))
+}
+
+func Test_UploadTxDataToPeers(t *testing.T) {
+	data := make([]byte, 1103732)
+	data[0] = byte('z')
+	data[1] = byte('y')
+	data[2] = byte('j')
+	for i := 3; i < len(data); i++ {
+		data[i] = byte('a')
+	}
+
+	cli := NewClient("https://arweave.net")
+	txId := "D3GOny9cItUEc8qAl1oLUtnoLOB3OfSB-wKbw8TUIRc"
+	err = cli.UploadTxDataToPeers(txId, data)
+	assert.NoError(t, err)
+}
+
+func Test_GetTxDataFromPeers(t *testing.T) {
+	cli := NewClient("https://arweave.net")
+	txId := "D3GOny9cItUEc8qAl1oLUtnoLOB3OfSB-wKbw8TUIRc"
+	data, err := cli.GetTxDataFromPeers(txId)
+	assert.NoError(t, err)
+	assert.Equal(t, 1471643, len(data))
+}