From c1a35325d632afe173c9403721c59335b33f8e76 Mon Sep 17 00:00:00 2001 From: gfxv Date: Wed, 12 Mar 2025 18:58:23 +0300 Subject: [PATCH] feat: add role initializer --- .../java/dev/gfxv/blps/BlpsApplication.java | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/src/main/java/dev/gfxv/blps/BlpsApplication.java b/src/main/java/dev/gfxv/blps/BlpsApplication.java index da49477..c43c447 100644 --- a/src/main/java/dev/gfxv/blps/BlpsApplication.java +++ b/src/main/java/dev/gfxv/blps/BlpsApplication.java @@ -1,7 +1,11 @@ package dev.gfxv.blps; +import dev.gfxv.blps.entity.Role; +import dev.gfxv.blps.repository.RoleRepository; +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 BlpsApplication { @@ -10,4 +14,20 @@ public class BlpsApplication { SpringApplication.run(BlpsApplication.class, args); } + + @Bean + CommandLineRunner initRoles(RoleRepository roleRepository) { + return args -> { + if (roleRepository.findByName("ROLE_USER").isEmpty()) { + Role userRole = new Role(); + userRole.setName("ROLE_USER"); + roleRepository.save(userRole); + } + if (roleRepository.findByName("ROLE_MODERATOR").isEmpty()) { + Role modRole = new Role(); + modRole.setName("ROLE_MODERATOR"); + roleRepository.save(modRole); + } + }; + } }