You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.0 KiB
44 lines
1.0 KiB
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"harmony-benchmark/p2p"
|
|
"log"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// ConvertFixedDataIntoByteArray converts an empty interface data to a byte array
|
|
func ConvertFixedDataIntoByteArray(data interface{}) []byte {
|
|
buff := new(bytes.Buffer)
|
|
err := binary.Write(buff, binary.BigEndian, data)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
return buff.Bytes()
|
|
}
|
|
|
|
// ConvertIntoInts is to convert '1,2,3,4' into []int{1,2,3,4}.
|
|
func ConvertIntoInts(data string) []int {
|
|
var res = []int{}
|
|
items := strings.Split(data, ",")
|
|
for _, value := range items {
|
|
intValue, err := strconv.Atoi(value)
|
|
if err == nil {
|
|
res = append(res, intValue)
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
func GetUniqueIdFromPeer(peer p2p.Peer) uint16 {
|
|
reg, err := regexp.Compile("[^0-9]+")
|
|
if err != nil {
|
|
log.Panic("Regex Compilation Failed", "err", err)
|
|
}
|
|
socketId := reg.ReplaceAllString(peer.Ip+peer.Port, "") // A integer Id formed by unique IP/PORT pair
|
|
value, _ := strconv.Atoi(socketId)
|
|
return uint16(value)
|
|
}
|
|
|