From 0104db48f5c22e0363eefa875e1aa9223c227056 Mon Sep 17 00:00:00 2001
From: ScarletRedMan <terentev.andrey.2002@gmail.com>
Date: Thu, 16 Nov 2023 17:33:00 +0700
Subject: [PATCH] Added methods hashCode() and equals() for models

---
 .../dragonestia/loadbalancer/model/Bucket.java | 15 +++++++++++++++
 .../dragonestia/loadbalancer/model/User.java   | 18 +++++++++++++++++-
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/src/main/java/ru/dragonestia/loadbalancer/model/Bucket.java b/src/main/java/ru/dragonestia/loadbalancer/model/Bucket.java
index 97b9b0b..2f00b41 100644
--- a/src/main/java/ru/dragonestia/loadbalancer/model/Bucket.java
+++ b/src/main/java/ru/dragonestia/loadbalancer/model/Bucket.java
@@ -28,4 +28,19 @@ public class Bucket {
         if (slots.isUnlimited()) return true;
         return slots.getSlots() >= usedSlots + requiredSlots;
     }
+
+    @Override
+    public int hashCode() {
+        return identifier.hashCode();
+    }
+
+    @Override
+    public boolean equals(Object object) {
+        if (object == this) return true;
+        if (object == null) return false;
+        if (object instanceof Bucket other) {
+            return identifier.equals(other.identifier);
+        }
+        return false;
+    }
 }
diff --git a/src/main/java/ru/dragonestia/loadbalancer/model/User.java b/src/main/java/ru/dragonestia/loadbalancer/model/User.java
index 3014a28..7ba7bef 100644
--- a/src/main/java/ru/dragonestia/loadbalancer/model/User.java
+++ b/src/main/java/ru/dragonestia/loadbalancer/model/User.java
@@ -2,4 +2,20 @@ package ru.dragonestia.loadbalancer.model;
 
 import lombok.NonNull;
 
-public record User(@NonNull String identifier) {}
+public record User(@NonNull String identifier) {
+
+    @Override
+    public int hashCode() {
+        return identifier.hashCode();
+    }
+
+    @Override
+    public boolean equals(Object object) {
+        if (object == this) return true;
+        if (object == null) return false;
+        if (object instanceof User other) {
+            return identifier.equals(other.identifier);
+        }
+        return false;
+    }
+}