mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-05-22 15:53:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reexport basic mongo structs

Browse Source
This commit is contained in:
Nikita Tokarchuk
2020-03-13 00:40:33 +01:00
parent 8267abe5be
commit e7a05d94e1
9 changed files with 68 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mongox/database/count.go
+2 -2
View File
@@ -3,9 +3,9 @@ package database
import (
"fmt"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
)
@@ -21,7 +21,7 @@ func (d *Database) Count(target interface{}, filters ...interface{}) (int64, err
opts.Skip = composed.Skipper()
result, err := collection.CountDocuments(d.Context(), composed.M(), opts)
if err == mongo.ErrNoDocuments {
if err == mongox.ErrNoDocuments {
return 0, err
}
if err != nil {
mongox/database/database.go
+6 -7
View File
@@ -8,7 +8,6 @@ import (
"strings"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/mainnika/mongox-go-driver/v2/mongox"
@@ -17,13 +16,13 @@ import (
// Database handler
type Database struct {
client *mongo.Client
client *mongox.Client
dbname string
ctx context.Context
}
// NewDatabase function creates new database instance with mongo client and empty context
func NewDatabase(client *mongo.Client, dbname string) mongox.Database {
func NewDatabase(client *mongox.Client, dbname string) mongox.Database {
db := &Database{}
db.client = client
@@ -33,7 +32,7 @@ func NewDatabase(client *mongo.Client, dbname string) mongox.Database {
}
// Client function returns a mongo client
func (d *Database) Client() *mongo.Client {
func (d *Database) Client() *mongox.Client {
return d.client
}
@@ -67,7 +66,7 @@ func (d *Database) New(ctx context.Context) mongox.Database {
// base.ObjectID `bson:",inline" json:",inline" collection:"foobars"`
// ...
// Will panic if there is no «collection» tag
func (d *Database) GetCollectionOf(document interface{}) *mongo.Collection {
func (d *Database) GetCollectionOf(document interface{}) *mongox.Collection {
el := reflect.TypeOf(document).Elem()
numField := el.NumField()
@@ -86,7 +85,7 @@ func (d *Database) GetCollectionOf(document interface{}) *mongo.Collection {
panic(fmt.Errorf("document %v does not have a collection tag", document))
}
func (d *Database) createSimpleLoad(target interface{}, composed *query.Query) (cursor *mongo.Cursor, err error) {
func (d *Database) createSimpleLoad(target interface{}, composed *query.Query) (cursor *mongox.Cursor, err error) {
collection := d.GetCollectionOf(target)
opts := options.Find()
@@ -98,7 +97,7 @@ func (d *Database) createSimpleLoad(target interface{}, composed *query.Query) (
return collection.Find(d.Context(), composed.M(), opts)
}
func (d *Database) createAggregateLoad(target interface{}, composed *query.Query) (cursor *mongo.Cursor, err error) {
func (d *Database) createAggregateLoad(target interface{}, composed *query.Query) (cursor *mongox.Cursor, err error) {
collection := d.GetCollectionOf(target)
opts := options.Aggregate()
mongox/database/deleteone.go
+2 -2
View File
@@ -5,9 +5,9 @@ import (
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
)
@@ -38,7 +38,7 @@ func (d *Database) DeleteOne(target interface{}, filters ...interface{}) error {
}
err := result.Decode(target)
if err == mongo.ErrNoDocuments {
if err == mongox.ErrNoDocuments {
return err
}
if err != nil {
mongox/database/loadarray.go
+2 -3
View File
@@ -4,8 +4,7 @@ import (
"fmt"
"reflect"
"go.mongodb.org/mongo-driver/mongo"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
)
@@ -36,7 +35,7 @@ func (d *Database) LoadArray(target interface{}, filters ...interface{}) error {
zeroElem := reflect.Zero(targetSliceElemT)
hasPreloader, _ := composed.Preloader()
var result *mongo.Cursor
var result *mongox.Cursor
var err error
if hasPreloader {
mongox/database/loadone.go
+3 -4
View File
@@ -3,8 +3,7 @@ package database
import (
"fmt"
"go.mongodb.org/mongo-driver/mongo"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
)
@@ -15,7 +14,7 @@ func (d *Database) LoadOne(target interface{}, filters ...interface{}) error {
composed := query.Compose(append(filters, query.Limit(1))...)
hasPreloader, _ := composed.Preloader()
var result *mongo.Cursor
var result *mongox.Cursor
var err error
if hasPreloader {
@@ -32,7 +31,7 @@ func (d *Database) LoadOne(target interface{}, filters ...interface{}) error {
return err
}
if !hasNext {
return mongo.ErrNoDocuments
return mongox.ErrNoDocuments
}
base.Reset(target)
mongox/database/loadstream.go
+2 -4
View File
@@ -3,8 +3,6 @@ package database
import (
"fmt"
"go.mongodb.org/mongo-driver/mongo"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
)
@@ -12,7 +10,7 @@ import (
// LoadStream function loads documents one by one into a target channel
func (d *Database) LoadStream(target interface{}, filters ...interface{}) (mongox.StreamLoader, error) {
var cursor *mongo.Cursor
var cursor *mongox.Cursor
var err error
composed := query.Compose(filters...)
@@ -27,7 +25,7 @@ func (d *Database) LoadStream(target interface{}, filters ...interface{}) (mongo
return nil, fmt.Errorf("can't create find result: %w", err)
}
l := &StreamLoader{Cursor: cursor, ctx: d.Context(), target: target}
l := &StreamLoader{cur: cursor, ctx: d.Context(), target: target}
return l, nil
}
mongox/database/streamloader.go
+21 -15
View File
@@ -4,14 +4,13 @@ import (
"context"
"fmt"
"go.mongodb.org/mongo-driver/mongo"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
)
// StreamLoader is a controller for a database cursor
type StreamLoader struct {
*mongo.Cursor
cur *mongox.Cursor
ctx context.Context
target interface{}
}
@@ -19,18 +18,18 @@ type StreamLoader struct {
// DecodeNext loads next documents to a target or returns an error
func (l *StreamLoader) DecodeNext() error {
hasNext := l.Cursor.Next(l.ctx)
hasNext := l.cur.Next(l.ctx)
if l.Cursor.Err() != nil {
return l.Cursor.Err()
if l.cur.Err() != nil {
return l.cur.Err()
}
if !hasNext {
return mongo.ErrNoDocuments
return mongox.ErrNoDocuments
}
base.Reset(l.target)
err := l.Cursor.Decode(l.target)
err := l.cur.Decode(l.target)
if err != nil {
return fmt.Errorf("can't decode desult: %w", err)
}
@@ -43,7 +42,7 @@ func (l *StreamLoader) Decode() error {
base.Reset(l.target)
err := l.Cursor.Decode(l.target)
err := l.cur.Decode(l.target)
if err != nil {
return fmt.Errorf("can't decode desult: %w", err)
}
@@ -54,20 +53,27 @@ func (l *StreamLoader) Decode() error {
// Next loads next documents but doesn't perform decoding
func (l *StreamLoader) Next() error {
hasNext := l.Cursor.Next(l.ctx)
hasNext := l.cur.Next(l.ctx)
if l.Cursor.Err() != nil {
return l.Cursor.Err()
if l.cur.Err() != nil {
return l.cur.Err()
}
if !hasNext {
return mongo.ErrNoDocuments
return mongox.ErrNoDocuments
}
return nil
}
func (l *StreamLoader) Cursor() *mongox.Cursor {
return l.cur
}
// Close cursor
func (l *StreamLoader) Close() error {
return l.Cursor.Close(l.ctx)
return l.cur.Close(l.ctx)
}
func (l *StreamLoader) Err() error {
return l.cur.Err()
}