database.go

database.go

package tracking

import (
	"context"
)

type InsertOneResult struct {
	InsertedID interface{} `bson:"insertedId"`
}

type Collection interface {
	InsertOne(ctx context.Context, document interface{}) (*InsertOneResult, error)
	UpdateOne(ctx context.Context, filter interface{}, update interface{}) (interface{}, error)
	DeleteOne(ctx context.Context, filter interface{}) (interface{}, error)
	FindOne(ctx context.Context, filter interface{}) (interface{}, error)
	Find(ctx context.Context, filter interface{}) ([]interface{}, error)
}

type Database interface {
	Collection(name string) Collection
}



// mongodb driver mongo.go

package tracking

import (
	"context"
	"go.mongodb.org/mongo-driver/mongo"
)

type mongoDB struct {
	db *mongo.Database
}

type mongoCollection struct {
	name string
	col  *mongo.Collection
}

func (m *mongoCollection) DeleteOne(ctx context.Context, filter interface{}) (interface{}, error) {
	//TODO implement me
	panic("implement me")
}

func (m *mongoCollection) InsertOne(ctx context.Context, document interface{}) (*InsertOneResult, error) {
	res, err := m.col.InsertOne(ctx, document)
	if err != nil {
		return nil, err
	}
	return &InsertOneResult{InsertedID: res.InsertedID}, nil
}

func (m *mongoCollection) UpdateOne(ctx context.Context, filter interface{}, update interface{}) (interface{}, error) {
	return m.col.UpdateOne(ctx, filter, update)
}

func newMongoDB(db *mongo.Database) *mongoDB {
	return &mongoDB{
		db: db,
	}
}

func (m *mongoCollection) FindOne(ctx context.Context, filter interface{}) (interface{}, error) {
	//TODO implement me
	panic("implement me")
}

func (m *mongoCollection) Find(ctx context.Context, filter interface{}) ([]interface{}, error) {
	//TODO implement me
	panic("implement me")
}

func (m *mongoDB) Collection(name string) Collection {
	return &mongoCollection{
		name: name,
		col:  m.db.Collection(name),
	}
}