package com.ygj.yuemum.controller.admin;


import com.alibaba.fastjson.JSONObject;
import com.ygj.yuemum.domain.admin.Train;
import com.ygj.yuemum.service.admin.TrainService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.Map;

@RestController
public class TrainController {

    @Autowired
    private TrainService trainService;

    @GetMapping("/getTrains")
    public String getTrains(@RequestParam("page") Integer page,@RequestParam("limit") Integer limit) {
        Map<String, Object> trains= trainService.getTrains(page,limit);
        String jso = JSONObject.toJSONString(trains);
        return jso;
    }

    @PostMapping("/addTrain")
    public int add(@ModelAttribute Train train) {
        return trainService.addTrain(train);
    }

    @GetMapping("/deleteTrain/{id}")
    public int delete(@PathVariable("id") Integer id) {
        return trainService.deleteTrain(id);
    }

    @PostMapping("/updateTrain")
    public int update(@ModelAttribute Train train) {
        return trainService.updateTrain(train);
    }

    @GetMapping("/getTrain/{id}")
    public Train getOne(@PathVariable("id") Integer id) {
        return trainService.getTrain(id);
    }
}