Implement ID bases in packages

mongox/base/oidbased/id.go

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

mongox/base/oidbased/id_test.go

@@ -0,0 +1,80 @@
+package oidbased
+
+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 DocWithObjectID struct {
+		Primary `bson:",inline" json:",inline" collection:"1"`
+	}
+
+	doc := &DocWithObjectID{Primary{[12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}}}
+
+	assert.Equal(t, primitive.ObjectID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}), doc.GetID())
+}
+
+func Test_SetID(t *testing.T) {
+
+	type DocWithObjectID struct {
+		Primary `bson:",inline" json:",inline" collection:"1"`
+	}
+
+	doc := &DocWithObjectID{}
+
+	doc.SetID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2})
+
+	assert.Equal(t, primitive.ObjectID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}), doc.Primary.ID)
+	assert.Equal(t, primitive.ObjectID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}), 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{}
+	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.ObjectIDFromHex("feadbeeffeadbeeffeadbeef")
+	doc := &DocWithObjectID{Primary{id}}
+
+	bytes, err := json.Marshal(doc)
+	assert.NoError(t, err)
+	assert.Equal(t, `{"_id":"feadbeeffeadbeeffeadbeef"}`, string(bytes))
+}