1 files changed, 46 insertions, 0 deletions
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