source: bus-n-go-do-5ta-faza/src/main/java/mk/ukim/finki/busngo/service/impl/KaznaServiceImpl.java

package mk.ukim.finki.busngo.service.impl;

import mk.ukim.finki.busngo.model.entities.Kazna;
import mk.ukim.finki.busngo.model.entities.Kontroli;
import mk.ukim.finki.busngo.repository.KaznaRepository;
import mk.ukim.finki.busngo.service.KaznaService;
import mk.ukim.finki.busngo.service.KontrolaService;
import org.springframework.stereotype.Service;

import java.sql.Timestamp;
import java.time.LocalDateTime;
import java.util.List;

@Service
public class KaznaServiceImpl implements KaznaService {
    private final KaznaRepository kaznaRepository;
    private final KontrolaService kontrolaService;

    public KaznaServiceImpl(KaznaRepository kaznaRepository, KontrolaService kontrolaService) {
        this.kaznaRepository = kaznaRepository;
        this.kontrolaService = kontrolaService;
    }

    @Override
    public List<Kazna> findAll() {
        return kaznaRepository.findAll();
    }

    @Override
    public List<Kazna> findAllByKontrolaId(Long kontrolaId) {
        Kontroli kontroli = kontrolaService.findById(kontrolaId);
        return kaznaRepository.findAllByKontroliByKontrolaId(kontroli);
    }

    @Override
    public Kazna pay(Long kaznaId) {
        Kazna kazna = this.findById(kaznaId);
        kazna.setKzPlateno(true);
        kazna.setKzDatumPlateno(Timestamp.valueOf(LocalDateTime.now()));
        return kaznaRepository.save(kazna);
    }

    @Override
    public Kazna findById(Long id) {
        return kaznaRepository.findById(id).orElseThrow(RuntimeException::new);
    }
}