Add redis
This commit is contained in:
20
mongo.go
20
mongo.go
@@ -5,12 +5,12 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/mongodb/mongo-go-driver/mongo"
|
||||
"github.com/mongodb/mongo-go-driver/mongo/options"
|
||||
"github.com/mongodb/mongo-go-driver/mongo/readconcern"
|
||||
"github.com/mongodb/mongo-go-driver/mongo/readpref"
|
||||
"github.com/mongodb/mongo-go-driver/mongo/writeconcern"
|
||||
"gopkg.in/mgo.v2/bson"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
"go.mongodb.org/mongo-driver/mongo/readconcern"
|
||||
"go.mongodb.org/mongo-driver/mongo/readpref"
|
||||
"go.mongodb.org/mongo-driver/mongo/writeconcern"
|
||||
)
|
||||
|
||||
type Mongo struct {
|
||||
@@ -26,8 +26,8 @@ func NewMongo(addr string, auth ...string) (*Mongo, error) {
|
||||
Password: auth[1],
|
||||
})
|
||||
}
|
||||
db, err := mongo.NewClientWithOptions(
|
||||
"mongodb://"+addr,
|
||||
db, err := mongo.NewClient(
|
||||
options.Client().ApplyURI("mongodb://"+addr),
|
||||
options.Client().SetReadConcern(readconcern.Local()),
|
||||
options.Client().SetReadPreference(readpref.PrimaryPreferred()),
|
||||
options.Client().SetWriteConcern(writeconcern.New(
|
||||
@@ -68,8 +68,8 @@ func (mg *Mongo) Get(key string, ns ...string) ([]byte, error) {
|
||||
return nil, ErrNotFound
|
||||
}
|
||||
|
||||
elem, err := cursor.DecodeBytes()
|
||||
if err != nil {
|
||||
var elem bson.Raw
|
||||
if err := cursor.Decode(&elem); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
raw, err := elem.LookupErr("value")
|
||||
|
||||
Reference in New Issue
Block a user