woop/client/utils.go

package client

import (
	"bytes"
	"github.com/simple-rules/harmony-benchmark/crypto/pki"
	"io"
	"log"
	"net/http"
	"sync"

)

var AddressToIntPriKeyMap map[[20]byte]int // For convenience, we use int as the secret seed for generating private key
var AddressToIntPriKeyMapLock sync.Mutex

func InitLookUpIntPriKeyMap() {
	if AddressToIntPriKeyMap == nil {
		AddressToIntPriKeyMapLock.Lock()
		AddressToIntPriKeyMap = make(map[[20]byte]int)
		for i := 1; i <= 10000; i++ {
			AddressToIntPriKeyMap[pki.GetAddressFromInt(i)] = i
		}
		AddressToIntPriKeyMapLock.Unlock()
	}
}

func LookUpIntPriKey(address [20]byte) (int, bool) {
	value, ok := AddressToIntPriKeyMap[address]
	return value, ok
}

func DownloadUrlAsString(url string) (string, error) {
	response, err := http.Get(url)
	buf := bytes.NewBufferString("")
	if err != nil {
		log.Fatal(err)
	} else {
		defer response.Body.Close()
		_, err := io.Copy(buf, response.Body)
		if err != nil {
			log.Fatal(err)
		}
	}
	return buf.String(), err
}