40 lines
1.2 KiB
Java
40 lines
1.2 KiB
Java
package org.baeldung;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.HashSet;
|
|
|
|
import javax.annotation.PostConstruct;
|
|
|
|
import org.baeldung.persistence.dao.RoleRepository;
|
|
import org.baeldung.persistence.dao.UserRepository;
|
|
import org.baeldung.persistence.model.Role;
|
|
import org.baeldung.persistence.model.User;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
@Component
|
|
public class Setup {
|
|
|
|
@Autowired
|
|
private UserRepository userRepository;
|
|
|
|
@Autowired
|
|
private RoleRepository roleRepository;
|
|
|
|
@PostConstruct
|
|
private void setupData() {
|
|
Role roleUser = new Role("ROLE_USER");
|
|
roleUser = roleRepository.save(roleUser);
|
|
Role roleAdmin = new Role("ROLE_ADMIN");
|
|
roleAdmin = roleRepository.save(roleAdmin);
|
|
|
|
final User userJohn = new User("john", "john@test.com");
|
|
userJohn.setRoles(new HashSet<Role>(Arrays.asList(roleUser, roleAdmin)));
|
|
userRepository.save(userJohn);
|
|
|
|
final User userTom = new User("tom", "tom@test.com");
|
|
userTom.setRoles(new HashSet<Role>(Arrays.asList(roleUser)));
|
|
userRepository.save(userTom);
|
|
}
|
|
|
|
} |