aswi2020vldc-gitlab/src/main/java/vldc/aswi/service/LocationManagerImpl.java @ 55515be9

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.LocationRepository;
import vldc.aswi.domain.Location;

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

/**
 * Manager for Location.
 */
@Service
@Slf4j
public class LocationManagerImpl implements LocationManager {

    /** Autowired location repository. */
    @Autowired
    private LocationRepository locationRepository;

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

    /**
     * Get all locations from database.
     * @return List of locations.
     */
    @Override
    public List<Location> getLocations() {
        List<Location> retVal = new LinkedList<>();
        this.locationRepository.findAll().forEach(retVal::add);
        return retVal;
    }

    /**
     * Get location from database by ID.
     * @return location by ID.
     */
    @Override
    public Location getLocationById(long id) {
        return locationRepository.getById(id);
    }
}