source: src/main/java/com/project/beautycenter/service/impl/RezervacijaUslugiServiceImpl.java

package com.project.beautycenter.service.impl;

import com.project.beautycenter.model.Rezervacija;
import com.project.beautycenter.model.RezervacijaUslugi;
import com.project.beautycenter.model.RezervacijaUslugiId;
import com.project.beautycenter.model.Uslugi;
import com.project.beautycenter.repository.RezervacijaUslugiRepository;
import com.project.beautycenter.service.RezervacijaUslugiService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

@Service
public class RezervacijaUslugiServiceImpl implements RezervacijaUslugiService {
    private final RezervacijaUslugiRepository repository;

    public RezervacijaUslugiServiceImpl(RezervacijaUslugiRepository repository) {
        this.repository = repository;
    }

    @Override
    public List<RezervacijaUslugi> listAll() {
        return this.repository.findAll();
    }

    @Override
    public List<RezervacijaUslugi> listByRezervacija(Rezervacija r) {
        return this.repository.findAllByBrRez(r);
    }

    @Override
    public List<RezervacijaUslugi> create(Rezervacija rezervacija, List<Uslugi> uslugi) {
        List<RezervacijaUslugi> lista = new ArrayList<>();
        for (Uslugi u : uslugi) {
            RezervacijaUslugi ru = new RezervacijaUslugi
                    (new RezervacijaUslugiId(rezervacija.getId(), u.getId()), rezervacija, u);
            lista.add(ru);
            this.repository.save(ru);
        }
        return lista;
    }

    @Override
    public List<Uslugi> listAllByBrRez(Rezervacija rezervacija) {
        List<RezervacijaUslugi> rezUslugi = this.repository.findAllByBrRez(rezervacija);
        List<Uslugi> listaOutput = new ArrayList<>();
        for (var i = 0; i < rezUslugi.size(); i++) {
            listaOutput.add(rezUslugi.get(i).getUsluga());
        }

        return listaOutput;
    }

}