Better attempt at an appengine wrapper.

go/src/infra/gae/libs/wrapper/gae/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 gae
+
+import (
+        "golang.org/x/net/context"
+
+        "appengine/datastore"
+
+        "github.com/mjibson/goon"
+
+        "infra/gae/libs/wrapper"
+)
+
+// UseDS adds a wrapper.Datastore implementation to context, accessible
+// by wrapper.DS(c)

[Vadim Sh., 2015/05/24 19:01:57]

wrapper.GetDS(c)?

[iannucci, 2015/05/24 20:04:31]

ah, oops. Fixed.

+func UseDS(c context.Context) context.Context {
+        return wrapper.SetDSFactory(c, func(ci context.Context) wrapper.Datastore {
+                return dsImpl{ctx(ci), ci}

[Vadim Sh., 2015/05/24 19:01:57]

how do you choose when to put (or not to put) & when returning interfaces?

[iannucci, 2015/05/24 20:04:31]

Not sure I understand what you mean?

[Vadim Sh., 2015/05/24 20:15:36]

I would have written:
return &dsImpl{ctx(ci), ci}

Because value types boxed into an interface look alien to me (since I'm used to
"traditional" ref-based languages).

+        })
+}
+
+////////// Query
+
+type queryImpl struct{ *datastore.Query }
+
+func (q queryImpl) Distinct() wrapper.DSQuery { return queryImpl{q.Query.Distinct()} }
+func (q queryImpl) End(c wrapper.DSCursor) wrapper.DSQuery {
+        return queryImpl{q.Query.End(c.(datastore.Cursor))}
+}
+func (q queryImpl) EventualConsistency() wrapper.DSQuery {
+        return queryImpl{q.Query.EventualConsistency()}
+}
+func (q queryImpl) KeysOnly() wrapper.DSQuery              { return queryImpl{q.Query.KeysOnly()} }
+func (q queryImpl) Limit(limit int) wrapper.DSQuery        { return queryImpl{q.Query.Limit(limit)} }
+func (q queryImpl) Offset(offset int) wrapper.DSQuery      { return queryImpl{q.Query.Offset(offset)} }
+func (q queryImpl) Order(fieldName string) wrapper.DSQuery { return queryImpl{q.Query.Order(fieldName)} }
+func (q queryImpl) Start(c wrapper.DSCursor) wrapper.DSQuery {
+        return queryImpl{q.Query.Start(c.(datastore.Cursor))}
+}
+func (q queryImpl) Ancestor(ancestor *datastore.Key) wrapper.DSQuery {
+        return queryImpl{q.Query.Ancestor(ancestor)}
+}
+func (q queryImpl) Project(fieldNames ...string) wrapper.DSQuery {
+        return queryImpl{q.Query.Project(fieldNames...)}
+}
+func (q queryImpl) Filter(filterStr string, value interface{}) wrapper.DSQuery {
+        return queryImpl{q.Query.Filter(filterStr, value)}
+}
+
+////////// Iterator
+
+type iteratorImpl struct{ *goon.Iterator }
+
+func (i iteratorImpl) Cursor() (wrapper.DSCursor, error) { return i.Iterator.Cursor() }
+
+////////// Datastore
+
+type dsImpl struct {
+        *goon.Goon
+        c context.Context
+}
+
+// logger
+func (g dsImpl) Debugf(format string, args ...interface{})   { g.Goon.Context.Debugf(format, args...) }
+func (g dsImpl) Infof(format string, args ...interface{})    { g.Goon.Context.Infof(format, args...) }
+func (g dsImpl) Warningf(format string, args ...interface{}) { g.Goon.Context.Warningf(format, args...) }
+func (g dsImpl) Errorf(format string, args ...interface{})   { g.Goon.Context.Errorf(format, args...) }
+
+// Kinder
+func (g dsImpl) KindNameResolver() goon.KindNameResolver       { return g.Goon.KindNameResolver }
+func (g dsImpl) SetKindNameResolver(knr goon.KindNameResolver) { g.Goon.KindNameResolver = knr }
+
+// NewKeyer
+func (g dsImpl) NewKey(kind, stringID string, intID int64, parent *datastore.Key) *datastore.Key {
+        return datastore.NewKey(g.Goon.Context, kind, stringID, intID, parent)
+}
+func (g dsImpl) NewKeyObj(obj interface{}) *datastore.Key               { return g.Key(obj) }
+func (g dsImpl) NewKeyObjError(obj interface{}) (*datastore.Key, error) { return g.KeyError(obj) }
+
+// DSQueryer
+func (g dsImpl) NewQuery(kind string) wrapper.DSQuery { return queryImpl{datastore.NewQuery(kind)} }
+func (g dsImpl) Run(q wrapper.DSQuery) wrapper.DSIterator {
+        return iteratorImpl{g.Goon.Run(q.(queryImpl).Query)}
+}
+func (g dsImpl) Count(q wrapper.DSQuery) (int, error) { return g.Goon.Count(q.(queryImpl).Query) }
+func (g dsImpl) GetAll(q wrapper.DSQuery, dst interface{}) ([]*datastore.Key, error) {
+        return g.Goon.GetAll(q.(queryImpl).Query, dst)
+}
+
+// Transactioner
+func (g dsImpl) RunInTransaction(f func(c context.Context) error, opts *datastore.TransactionOptions) error {
+        return g.Goon.RunInTransaction(func(ig *goon.Goon) error {
+                return f(context.WithValue(g.c, goonContextKey, ig))
+        }, (*datastore.TransactionOptions)(opts))
+}