login 💀

Common/Database/AutoIncrement.cs

@@ -0,0 +1,28 @@
+using Common.Resources.Proto;
+using MongoDB.Bson.Serialization.Attributes;
+using MongoDB.Bson;
+using MongoDB.Driver;
+
+namespace Common.Database
+{
+    public class AutoIncrement
+    {
+        public static readonly IMongoCollection<AI> collection = Global.db.GetCollection<AI>("AutoIncrements");
+
+        public static int GetNextNumber(string name, int starting = 100)
+        {
+            AI AutoIncrement = collection.FindOneAndUpdate(Builders<AI>.Filter.Eq("Name", name), Builders<AI>.Update.SetOnInsert("Count", starting), new FindOneAndUpdateOptions<AI> { IsUpsert = true, ReturnDocument = ReturnDocument.After });
+            collection.UpdateOne(Builders<AI>.Filter.Eq("Name", AutoIncrement.Name), Builders<AI>.Update.Inc("Count", 1));
+            return AutoIncrement.Count + 1;
+        }
+
+#pragma warning disable CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider declaring as nullable.
+        public class AI
+        {
+            public ObjectId Id { get; set; }
+            public string Name { get; set; }
+            public int Count { get; set; }
+        }
+#pragma warning restore CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider declaring as nullable.
+    }
+}