client set key after get

acoshift · Mar 13, 2017 · 29fd95a · 29fd95a
1 parent f3cf382
commit 29fd95a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/get.go b/get.go
@@ -10,6 +10,7 @@ import (
 // GetByKey retrieves model from datastore by key
 func (client *Client) GetByKey(ctx context.Context, key *datastore.Key, dst interface{}) error {
 	err := client.Get(ctx, key, dst)
+	SetKey(key, dst)
 	if err != nil {
 		return err
 	}
@@ -29,6 +30,7 @@ func (client *Client) GetByKeys(ctx context.Context, keys []*datastore.Key, dst
 	}
 
 	err := client.GetMulti(ctx, keys, dst)
+	SetKeys(keys, dst)
 	if err != nil {
 		return err
 	}