Reland: Refactor current GAE abstraction library to be free of the SDK*

go/src/infra/gae/libs/wrapper/memory/datastore.go

-// 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 (
-        "errors"
-        "fmt"
-        "infra/gae/libs/wrapper"
-        "strings"
-
-        "github.com/mjibson/goon"
-        "golang.org/x/net/context"
-
-        "appengine/datastore"
-        "appengine_internal"
-        pb "appengine_internal/datastore"
-)
-
-//////////////////////////////////// public ////////////////////////////////////
-
-// useDS adds a wrapper.Datastore implementation to context, accessible
-// by wrapper.GetDS(c)
-func useDS(c context.Context) context.Context {
-        return wrapper.SetDSFactory(c, func(ic context.Context) wrapper.Datastore {
-                dsd := cur(ic).Get(memContextDSIdx)
-
-                switch x := dsd.(type) {
-                case *dataStoreData:
-                        return &dsImpl{wrapper.DummyDS(), x, curGID(ic).namespace, ic}
-                case *txnDataStoreData:
-                        return &txnDsImpl{wrapper.DummyDS(), x, curGID(ic).namespace}
-                default:
-                        panic(fmt.Errorf("DS: bad type: %v in context %v", dsd, ic))
-                }
-        })
-}
-
-//////////////////////////////////// dsImpl ////////////////////////////////////
-
-// dsImpl exists solely to bind the current c to the datastore data.
-type dsImpl struct {
-        wrapper.Datastore
-
-        data *dataStoreData
-        ns   string
-        c    context.Context
-}
-
-var (
-        _ = wrapper.Datastore((*dsImpl)(nil))
-        _ = wrapper.Testable((*dsImpl)(nil))
-)
-
-func (d *dsImpl) BreakFeatures(err error, features ...string) {
-        d.data.BreakFeatures(err, features...)
-}
-func (d *dsImpl) UnbreakFeatures(features ...string) {
-        d.data.UnbreakFeatures(features...)
-}
-
-func (d *dsImpl) Kind(src interface{}) string {
-        return kind(d.ns, d.KindNameResolver(), src)
-}
-
-func (d *dsImpl) KindNameResolver() goon.KindNameResolver {
-        return d.data.KindNameResolver()
-}
-func (d *dsImpl) SetKindNameResolver(knr goon.KindNameResolver) {
-        d.data.SetKindNameResolver(knr)
-}
-
-func (d *dsImpl) NewKey(kind, stringID string, intID int64, parent *datastore.Key) *datastore.Key {
-        return newKey(d.ns, kind, stringID, intID, parent)
-}
-func (d *dsImpl) NewKeyObj(src interface{}) *datastore.Key {
-        return newKeyObj(d.ns, d.KindNameResolver(), src)
-}
-func (d *dsImpl) NewKeyObjError(src interface{}) (*datastore.Key, error) {
-        return newKeyObjError(d.ns, d.KindNameResolver(), src)
-}
-
-func (d *dsImpl) Put(src interface{}) (*datastore.Key, error) {
-        if err := d.data.IsBroken(); err != nil {
-                return nil, err
-        }
-        return d.data.put(d.ns, src)
-}
-
-func (d *dsImpl) Get(dst interface{}) error {
-        if err := d.data.IsBroken(); err != nil {
-                return err
-        }
-        return d.data.get(d.ns, dst)
-}
-
-func (d *dsImpl) Delete(key *datastore.Key) error {
-        if err := d.data.IsBroken(); err != nil {
-                return err
-        }
-        return d.data.del(d.ns, key)
-}
-
-////////////////////////////////// txnDsImpl ///////////////////////////////////
-
-type txnDsImpl struct {
-        wrapper.Datastore
-
-        data *txnDataStoreData
-        ns   string
-}
-
-var (
-        _ = wrapper.Datastore((*txnDsImpl)(nil))
-        _ = wrapper.Testable((*txnDsImpl)(nil))
-)
-
-func (d *dsImpl) NewQuery(kind string) wrapper.DSQuery {
-        return &queryImpl{DSQuery: wrapper.DummyQY(), ns: d.ns, kind: kind}
-}
-
-func (d *dsImpl) Run(q wrapper.DSQuery) wrapper.DSIterator {
-        rq := q.(*queryImpl)
-        rq = rq.normalize().checkCorrectness(d.ns, false)
-        return &queryIterImpl{rq}
-}
-
-func (d *dsImpl) GetAll(q wrapper.DSQuery, dst interface{}) ([]*datastore.Key, error) {
-        // TODO(riannucci): assert that dst is a slice of structs
-        return nil, nil
-}
-
-func (d *dsImpl) Count(q wrapper.DSQuery) (ret int, err error) {
-        itr := d.Run(q.KeysOnly())
-        for _, err = itr.Next(nil); err != nil; _, err = itr.Next(nil) {
-                ret++
-        }
-        if err == datastore.Done {
-                err = nil
-        }
-        return
-}
-
-func (d *txnDsImpl) BreakFeatures(err error, features ...string) {
-        d.data.BreakFeatures(err, features...)
-}
-func (d *txnDsImpl) UnbreakFeatures(features ...string) {
-        d.data.UnbreakFeatures(features...)
-}
-
-func (d *txnDsImpl) Kind(src interface{}) string {
-        return kind(d.ns, d.KindNameResolver(), src)
-}
-
-func (d *txnDsImpl) KindNameResolver() goon.KindNameResolver {
-        return d.data.KindNameResolver()
-}
-func (d *txnDsImpl) SetKindNameResolver(knr goon.KindNameResolver) {
-        d.data.SetKindNameResolver(knr)
-}
-
-func (d *txnDsImpl) NewKey(kind, stringID string, intID int64, parent *datastore.Key) *datastore.Key {
-        return newKey(d.ns, kind, stringID, intID, parent)
-}
-func (d *txnDsImpl) NewKeyObj(src interface{}) *datastore.Key {
-        return newKeyObj(d.ns, d.KindNameResolver(), src)
-}
-func (d *txnDsImpl) NewKeyObjError(src interface{}) (*datastore.Key, error) {
-        return newKeyObjError(d.ns, d.KindNameResolver(), src)
-}
-
-func (d *txnDsImpl) Put(src interface{}) (*datastore.Key, error) {
-        if err := d.data.IsBroken(); err != nil {
-                return nil, err
-        }
-        return d.data.put(d.ns, src)
-}
-
-func (d *txnDsImpl) Get(dst interface{}) error {
-        if err := d.data.IsBroken(); err != nil {
-                return err
-        }
-        return d.data.get(d.ns, dst)
-}
-
-func (d *txnDsImpl) Delete(key *datastore.Key) error {
-        if err := d.data.IsBroken(); err != nil {
-                return err
-        }
-        return d.data.del(d.ns, key)
-}
-
-func (*txnDsImpl) RunInTransaction(func(c context.Context) error, *datastore.TransactionOptions) error {
-        return errors.New("datastore: nested transactions are not supported")
-}
-
-////////////////////////////// private functions ///////////////////////////////
-
-func newDSError(code pb.Error_ErrorCode, message ...string) *appengine_internal.APIError {
-        return &appengine_internal.APIError{
-                Detail:  strings.Join(message, ""),
-                Service: "datastore_v3",
-                Code:    int32(code),
-        }
-}