src/main/java/ru/sber/arch/restExample/controller/ReviewController.java - master - Vagabond132/RestExample

src/main/java/ru/sber/arch/restExample/controller/ReviewController.java
69 строк3 KB
Vagabond132
Добавление DTO
22 фев 2025, 16:40
22 фев 2025, 16:40e8183e1
package ru.sber.arch.restExample.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import ru.sber.arch.restExample.dto.Review;
import ru.sber.arch.restExample.model.Movie;
import ru.sber.arch.restExample.model.Rate;
import ru.sber.arch.restExample.model.User;
import ru.sber.arch.restExample.service.MovieService;
import ru.sber.arch.restExample.service.UserService;

@RestController("/review")
public class ReviewController {

    @Autowired
    UserService userService;
    @Autowired
    MovieService movieService;


    @PostMapping("/add")
    public ResponseEntity<?> AddRate(Review review){
        User tempUser;
        var userOptional = userService.getUserByLogin(review.getUser_login());
        if (userOptional.isEmpty())
            return new ResponseEntity<>(HttpStatus.OK);
        else
            tempUser = userOptional.get();
        tempUser.addRate(new Rate(review.getMovie().getName(), review.getMovie().getDate(), review.getReviewText(), review.getRate()));
        var movieOptional = movieService.getMovie(review.getMovie().getName(), review.getMovie().getDate());

        Movie tempMovie;
        if (movieOptional.isEmpty())
            return new ResponseEntity<>(HttpStatus.OK);
        else
            tempMovie = movieOptional.get();
        tempMovie.AddRate(review.getRate());
        userService.updateUser(tempUser);
        movieService.updateMovie(tempMovie);
        return new ResponseEntity<>(HttpStatus.OK);
    }

    @PostMapping("/delete")
    public ResponseEntity<?> DeleteRate(Review review){
        User tempUser;
        var userOptional = userService.getUserByLogin(review.getUser_login());
        if (userOptional.isEmpty())
            return new ResponseEntity<>(HttpStatus.OK);
        else
            tempUser = userOptional.get();
        tempUser.removeRate(new Rate(review.getMovie().getName(), review.getMovie().getDate(), review.getReviewText(), review.getRate()));
        var movieOptional = movieService.getMovie(review.getMovie().getName(), review.getMovie().getDate());
        Movie tempMovie;
        if (movieOptional.isEmpty())
            return new ResponseEntity<>(HttpStatus.OK);
        else
            tempMovie = movieOptional.get();
        tempMovie.DeleteRate(review.getRate());
        userService.updateUser(tempUser);
        movieService.updateMovie(tempMovie);
        return new ResponseEntity<>(HttpStatus.OK);
    }



}