Merge branch 'master' of github.com:gophergala2016/source

Author: gotokatsuya

core/foundation/context.go

@@ -53,6 +53,7 @@ type (
 	}
 
 	Request interface {
+		Ref() *http.Request
 		Clone() *http.Request
 		Body() []byte
 		Method() string
@@ -204,6 +205,11 @@ func (c *FoundationContext) JSON(code int, obj interface{}) {
 /// Request interface
 ////////////////////////////////////////
 
+// Ref returns the request of the Context.
+func (c *FoundationContext) Ref() *http.Request {
+	return c.opt.Request
+}
+
 // Clone returns the cloned request of the Context.
 func (c *FoundationContext) Clone() *http.Request {
 	cp := *c.opt.Request

core/models/appengine/datastore.go

@@ -0,0 +1,42 @@
+package appengine
+
+import (
+	"github.com/gophergala2016/source/core/foundation"
+	"golang.org/x/net/context"
+
+	"google.golang.org/appengine"
+	"google.golang.org/appengine/datastore"
+)
+
+type (
+	Datastore struct {
+		ctx context.Context
+	}
+	IDatastore interface {
+	}
+	Key struct {
+		o *datastore.Key
+	}
+)
+
+func NewDatastore(ctx foundation.Context) *Datastore {
+	return &Datastore{
+		ctx: appengine.NewContext(ctx.Request().Ref()),
+	}
+}
+
+func (d *Datastore) NewKey(kind, stringID string, intID int64, parent *Key) *Key {
+	return &Key{datastore.NewKey(d.ctx, kind, stringID, intID, parent.o)}
+}
+
+func (d *Datastore) Put(key *Key, src interface{}) (*Key, error) {
+	k, err := datastore.Put(d.ctx, key.o, src)
+	if err != nil {
+		return nil, err
+	}
+	return &Key{k}, nil
+}
+
+func (d *Datastore) Get(key *Key, dst interface{}) error {
+	return datastore.Get(d.ctx, key.o, dst)
+}

core/models/tag_entity_repository.go

@@ -2,15 +2,18 @@ package models
 
 import (
 	"github.com/gophergala2016/source/core/foundation"
+	"github.com/gophergala2016/source/core/models/appengine"
 )
 
 type TagRepository struct {
 	RootRepository
+	*appengine.Datastore
 }
 
 func NewTagRepository(ctx foundation.Context) *TagRepository {
 	return &TagRepository{
 		RootRepository: NewRootRepository(ctx),
+		Datastore:      appengine.NewDatastore(ctx),
 	}
 }
 

core/models/tag_entity_repository_test.go

@@ -0,0 +1,22 @@
+package models
+
+import (
+	"testing"
+
+	"github.com/gophergala2016/source/core/foundation/foundationtest"
+	"github.com/stretchr/testify/assert"
+)
+
+// TestTagEntityRepository ...
+func TestTagEntityRepository(t *testing.T) {
+	assert := assert.New(t)
+
+	ctx := foundationtest.NewContext()
+	repo := NewTagRepository(ctx)
+
+	ent := NewTag("foo", "#000000", 0)
+	k := repo.NewKey("Tag", "", 0, nil)
+	key, err := repo.Put(k, &ent)
+	assert.NoError(err)
+	_ = key
+}