EntryServiceImpl.java

package com.exadel.service.impl;

import com.exadel.exception.EntryNotFoundException;
import com.exadel.model.entity.training.Entry;
import com.exadel.repository.EntryRepository;
import com.exadel.service.EntryService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.Date;
import java.util.List;
import java.util.Optional;

@Service
@Transactional(rollbackFor = {EntryNotFoundException.class})
public class EntryServiceImpl implements EntryService {
    @Autowired
    private EntryRepository entryRepository;

    @Override
    public List<Entry> getAllEntries() {
        return entryRepository.findAll();
    }

    @Override
    public Entry getEntryById(String id) {
        try {
            long entryId = Long.parseLong(id);
            return getEntryById(entryId);
        } catch (NumberFormatException ex) {
            throw new EntryNotFoundException(id);
        }
    }

    @Override
    public Entry getEntryById(long id) {
        Entry entry = entryRepository.findOne(id);
        if (entry != null) {
            return entry;
        } else {
            throw new EntryNotFoundException(String.valueOf(id));
        }
    }

    @Override
    public List<Entry> getAllEntriesByTrainingId(long trainingId) {
        return entryRepository.findByTrainingId(trainingId);
    }

    @Override
    public List<Entry> findFutureEntriesByTrainingId(Date time, long trainingId) {
        return entryRepository.findByTrainingIdAndBeginTimeAfter(trainingId, time);
    }

    @Override
    public List<Entry> findEntriesForJournal(Date beginDay, Date endDay, long trainingId) {
        return entryRepository.findByTrainingIdAndEndTimeAfterAndBeginTimeBefore(trainingId,
                beginDay, endDay);
    }

    @Override
    public List<Entry> findEntriesForJournal(Date endDay, long trainingId) {
        return entryRepository.findByTrainingIdAndBeginTimeBefore(trainingId, endDay);
    }

    @Override
    public Entry findNextEntryByTrainingId(Date time, long trainingId) {
        return entryRepository.findFirstByTrainingIdAndBeginTimeAfter(trainingId, time);
    }

    @Override
    public Optional<Entry> addEntry(Entry entry) {
        return Optional.ofNullable(entryRepository.save(entry));
    }

    @Override
    public void deleteEntry(long id) {
        entryRepository.delete(id);
    }

    @Override
    public List<Entry> getBetweenDates(Date before, Date after) {
        return entryRepository.findBetweenDates(before, after);
    }
}