aswi2020vldc-gitlab/src/main/java/vldc/aswi/service/RoleManagerImpl.java @ d90fc9de

package vldc.aswi.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.event.EventListener;
import org.springframework.core.annotation.Order;
import org.springframework.dao.InvalidDataAccessResourceUsageException;
import org.springframework.stereotype.Service;
import vldc.aswi.dao.RoleRepository;
import vldc.aswi.domain.Role;

import javax.transaction.Transactional;
import java.util.LinkedList;
import java.util.List;

/**
 * Manager for Role.
 */
@Service
@Slf4j
public class RoleManagerImpl implements RoleManager {

    /** Autowired role repository. */
    @Autowired
    private RoleRepository roleRepository;

    /**
     * Initialization setup for role manager.
     * Check if table "Role" exists.
     */
    @EventListener(classes = {
            ContextRefreshedEvent.class
    })
    @Order(1)
    @Transactional
    public void setup() {
        try {
            if (this.roleRepository.count() == 0) {
                // Just checking if table exists.
            }
        }
        catch (InvalidDataAccessResourceUsageException e) {
            log.error("Table \"Role\" did not exists in database!");
            System.exit(1);
        }
    }

    /**
     * Add newly created role into database.
     * @param name - Name of role.
     */
    public void addRole(String name) {
        Role role = new Role(name);
        this.roleRepository.save(role);
    }

    /**
     * Get all roles from database.
     * @return List of roles.
     */
    @Override
    public List<Role> getRoles() {
        List<Role> retVal = new LinkedList<>();
        this.roleRepository.findAll().forEach(retVal::add);
        return retVal;
    }

    /**
     * Get role from database by name.
     * @param name name of the role
     * @return List of roles.
     */
    @Override
    public Role getRole(String name) {
        return roleRepository.getByName(name);
    }
}