Change path

1 files changed, 47 insertions, 0 deletions

diff --git a/src/main/java/com/mavlushechka/animarfo/database/Database.java b/src/main/java/com/mavlushechka/animarfo/database/Database.java
new file mode 100644
index 0000000..5de20a1
--- /dev/null
+++ b/src/main/java/com/mavlushechka/animarfo/database/Database.java
@@ -0,0 +1,47 @@
+package com.mavlushechka.animarfo.database;
+
+import com.mavlushechka.animarfo.App;
+import com.mavlushechka.animarfo.telegram.user.TelegramUser;
+import com.mongodb.MongoClient;
+import com.mongodb.MongoClientSettings;
+import com.mongodb.MongoClientURI;
+import com.mongodb.client.MongoCollection;
+import com.mongodb.client.MongoDatabase;
+import com.mongodb.client.model.Filters;
+import org.bson.Document;
+import org.bson.codecs.configuration.CodecRegistries;
+import org.bson.codecs.pojo.PojoCodecProvider;
+
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+public class Database {
+    private static final String MONGO_URI = App.PROPERTIES.getProperty("DATABASE_URL");
+    private static final MongoDatabase database = new MongoClient(new MongoClientURI(MONGO_URI))
+            .getDatabase("animarfo")
+            .withCodecRegistry(
+                    CodecRegistries.fromRegistries(
+                        MongoClientSettings.getDefaultCodecRegistry(),
+                        CodecRegistries.fromProviders(PojoCodecProvider.builder().automatic(true).build())
+                    )
+            );
+    private static final Logger LOGGER = Logger.getLogger(Database.class.getName());
+    static {
+        LOGGER.setLevel(Level.ALL);
+    }
+
+    public static void saveUser(TelegramUser telegramUser) {
+        LOGGER.info("Trying to save user to database");
+        MongoCollection<TelegramUser> collection = database.getCollection("users", TelegramUser.class);
+        LOGGER.fine("Connected to database and got collection");
+        TelegramUser searchedTelegramUser = collection.find(Filters.eq("_id", telegramUser.getId())).first();
+
+        if (searchedTelegramUser == null) {
+            collection.insertOne(telegramUser);
+            LOGGER.fine("User is added to the database");
+        } else {
+            collection.updateOne(Filters.eq("_id", telegramUser.getId()), new Document("$set", telegramUser));
+            LOGGER.fine("User is updated");
+        }
+    }
+}