Implement ID bases in packages

mongox/base/jsonbased/id.go

@@ -0,0 +1,24 @@
+package jsonbased
+
+import (
+	"go.mongodb.org/mongo-driver/bson/primitive"
+
+	"github.com/mainnika/mongox-go-driver/v2/mongox"
+)
+
+var _ mongox.JSONBased = (*Primary)(nil)
+
+// Primary is a structure with object as an _id field
+type Primary struct {
+	ID primitive.D `bson:"_id" json:"_id"`
+}
+
+// GetID returns an _id
+func (p *Primary) GetID() primitive.D {
+	return p.ID
+}
+
+// SetID sets an _id
+func (p *Primary) SetID(id primitive.D) {
+	p.ID = id
+}

mongox/base/jsonbased/id_test.go

@@ -0,0 +1,80 @@
+package jsonbased
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+
+	"github.com/mainnika/mongox-go-driver/v2/mongox/tempdb"
+)
+
+func Test_GetID(t *testing.T) {
+
+	type DocWithObject struct {
+		Primary `bson:",inline" json:",inline" collection:"1"`
+	}
+
+	doc := &DocWithObject{Primary{primitive.D{{"1", "one"}, {"2", "two"}}}}
+
+	assert.Equal(t, primitive.D{{"1", "one"}, {"2", "two"}}, doc.GetID())
+}
+
+func Test_SetID(t *testing.T) {
+
+	type DocWithObject struct {
+		Primary `bson:",inline" json:",inline" collection:"1"`
+	}
+
+	doc := &DocWithObject{Primary{primitive.D{{"1", "one"}, {"2", "two"}}}}
+
+	doc.SetID(primitive.D{{"3", "three"}, {"4", "you"}})
+
+	assert.Equal(t, primitive.D{{"3", "three"}, {"4", "you"}}, doc.Primary.ID)
+	assert.Equal(t, primitive.D{{"3", "three"}, {"4", "you"}}, doc.GetID())
+}
+
+func Test_SaveLoad(t *testing.T) {
+
+	type DocWithObjectID struct {
+		Primary `bson:",inline" json:",inline" collection:"1"`
+	}
+
+	db, err := tempdb.NewTempDB("mongodb://localhost")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	defer db.Close()
+
+	doc1 := &DocWithObjectID{Primary{primitive.D{{"1", "one"}, {"2", "two"}}}}
+	doc2 := &DocWithObjectID{}
+
+	err = db.SaveOne(doc1)
+	assert.NoError(t, err)
+
+	err = db.LoadOne(doc2)
+	assert.NoError(t, err)
+
+	assert.Equal(t, doc1, doc2)
+
+	bytes1, _ := json.Marshal(doc1)
+	bytes2, _ := json.Marshal(doc2)
+
+	assert.Equal(t, bytes1, bytes2)
+}
+
+func Test_Marshal(t *testing.T) {
+
+	type DocWithObjectID struct {
+		Primary `bson:",inline" json:",inline" collection:"1"`
+	}
+
+	id := primitive.D{{"1", "one"}, {"2", "two"}}
+	doc := &DocWithObjectID{Primary{id}}
+
+	bytes, err := json.Marshal(doc)
+	assert.NoError(t, err)
+	assert.Equal(t, `{"_id":[{"Key":"1","Value":"one"},{"Key":"2","Value":"two"}]}`, string(bytes))
+}