40 lines
878 B
Go
40 lines
878 B
Go
package databases
|
|
|
|
import (
|
|
"log"
|
|
|
|
"git.wh64.net/muffin/goMuffin/configs"
|
|
"go.mongodb.org/mongo-driver/v2/mongo"
|
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
|
)
|
|
|
|
type MuffinDatabase struct {
|
|
Client *mongo.Client
|
|
Learns *mongo.Collection
|
|
Texts *mongo.Collection
|
|
Memory *mongo.Collection
|
|
}
|
|
|
|
var Database *MuffinDatabase
|
|
|
|
func init() {
|
|
var err error
|
|
|
|
Database, err = Connect()
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}
|
|
|
|
func Connect() (*MuffinDatabase, error) {
|
|
client, err := mongo.Connect(options.Client().ApplyURI(configs.Config.Database.URL))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &MuffinDatabase{
|
|
Client: client,
|
|
Learns: client.Database(configs.Config.Database.Name).Collection("learn"),
|
|
Texts: client.Database(configs.Config.Database.Name).Collection("text"),
|
|
Memory: client.Database(configs.Config.Database.Name).Collection("memory"),
|
|
}, nil
|
|
}
|