From 3d93e3f7f845ceca4cea62302cc238e21bcd5429 Mon Sep 17 00:00:00 2001 From: mavlonerkinboev Date: Tue, 27 Jul 2021 09:15:57 +0500 Subject: eighteenth commit --- src/main/java/files/Database.java | 41 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100644 src/main/java/files/Database.java (limited to 'src/main/java/files/Database.java') diff --git a/src/main/java/files/Database.java b/src/main/java/files/Database.java new file mode 100644 index 0000000..2d35c2d --- /dev/null +++ b/src/main/java/files/Database.java @@ -0,0 +1,41 @@ +package files; + +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 static org.fusesource.jansi.Ansi.*; +import static org.fusesource.jansi.Ansi.Color.*; + +public class Database { + private static final String MONGO_URI = Bot.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())) + ); + + public static void saveUser(TelegramUser telegramUser) { + System.out.println(ansi().fg(YELLOW).a("Trying to save user to database...")); + MongoCollection collection = database.getCollection("users", TelegramUser.class); + System.out.println(ansi().fg(GREEN).a("Connected to database and got collection!")); + TelegramUser searchedTelegramUser = collection.find(Filters.eq("_id", telegramUser.getId())).first(); + + if (searchedTelegramUser == null) { + collection.insertOne(telegramUser); + System.out.println(ansi().fg(GREEN).a("User is added to the database!")); + } else { + collection.updateOne(Filters.eq("_id", telegramUser.getId()), new Document("$set", telegramUser)); + System.out.println(ansi().fg(GREEN).a("User is updated")); + } + } +} -- cgit v1.2.3