package com.mavlonerkinboev.animarfo.database; import com.mavlonerkinboev.animarfo.App; import com.mavlonerkinboev.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 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"); } } }