add update document (#4)

Author: chessman

client.go

@@ -69,7 +69,7 @@ type Clienter interface {
 	Delete(ctx context.Context, link string, headers map[string] string) error
 	Query(ctx context.Context, link string, qu *Query, ret interface{}) (token string, err error)
 	Create(ctx context.Context, link string, body, ret interface{}, headers map[string]string) error
-	Replace(ctx context.Context, link string, body, ret interface{}) error
+	Replace(ctx context.Context, link string, body, ret interface{}, headers map[string]string) error
 	Execute(ctx context.Context, link string, body, ret interface{}) error
 }
 
@@ -134,13 +134,13 @@ func (c *Client) Create(ctx context.Context, link string, body, ret interface{},
 }
 
 // Replace resource
-func (c *Client) Replace(ctx context.Context, link string, body, ret interface{}) error {
+func (c *Client) Replace(ctx context.Context, link string, body, ret interface{}, headers map[string]string) error {
 	data, err := stringify(body)
 	if err != nil {
 		return err
 	}
 	buf := bytes.NewBuffer(data)
-	_, err = c.method(ctx, "PUT", link, ret, buf, nil)
+	_, err = c.method(ctx, "PUT", link, ret, buf, headers)
 	return err
 }
 

client_test.go

@@ -3,11 +3,12 @@ package documentdb
 import (
 	"context"
 	"fmt"
-	"github.com/stretchr/testify/assert"
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 // I more interested in the request, instead of the response
@@ -161,21 +162,21 @@ func TestReplace(t *testing.T) {
 
 	// First call
 	var db Database
-	err := client.Replace(ctx, "dbs", `{"id": 3}`, &db)
+	err := client.Replace(ctx, "dbs", `{"id": 3}`, &db, nil)
 	s.AssertHeaders(t, HEADER_XDATE, HEADER_AUTH, HEADER_VER)
 	assert.Equal(db.Colls, "colls", "Should fill the fields from response body")
 	assert.Nil(err, "err should be nil")
 
 	// Second call
 	var doc, tDoc Document
 	tDoc.Id = "9"
-	err = client.Replace(ctx, "dbs", tDoc, &doc)
+	err = client.Replace(ctx, "dbs", tDoc, &doc, nil)
 	s.AssertHeaders(t, HEADER_XDATE, HEADER_AUTH, HEADER_VER)
 	assert.Equal(doc.Id, "9", "Should fill the fields from response body")
 	assert.Nil(err, "err should be nil")
 
 	// Last Call, when StatusCode != StatusOK && StatusCreated
-	err = client.Replace(ctx, "dbs", tDoc, &doc)
+	err = client.Replace(ctx, "dbs", tDoc, &doc, nil)
 	assert.Equal(err.Error(), "500, DocumentDB error")
 }
 

documentdb.go

@@ -132,6 +132,7 @@ type Col struct {
 	db *DB
 	Collection
 }
+
 func (c *Col) ctx(ctx context.Context) context.Context {
 	return context.WithValue(ctx, collKey{}, string(c.Collection.Id))
 }
@@ -147,6 +148,10 @@ func (c *Col) CreateDocument(ctx context.Context, doc interface{}) error {
 	return c.db.c.CreateDocument(c.ctx(ctx), c.Self, doc)
 }
 
+func (c *Col) UpdateDocument(ctx context.Context, doc interface{}, etag string) error {
+	return c.db.c.UpdateDocument(c.ctx(ctx), c.Self, doc, etag)
+}
+
 func (c *Col) UpsertDocument(ctx context.Context, doc interface{}, etag string) error {
 	return c.db.c.UpsertDocument(c.ctx(ctx), c.Self, doc, etag)
 }
@@ -370,6 +375,36 @@ func (c *DocumentDB) CreateDocument(ctx context.Context, coll string, doc interf
 	return c.createDocument(ctx, coll, doc, nil)
 }
 
+func (c *DocumentDB) UpdateDocument(ctx context.Context, coll string, doc interface{}, etag string) error {
+	rv := reflect.ValueOf(doc)
+	if rv.Kind() == reflect.Ptr {
+		rv = rv.Elem()
+	}
+
+	id := rv.FieldByName("Id")
+	if !id.IsValid() || id.String() == "" {
+		id = rv.FieldByName("ID")
+		if !id.IsValid() || id.String() == "" {
+			return errors.New("document doesn't have id")
+		}
+	}
+
+	var docs []Document
+	_, err := c.QueryDocuments(ctx, coll, selectByID(id.String()), &docs)
+	if err != nil {
+		return err
+	}
+	if len(docs) == 0 {
+		return ErrNotFound
+	}
+
+	headers := make(map[string]string)
+	if etag != "" {
+		headers[HEADER_IF_MATCH] = etag
+	}
+	return c.ReplaceDocument(ctx, coll+"docs/"+docs[0].Id, doc, headers)
+}
+
 // Create document
 func (c *DocumentDB) UpsertDocument(ctx context.Context, coll string, doc interface{}, etag string) error {
 	headers := map[string]string{
@@ -413,21 +448,21 @@ func (c *DocumentDB) DeleteUserDefinedFunction(ctx context.Context, link string)
 
 // Replace database
 func (c *DocumentDB) ReplaceDatabase(ctx context.Context, link string, body interface{}) (db *Database, err error) {
-	err = c.client.Replace(ctx, link, body, &db)
+	err = c.client.Replace(ctx, link, body, &db, nil)
 	if err != nil {
 		return nil, err
 	}
 	return
 }
 
 // Replace document
-func (c *DocumentDB) ReplaceDocument(ctx context.Context, link string, doc interface{}) error {
-	return c.client.Replace(ctx, link, doc, &doc)
+func (c *DocumentDB) ReplaceDocument(ctx context.Context, link string, doc interface{}, headers map[string]string) error {
+	return c.client.Replace(ctx, link, doc, &doc, headers)
 }
 
 // Replace stored procedure
 func (c *DocumentDB) ReplaceStoredProcedure(ctx context.Context, link string, body interface{}) (sproc *Sproc, err error) {
-	err = c.client.Replace(ctx, link, body, &sproc)
+	err = c.client.Replace(ctx, link, body, &sproc, nil)
 	if err != nil {
 		return nil, err
 	}
@@ -436,7 +471,7 @@ func (c *DocumentDB) ReplaceStoredProcedure(ctx context.Context, link string, bo
 
 // Replace stored procedure
 func (c *DocumentDB) ReplaceUserDefinedFunction(ctx context.Context, link string, body interface{}) (udf *UDF, err error) {
-	err = c.client.Replace(ctx, link, body, &udf)
+	err = c.client.Replace(ctx, link, body, &udf, nil)
 	if err != nil {
 		return nil, err
 	}

documentdb_test.go

@@ -2,9 +2,10 @@ package documentdb
 
 import (
 	"context"
+	"testing"
+
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
-	"testing"
 )
 
 type ClientStub struct {
@@ -21,12 +22,12 @@ func (c *ClientStub) Create(ctx context.Context, link string, body, ret interfac
 	return nil
 }
 
-func (c *ClientStub) Delete(ctx context.Context, link string, headers map[string] string) error {
+func (c *ClientStub) Delete(ctx context.Context, link string, headers map[string]string) error {
 	c.Called(link)
 	return nil
 }
 
-func (c *ClientStub) Replace(ctx context.Context, link string, body, ret interface{}) error {
+func (c *ClientStub) Replace(ctx context.Context, link string, body, ret interface{}, headers map[string]string) error {
 	c.Called(link, body)
 	return nil
 }
@@ -279,7 +280,7 @@ func TestReplaceDocument(t *testing.T) {
 	c := &DocumentDB{client}
 	client.On("Replace", "doc_link", "{}").Return(nil)
 	ctx := context.Background()
-	c.ReplaceDocument(ctx, "doc_link", "{}")
+	c.ReplaceDocument(ctx, "doc_link", "{}", nil)
 	client.AssertCalled(t, "Replace", "doc_link", "{}")
 }