summaryrefslogtreecommitdiff
path: root/database
diff options
context:
space:
mode:
Diffstat (limited to 'database')
-rw-r--r--database/database.go66
1 files changed, 66 insertions, 0 deletions
diff --git a/database/database.go b/database/database.go
new file mode 100644
index 0000000..a970ee1
--- /dev/null
+++ b/database/database.go
@@ -0,0 +1,66 @@
+/* OpenBills-server - Server for libre billing software OpenBills-web
+ * Copyright (C) 2022 Vidhu Kant Sharma <vidhukant@vidhukant.xyz>
+
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package database
+
+import (
+ "context"
+ "log"
+ "time"
+
+ "go.mongodb.org/mongo-driver/mongo"
+ "go.mongodb.org/mongo-driver/mongo/options"
+)
+
+/* This creates a new client and sets
+ * it to the value of MongoClient which
+ * can be imported by other packages
+ *
+ * I am not at all sure if this is the best
+ * actually no, even if this is a decent way to do it.
+ * But yea this seems to work
+ * (remember to close the connections!)
+ */
+var DB *mongo.Database
+var client *mongo.Client
+
+func init() {
+ ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+ defer cancel()
+
+ var err error
+ client, err = mongo.Connect(ctx, options.Client().ApplyURI("mongodb://127.0.0.1:27017"))
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ log.Println("Successfully connected to MongoDB database.")
+ DB = client.Database("OpenBillsDB")
+}
+
+func DisconnectDB() {
+ if client == nil {
+ return
+ }
+
+ err := client.Disconnect(context.TODO())
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ log.Println("Successfully closed connection with MongoDB.")
+}