61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"reflect"
|
|
"time"
|
|
|
|
"git.cliffbreak.de/haveachin/scoreboard/config"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"go.mongodb.org/mongo-driver/mongo/readpref"
|
|
)
|
|
|
|
type DataService struct {
|
|
client *mongo.Client
|
|
db *mongo.Database
|
|
collections map[reflect.Type]*mongo.Collection
|
|
}
|
|
|
|
func New(ds config.DataService) (*DataService, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
|
defer cancel()
|
|
|
|
connectionString := fmt.Sprintf("mongodb://%s:%s@%s:%d/?authSource=admin", url.QueryEscape(ds.Username), url.QueryEscape(ds.Password), ds.IP, ds.Port)
|
|
client, err := mongo.Connect(ctx, options.Client().ApplyURI(connectionString))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err = client.Ping(ctx, readpref.Primary()); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &DataService{
|
|
client: client,
|
|
db: client.Database(ds.Database),
|
|
}, nil
|
|
}
|
|
|
|
func (s DataService) collection(t reflect.Type) *mongo.Collection {
|
|
for t.Kind() != reflect.Struct {
|
|
t = t.Elem()
|
|
}
|
|
|
|
collection, ok := s.collections[t]
|
|
if !ok {
|
|
panic(fmt.Sprintf("No collection registered for %T", t))
|
|
}
|
|
|
|
return collection
|
|
}
|
|
|
|
func (s DataService) RegisterCollection(t reflect.Type, collection string) {
|
|
for t.Kind() != reflect.Struct {
|
|
t = t.Elem()
|
|
}
|
|
|
|
s.collections[t] = s.db.Collection(collection)
|
|
}
|