Advertisement
Guest User

Untitled

a guest
Dec 12th, 2017
536
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 2.77 KB | None | 0 0
  1. package com.buildingmanagementapp;
  2.  
  3. import org.springframework.beans.factory.annotation.Autowired;
  4. import org.springframework.boot.ApplicationArguments;
  5. import org.springframework.boot.ApplicationRunner;
  6. import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
  7. import org.springframework.stereotype.Component;
  8.  
  9. import com.buildingmanagementapp.models.Address;
  10. import com.buildingmanagementapp.models.Building;
  11. import com.buildingmanagementapp.models.Role;
  12. import com.buildingmanagementapp.models.User;
  13. import com.buildingmanagementapp.repositories.AddressRepo;
  14. import com.buildingmanagementapp.repositories.BuildingRepo;
  15. import com.buildingmanagementapp.repositories.RoleRepo;
  16. import com.buildingmanagementapp.repositories.UserRepo;
  17.  
  18. @Component
  19. public class DataInitializer implements ApplicationRunner {
  20.  
  21.     @Autowired
  22.     RoleRepo roleRepo;
  23.  
  24.     @Autowired
  25.     UserRepo userRepo;
  26.  
  27.     @Autowired
  28.     BuildingRepo buildingRepo;
  29.  
  30.     @Autowired
  31.     AddressRepo addressRepo;
  32.  
  33.     @Autowired
  34.     public DataInitializer(RoleRepo roleRepo) {
  35.         this.roleRepo = roleRepo;
  36.     }
  37.  
  38.     public void run(ApplicationArguments args) {
  39.         try {
  40.             BCryptPasswordEncoder encoder = new BCryptPasswordEncoder();
  41.  
  42.             User user = new User();
  43.             Building building = new Building();
  44.             Address address = new Address();
  45.  
  46.             String[] roles = { "ADMIN", "EMPLOYEE", "TRUMP" };
  47.             for (int i = 0; i < roles.length; i++) {
  48.                 Role r = roleRepo.findByName(roles[i]);
  49.                 if (r == null) {
  50.                     Role role = new Role();
  51.                     role.setName(roles[i]);
  52.                     roleRepo.save(role);
  53.                 }
  54.             }
  55.  
  56.             user.setActivated(true);
  57.             user.setEmail("riggy.ruter@gmail.com");
  58.             user.setLast_name("ruter");
  59.             user.setName("riggy");
  60.             user.setPassword(encoder.encode("123"));
  61.             user.setUsername("riggy");
  62.             user.getRoles().add(roleRepo.findByName("ADMIN"));
  63.             userRepo.save(user);
  64.  
  65.             // Dodavanje Address
  66.             address.setCity("Novi Sad");
  67.             address.setZip("21000");
  68.             address.setNumber(15);
  69.             address.setStreet("Brace Krkljus");
  70.             addressRepo.save(address);
  71.  
  72.             // Dodavanje zgrade
  73.             building.setAddress(address);
  74.             buildingRepo.save(building);
  75.            
  76.             /*
  77.              * Another inactive user for testing
  78.             */
  79.             User bleki = new User();
  80.             bleki.setActivated(false);
  81.             bleki.setEmail("bleki@gmail.com");
  82.             bleki.setLast_name("bleki");
  83.             bleki.setName("bleki");
  84.             bleki.setPassword(encoder.encode("123"));
  85.             bleki.setUsername("bleki");
  86.             userRepo.save(bleki);
  87.            
  88.             /*
  89.              * Another active user for testing
  90.             */
  91.             User nn = new User();
  92.             nn.setActivated(false);
  93.             nn.setEmail("nn@gmail.com");
  94.             nn.setLast_name("nn");
  95.             nn.setName("nn");
  96.             nn.setPassword(encoder.encode("1234"));
  97.             nn.setUsername("nn");
  98.             userRepo.save(nn);
  99.         } catch (Exception e) {
  100.             e.printStackTrace();
  101.         }
  102.     }
  103. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement