| Index: go/src/infra/gae/libs/wrapper/memory/binutils.go
|
| diff --git a/go/src/infra/gae/libs/wrapper/memory/binutils.go b/go/src/infra/gae/libs/wrapper/memory/binutils.go
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..4acf30b6f54556356046990995a3840455f7fc57
|
| --- /dev/null
|
| +++ b/go/src/infra/gae/libs/wrapper/memory/binutils.go
|
| @@ -0,0 +1,98 @@
|
| +// Copyright 2015 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +package memory
|
| +
|
| +import (
|
| + "bytes"
|
| + "encoding/binary"
|
| + "fmt"
|
| + "math"
|
| + "time"
|
| +
|
| + "appengine"
|
| +
|
| + "github.com/luci/luci-go/common/cmpbin"
|
| +)
|
| +
|
| +func writeString(buf *bytes.Buffer, s string) {
|
| + cmpbin.WriteUint(buf, uint64(len(s)))
|
| + buf.WriteString(s)
|
| +}
|
| +
|
| +func readString(buf *bytes.Buffer) (string, error) {
|
| + b, err := readBytes(buf)
|
| + if err != nil {
|
| + return "", err
|
| + }
|
| + return string(b), nil
|
| +}
|
| +
|
| +func writeBytes(buf *bytes.Buffer, b []byte) {
|
| + cmpbin.WriteUint(buf, uint64(len(b)))
|
| + buf.Write(b)
|
| +}
|
| +
|
| +func readBytes(buf *bytes.Buffer) ([]byte, error) {
|
| + val, _, err := cmpbin.ReadUint(buf)
|
| + if err != nil {
|
| + return nil, err
|
| + }
|
| + if val > 2*1024*1024 { // 2MB
|
| + return nil, fmt.Errorf("readBytes: tried to read %d bytes (> 2MB)", val)
|
| + }
|
| + retBuf := make([]byte, val)
|
| + n, _ := buf.Read(retBuf) // err is either io.EOF or nil for bytes.Buffer
|
| + if uint64(n) != val {
|
| + return nil, fmt.Errorf("readBytes: expected %d bytes but read %d", val, n)
|
| + }
|
| + return retBuf, err
|
| +}
|
| +
|
| +func writeFloat64(buf *bytes.Buffer, v float64) {
|
| + // byte-ordered floats http://stereopsis.com/radix.html
|
| + bits := math.Float64bits(v)
|
| + bits = bits ^ (-(bits >> 63) | (1 << 63))
|
| + data := make([]byte, 8)
|
| + binary.BigEndian.PutUint64(data, bits)
|
| + buf.Write(data)
|
| +}
|
| +
|
| +func readFloat64(buf *bytes.Buffer) (float64, error) {
|
| + // byte-ordered floats http://stereopsis.com/radix.html
|
| + data := make([]byte, 8)
|
| + _, err := buf.Read(data)
|
| + if err != nil {
|
| + return 0, err
|
| + }
|
| + bits := binary.BigEndian.Uint64(data)
|
| + return math.Float64frombits(bits ^ (((bits >> 63) - 1) | (1 << 63))), nil
|
| +}
|
| +
|
| +// We truncate this to microseconds and drop the timezone, because that's the
|
| +// way that the appengine SDK does it. Awesome, right? Also: its not documented.
|
| +func writeTime(buf *bytes.Buffer, t time.Time) {
|
| + cmpbin.WriteUint(buf, uint64(t.Unix())*1e6+uint64(t.Nanosecond()/1e3))
|
| +}
|
| +
|
| +func readTime(buf *bytes.Buffer) (time.Time, error) {
|
| + v, _, err := cmpbin.ReadUint(buf)
|
| + if err != nil {
|
| + return time.Time{}, err
|
| + }
|
| + return time.Unix(int64(v/1e6), int64((v%1e6)*1e3)), nil
|
| +}
|
| +
|
| +func writeGeoPoint(buf *bytes.Buffer, gp appengine.GeoPoint) {
|
| + writeFloat64(buf, gp.Lat)
|
| + writeFloat64(buf, gp.Lng)
|
| +}
|
| +
|
| +func readGeoPoint(buf *bytes.Buffer) (pt appengine.GeoPoint, err error) {
|
| + if pt.Lat, err = readFloat64(buf); err != nil {
|
| + return
|
| + }
|
| + pt.Lng, err = readFloat64(buf)
|
| + return
|
| +}
|
|
|