|
2 | 2 |
|
3 | 3 | import com.podzilla.auth.model.ERole; |
4 | 4 | import com.podzilla.auth.model.Role; |
| 5 | +import com.podzilla.auth.repository.AddressRepository; |
| 6 | +import com.podzilla.auth.repository.RefreshTokenRepository; |
5 | 7 | import com.podzilla.auth.repository.RoleRepository; |
| 8 | +import com.podzilla.auth.repository.UserRepository; |
6 | 9 | import org.springframework.boot.CommandLineRunner; |
7 | 10 | import org.springframework.stereotype.Component; |
8 | 11 |
|
9 | 12 | @Component |
10 | 13 | public class DatabaseSeeder implements CommandLineRunner { |
11 | 14 |
|
12 | 15 | private final RoleRepository roleRepository; |
| 16 | + private final RefreshTokenRepository refreshTokenRepository; |
| 17 | + private final UserRepository userRepository; |
| 18 | + private final AddressRepository addressRepository; |
13 | 19 |
|
14 | | - public DatabaseSeeder(final RoleRepository roleRepository) { |
| 20 | + public DatabaseSeeder(final RoleRepository roleRepository, |
| 21 | + final RefreshTokenRepository refreshTokenRepository, |
| 22 | + final UserRepository userRepository, |
| 23 | + final AddressRepository addressRepository) { |
15 | 24 | this.roleRepository = roleRepository; |
| 25 | + this.refreshTokenRepository = refreshTokenRepository; |
| 26 | + this.userRepository = userRepository; |
| 27 | + this.addressRepository = addressRepository; |
16 | 28 | } |
17 | 29 |
|
18 | 30 | @Override |
19 | 31 | public void run(final String... args) throws Exception { |
| 32 | + userRepository.findAll().forEach(user -> { |
| 33 | + user.getRoles().clear(); |
| 34 | + userRepository.save(user); |
| 35 | + }); |
| 36 | + userRepository.deleteAll(); |
20 | 37 | roleRepository.deleteAll(); |
| 38 | + refreshTokenRepository.deleteAll(); |
| 39 | + addressRepository.deleteAll(); |
21 | 40 | roleRepository.save(new Role(ERole.ROLE_USER)); |
22 | 41 | roleRepository.save(new Role(ERole.ROLE_ADMIN)); |
23 | 42 | roleRepository.save(new Role(ERole.ROLE_COURIER)); |
|
0 commit comments