A transparent cache for datastore, backed by memcache.

filter/dscache/serialize.go

Index: filter/dscache/serialize.go
diff --git a/filter/dscache/serialize.go b/filter/dscache/serialize.go
new file mode 100644
index 0000000000000000000000000000000000000000..ecf685a028c64f28d66051510c3e2f18d25fb4c5
--- /dev/null
+++ b/filter/dscache/serialize.go
@@ -0,0 +1,61 @@
+// 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 dscache
+
+import (
+	"bytes"
+	"compress/zlib"
+	"io/ioutil"
+
+	ds "github.com/luci/gae/service/datastore"
+)
+
+func encodeItemValue(pm ds.PropertyMap) []byte {
+	pm, _ = pm.Save(false)
+
+	buf := bytes.Buffer{}
+	// errs can't happen, since we're using a byte buffer.
+	_ = buf.WriteByte(byte(NoCompression))
+	_ = pm.Write(&buf, ds.WithoutContext)
+
+	data := buf.Bytes()
+	if buf.Len() > CompressionThreshold {
+		buf2 := bytes.NewBuffer(make([]byte, 0, len(data)))
+		_ = buf2.WriteByte(byte(ZlibCompression))
+		writer := zlib.NewWriter(buf2)
+		_, _ = writer.Write(data[1:]) // skip the NoCompression byte
+		writer.Close()
+		data = buf2.Bytes()
+	}
+
+	return data
+}
+
+func decodeItemValue(val []byte, ns, aid string) (ds.PropertyMap, error) {
+	if len(val) == 0 {
+		return nil, ds.ErrNoSuchEntity
+	}
+	buf := bytes.NewBuffer(val)
+	compTypeByte, err := buf.ReadByte()
+	if err != nil {
+		return nil, err
+	}
+
+	if CompressionType(compTypeByte) == ZlibCompression {
+		reader, err := zlib.NewReader(buf)
+		if err != nil {
+			return nil, err
+		}
+		defer reader.Close()
+		data, err := ioutil.ReadAll(reader)
+		if err != nil {
+			return nil, err
+		}
+		buf = bytes.NewBuffer(data)
+	}
+	ret := ds.PropertyMap{}
+	err = ret.Read(buf, ds.WithoutContext, ns, aid)
+	return ret, err
+}