aswi2020vldc-gitlab/src/main/java/vldc/aswi/web/controller/AssemblyController.java @ 2868bb9a

package vldc.aswi.web.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.servlet.ModelAndView;
import vldc.aswi.domain.Assembly;
import vldc.aswi.service.AssemblyManager;
import vldc.aswi.service.SqlQueryManager;

import javax.validation.Valid;

@Controller
public class AssemblyController {

    @Autowired
    private SqlQueryManager sqlQueryManager;

    @Autowired
    private AssemblyManager assemblyManager;

    @GetMapping("/assembly")
    public ModelAndView assemblyIndex(@Valid @ModelAttribute("assemblyID") String id) {
        ModelAndView modelAndView = new ModelAndView("assembly");

        ModelMap modelMap = modelAndView.getModelMap();

        Assembly selectedAssembly = assemblyManager.getAssemblyById(Long.parseLong(id));

        modelMap.addAttribute("assemblies", assemblyManager.getAssemblies());
        modelMap.addAttribute("selectedAssembly", selectedAssembly);
        modelMap.addAttribute("contingencyTableRows", sqlQueryManager.getContingencyTableRow(selectedAssembly.getSQLQuery()));

        return modelAndView;
    }


    @PostMapping("/assembly")
    public ModelAndView indexPost(@Valid @ModelAttribute("assemblyID") String id) {
        ModelAndView modelAndView = new ModelAndView();

        ModelMap modelMap = modelAndView.getModelMap();

        Assembly selectedAssembly = assemblyManager.getAssemblyById(Long.parseLong(id));

        modelMap.addAttribute("assemblies", assemblyManager.getAssemblies());
        modelMap.addAttribute("selectedAssembly", selectedAssembly);
        modelMap.addAttribute("contingencyTableRows", sqlQueryManager.getContingencyTableRow(selectedAssembly.getSQLQuery()));

        return modelAndView;
    }
}