Skip to content

Commit ddfe85a

Browse files
authored
Refactor DatabaseSeeder to include repositories for user, address, and refresh tokens; clear user roles and delete all users during seeding (#31)
1 parent ab4c9f6 commit ddfe85a

1 file changed

Lines changed: 20 additions & 1 deletion

File tree

src/main/java/com/podzilla/auth/seeder/DatabaseSeeder.java

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,22 +2,41 @@
22

33
import com.podzilla.auth.model.ERole;
44
import com.podzilla.auth.model.Role;
5+
import com.podzilla.auth.repository.AddressRepository;
6+
import com.podzilla.auth.repository.RefreshTokenRepository;
57
import com.podzilla.auth.repository.RoleRepository;
8+
import com.podzilla.auth.repository.UserRepository;
69
import org.springframework.boot.CommandLineRunner;
710
import org.springframework.stereotype.Component;
811

912
@Component
1013
public class DatabaseSeeder implements CommandLineRunner {
1114

1215
private final RoleRepository roleRepository;
16+
private final RefreshTokenRepository refreshTokenRepository;
17+
private final UserRepository userRepository;
18+
private final AddressRepository addressRepository;
1319

14-
public DatabaseSeeder(final RoleRepository roleRepository) {
20+
public DatabaseSeeder(final RoleRepository roleRepository,
21+
final RefreshTokenRepository refreshTokenRepository,
22+
final UserRepository userRepository,
23+
final AddressRepository addressRepository) {
1524
this.roleRepository = roleRepository;
25+
this.refreshTokenRepository = refreshTokenRepository;
26+
this.userRepository = userRepository;
27+
this.addressRepository = addressRepository;
1628
}
1729

1830
@Override
1931
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();
2037
roleRepository.deleteAll();
38+
refreshTokenRepository.deleteAll();
39+
addressRepository.deleteAll();
2140
roleRepository.save(new Role(ERole.ROLE_USER));
2241
roleRepository.save(new Role(ERole.ROLE_ADMIN));
2342
roleRepository.save(new Role(ERole.ROLE_COURIER));

0 commit comments

Comments
 (0)