From 0225bdb772d1334cc1aa7ab0fc3678df0864df6b Mon Sep 17 00:00:00 2001
From: AlisaLinUwU <alisalinuwu@gmail.com>
Date: Sun, 26 Jan 2025 10:42:28 +0500
Subject: Initialize

---
 .../com/mavlushechka/barbershop/Application.java   | 46 ++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 src/main/java/com/mavlushechka/barbershop/Application.java

(limited to 'src/main/java/com/mavlushechka/barbershop/Application.java')

diff --git a/src/main/java/com/mavlushechka/barbershop/Application.java b/src/main/java/com/mavlushechka/barbershop/Application.java
new file mode 100644
index 0000000..00b8909
--- /dev/null
+++ b/src/main/java/com/mavlushechka/barbershop/Application.java
@@ -0,0 +1,46 @@
+package com.mavlushechka.barbershop;
+
+import com.mavlushechka.barbershop.domain.Level;
+import com.mavlushechka.barbershop.domain.Role;
+import com.mavlushechka.barbershop.repository.LevelRepository;
+import com.mavlushechka.barbershop.repository.RoleRepository;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Bean;
+
+@SpringBootApplication
+public class Application {
+    @Autowired
+    private RoleRepository roleRepository;
+    @Autowired
+    private LevelRepository levelRepository;
+
+    public static void main(String[] args) {
+        SpringApplication.run(Application.class, args);
+    }
+
+    @Bean
+    CommandLineRunner init() {
+        return args -> {
+            save(new Role("Admin"));
+            save(new Role("User"));
+            save(new Level("Junior"));
+            save(new Level("Middle"));
+            save(new Level("Senior"));
+        };
+    }
+
+    private void save(Role role) {
+        if (roleRepository.findByName(role.getName()) == null) {
+            roleRepository.save(role);
+        }
+    }
+
+    private void save(Level level) {
+        if (levelRepository.findByName(level.getName()) == null) {
+            levelRepository.save(level);
+        }
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3