Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package com.buildingmanagementapp;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.boot.ApplicationArguments;
- import org.springframework.boot.ApplicationRunner;
- import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
- import org.springframework.stereotype.Component;
- import com.buildingmanagementapp.models.Address;
- import com.buildingmanagementapp.models.Building;
- import com.buildingmanagementapp.models.Role;
- import com.buildingmanagementapp.models.User;
- import com.buildingmanagementapp.repositories.AddressRepo;
- import com.buildingmanagementapp.repositories.BuildingRepo;
- import com.buildingmanagementapp.repositories.RoleRepo;
- import com.buildingmanagementapp.repositories.UserRepo;
- @Component
- public class DataInitializer implements ApplicationRunner {
- @Autowired
- RoleRepo roleRepo;
- @Autowired
- UserRepo userRepo;
- @Autowired
- BuildingRepo buildingRepo;
- @Autowired
- AddressRepo addressRepo;
- @Autowired
- public DataInitializer(RoleRepo roleRepo) {
- this.roleRepo = roleRepo;
- }
- public void run(ApplicationArguments args) {
- try {
- BCryptPasswordEncoder encoder = new BCryptPasswordEncoder();
- User user = new User();
- Building building = new Building();
- Address address = new Address();
- String[] roles = { "ADMIN", "EMPLOYEE", "TRUMP" };
- for (int i = 0; i < roles.length; i++) {
- Role r = roleRepo.findByName(roles[i]);
- if (r == null) {
- Role role = new Role();
- role.setName(roles[i]);
- roleRepo.save(role);
- }
- }
- user.setActivated(true);
- user.setEmail("riggy.ruter@gmail.com");
- user.setLast_name("ruter");
- user.setName("riggy");
- user.setPassword(encoder.encode("123"));
- user.setUsername("riggy");
- user.getRoles().add(roleRepo.findByName("ADMIN"));
- userRepo.save(user);
- // Dodavanje Address
- address.setCity("Novi Sad");
- address.setZip("21000");
- address.setNumber(15);
- address.setStreet("Brace Krkljus");
- addressRepo.save(address);
- // Dodavanje zgrade
- building.setAddress(address);
- buildingRepo.save(building);
- /*
- * Another inactive user for testing
- */
- User bleki = new User();
- bleki.setActivated(false);
- bleki.setEmail("bleki@gmail.com");
- bleki.setLast_name("bleki");
- bleki.setName("bleki");
- bleki.setPassword(encoder.encode("123"));
- bleki.setUsername("bleki");
- userRepo.save(bleki);
- /*
- * Another active user for testing
- */
- User nn = new User();
- nn.setActivated(false);
- nn.setEmail("nn@gmail.com");
- nn.setLast_name("nn");
- nn.setName("nn");
- nn.setPassword(encoder.encode("1234"));
- nn.setUsername("nn");
- userRepo.save(nn);
- } catch (Exception e) {
- e.printStackTrace();
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement