Simplify memory package interface.

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
+// useDS adds a wrapper.Datastore implementation to context, accessible
 // by wrapper.GetDS(c)
-func UseDS(c context.Context) context.Context {
+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))
@@ skipping 135 matching lines @@
 
 ////////////////////////////// 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),
         }
 }