src/main/java/com/mavlushechka/animarfo/database/Database.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
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 MongoDatabase MONGO_DATABASE = new MongoClient(new MongoClientURI(App.PROPERTIES.getProperty("DATABASE_URL")))
            .getDatabase(App.PROPERTIES.getProperty("DATABASE_NAME"))
            .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 = MONGO_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");
        }
    }
}