| Index: go/src/infra/gae/epservice/example/service_add.go
|
| diff --git a/go/src/infra/gae/epservice/example/service_add.go b/go/src/infra/gae/epservice/example/service_add.go
|
| index da259b0d2b9840842aa40ca99c6b16322ac9a936..46064977add4f2c5bfea553187db16a4c48ba5ec 100644
|
| --- a/go/src/infra/gae/epservice/example/service_add.go
|
| +++ b/go/src/infra/gae/epservice/example/service_add.go
|
| @@ -6,9 +6,8 @@ package example
|
|
|
| import (
|
| "golang.org/x/net/context"
|
| - "infra/gae/libs/wrapper"
|
| - "infra/gae/libs/wrapper/gae"
|
| - "infra/gae/libs/wrapper/gae/commonErrors"
|
| + "infra/gae/libs/gae"
|
| + "infra/gae/libs/gae/prod"
|
|
|
| "github.com/GoogleCloudPlatform/go-endpoints/endpoints"
|
| )
|
| @@ -31,19 +30,21 @@ type AddRsp struct {
|
|
|
| // Add adds a value to the current counter, and returns the old+new values. It
|
| // may cause a counter to come into existance.
|
| -func (Example) Add(c endpoints.Context, r *AddReq) (rsp *AddRsp, err error) {
|
| +func (Example) Add(c context.Context, r *AddReq) (rsp *AddRsp, err error) {
|
| rsp = &AddRsp{}
|
|
|
| - ds := wrapper.GetDS(gae.Use(context.Background(), c))
|
| - err = ds.RunInTransaction(func(context.Context) error {
|
| - ctr := &Counter{ID: r.Name}
|
| - if err := ds.Get(ctr); err != nil && err != commonErrors.ErrNoSuchEntityDS {
|
| + c = prod.Use(c)
|
| + err = gae.GetRDS(c).RunInTransaction(func(c context.Context) error {
|
| + rds := gae.GetRDS(c)
|
| + ctr := &Counter{}
|
| + key := rds.NewKey("Counter", r.Name, 0, nil)
|
| + if err := rds.Get(key, ctr); err != nil && err != gae.ErrDSNoSuchEntity {
|
| return err
|
| }
|
| rsp.Prev = ctr.Val
|
| ctr.Val += r.Delta
|
| rsp.Cur = ctr.Val
|
| - _, err := ds.Put(ctr)
|
| + _, err := rds.Put(key, ctr)
|
| return err
|
| }, nil)
|
| return
|
|
|