use std encoding lib

Signed-off-by: divinerapier <poriter.coco@gmail.com>

2 files changed, 89 insertions, 6 deletions

diff --git a/weed/util/bytes.go b/weed/util/bytes.go
index dfa4ae665..68279cefc 100644
--- a/weed/util/bytes.go
+++ b/weed/util/bytes.go
@@ -1,8 +1,10 @@
 package util
 
+import "encoding/binary"
+
 // big endian
 
-func BytesToUint64(b []byte) (v uint64) {
+func BytesToUint64Old(b []byte) (v uint64) {
 	length := uint(len(b))
 	for i := uint(0); i < length-1; i++ {
 		v += uint64(b[i])
@@ -11,7 +13,7 @@ func BytesToUint64(b []byte) (v uint64) {
 	v += uint64(b[length-1])
 	return
 }
-func BytesToUint32(b []byte) (v uint32) {
+func BytesToUint32Old(b []byte) (v uint32) {
 	length := uint(len(b))
 	for i := uint(0); i < length-1; i++ {
 		v += uint32(b[i])
@@ -20,26 +22,45 @@ func BytesToUint32(b []byte) (v uint32) {
 	v += uint32(b[length-1])
 	return
 }
-func BytesToUint16(b []byte) (v uint16) {
+func BytesToUint16Old(b []byte) (v uint16) {
 	v += uint16(b[0])
 	v <<= 8
 	v += uint16(b[1])
 	return
 }
-func Uint64toBytes(b []byte, v uint64) {
+func Uint64toBytesOld(b []byte, v uint64) {
 	for i := uint(0); i < 8; i++ {
 		b[7-i] = byte(v >> (i * 8))
 	}
 }
-func Uint32toBytes(b []byte, v uint32) {
+func Uint32toBytesOld(b []byte, v uint32) {
 	for i := uint(0); i < 4; i++ {
 		b[3-i] = byte(v >> (i * 8))
 	}
 }
-func Uint16toBytes(b []byte, v uint16) {
+func Uint16toBytesOld(b []byte, v uint16) {
 	b[0] = byte(v >> 8)
 	b[1] = byte(v)
 }
+
+func BytesToUint64(b []byte) (v uint64) {
+	return binary.BigEndian.Uint64(b)
+}
+func BytesToUint32(b []byte) (v uint32) {
+	return binary.BigEndian.Uint32(b)
+}
+func BytesToUint16(b []byte) (v uint16) {
+	return binary.BigEndian.Uint16(b)
+}
+func Uint64toBytes(b []byte, v uint64) {
+	binary.BigEndian.PutUint64(b, v)
+}
+func Uint32toBytes(b []byte, v uint32) {
+	binary.BigEndian.PutUint32(b, v)
+}
+func Uint16toBytes(b []byte, v uint16) {
+	binary.BigEndian.PutUint16(b, v)
+}
 func Uint8toBytes(b []byte, v uint8) {
 	b[0] = byte(v)
 }
diff --git a/weed/util/bytes_test.go b/weed/util/bytes_test.go
new file mode 100644
index 000000000..4e6b3f294
--- /dev/null
+++ b/weed/util/bytes_test.go
@@ -0,0 +1,62 @@
+package util
+
+import (
+	"reflect"
+	"testing"
+)
+
+// go test github.com/chrislusf/seaweedfs/weed/util -run '^(TestAll)$' -v
+
+func TestAll(t *testing.T) {
+	for i := 0; i <= 0xff; i += 7 {
+		assertU16(t, uint16(i))
+	}
+	for i := 0; i <= 0xffff; i += 211 {
+		assertU32(t, uint32(i))
+	}
+	for i := 0; i <= 0xffffffff; i += 1000000007 {
+		assertU64(t, uint64(i))
+	}
+}
+
+func assertU16(t *testing.T, i uint16) {
+	var buf1, buf2 [2]byte
+	Uint16toBytesOld(buf1[:], i)
+	Uint16toBytes(buf2[:], i)
+	if !reflect.DeepEqual(buf1, buf2) {
+		t.Errorf("i: %d, buf1: %v, buf2: %v\n", i, buf1, buf2)
+	}
+	v1 := BytesToUint16Old(buf1[:])
+	v2 := BytesToUint16(buf2[:])
+	if v1 != v2 {
+		t.Errorf("buf1: %v, v1: %d, buf2: %v, v2: %d\n", buf1, v1, buf2, v2)
+	}
+}
+
+func assertU32(t *testing.T, i uint32) {
+	var buf1, buf2 [4]byte
+	Uint32toBytesOld(buf1[:], i)
+	Uint32toBytes(buf2[:], i)
+	if !reflect.DeepEqual(buf1, buf2) {
+		t.Errorf("i: %d, buf1: %v, buf2: %v\n", i, buf1, buf2)
+	}
+	v1 := BytesToUint32Old(buf1[:])
+	v2 := BytesToUint32(buf2[:])
+	if v1 != v2 {
+		t.Errorf("buf1: %v, v1: %d, buf2: %v, v2: %d\n", buf1, v1, buf2, v2)
+	}
+}
+
+func assertU64(t *testing.T, i uint64) {
+	var buf1, buf2 [8]byte
+	Uint64toBytesOld(buf1[:], i)
+	Uint64toBytes(buf2[:], i)
+	if !reflect.DeepEqual(buf1, buf2) {
+		t.Errorf("i: %d, buf1: %v, buf2: %v\n", i, buf1, buf2)
+	}
+	v1 := BytesToUint64Old(buf1[:])
+	v2 := BytesToUint64(buf2[:])
+	if v1 != v2 {
+		t.Errorf("buf1: %v, v1: %d, buf2: %v, v2: %d\n", buf1, v1, buf2, v2)
+	}
+}