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

package com.project.beautycenter.service.impl;

import com.project.beautycenter.model.Beautycenter;
import com.project.beautycenter.model.Users;
import com.project.beautycenter.model.Vraboteni;
import com.project.beautycenter.model.exceptions.InvalidUserIdException;
import com.project.beautycenter.repository.BeautyCenterRepository;
import com.project.beautycenter.repository.UsersRepository;
import com.project.beautycenter.repository.VraboteniRepository;
import com.project.beautycenter.service.VraboteniService;
import org.springframework.stereotype.Service;

import java.time.Instant;
import java.util.List;

@Service
public class VraboteniServiceImpl implements VraboteniService {
    private final VraboteniRepository vraboteniRepository;
    private final UsersRepository usersRepository;
    private final BeautyCenterRepository beautyCenterRepository;

    public VraboteniServiceImpl(VraboteniRepository vraboteniRepository, UsersRepository usersRepository, BeautyCenterRepository beautyCenterRepository) {
        this.vraboteniRepository = vraboteniRepository;
        this.usersRepository = usersRepository;
        this.beautyCenterRepository = beautyCenterRepository;
    }

    @Override
    public List<Vraboteni> findAll() {
        return this.vraboteniRepository.findAll();
    }

    @Override
    public Vraboteni findbyId(Integer id) {
        return this.vraboteniRepository.findById(id).orElseThrow(NullPointerException::new);
    }

    @Override
    public Vraboteni findVrabotenByUserId(Integer id) {
        return this.vraboteniRepository.getById(id);
    }

    @Override
    public Vraboteni create(Users users, String ime, String middleName, String prezime, String telBr, String email,
                            Integer rabIskustvo, Instant rabotiOd, String beautyCenterId) {

        Users user = this.usersRepository.findById(users.getId()).orElseThrow(InvalidUserIdException::new);
        Beautycenter beautycenter = this.beautyCenterRepository.getById(beautyCenterId);
        Vraboteni vraboteni = new Vraboteni(user, ime, middleName, prezime, telBr, email, rabIskustvo, rabotiOd, beautycenter);
        return this.vraboteniRepository.save(vraboteni);
    }
}