93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"strconv"
|
|
"truck-company/internal/usecase"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
type TruckCompanyHandler struct {
|
|
uc *usecase.UseCase
|
|
}
|
|
|
|
func NewTruckCompanyHandler(u *usecase.UseCase) *TruckCompanyHandler {
|
|
return &TruckCompanyHandler{
|
|
uc: u,
|
|
}
|
|
}
|
|
|
|
func (h *TruckCompanyHandler) RegisterRoutes(r *mux.Router) {
|
|
|
|
r.HandleFunc("/routes", h.GetRoutes).Methods("GET")
|
|
r.HandleFunc("/routes/{route_id}/transport", h.GetTransportByRoute).Methods("GET")
|
|
r.HandleFunc("/shipment/{cargo_id}/cost", h.GetShipmentCost).Methods("GET")
|
|
// r.HandleFunc("/routes/{route_id}/transport-types", h.GetTransportTypesByRoute).Methods("GET")
|
|
// r.HandleFunc("/routes/{route_id}/cargo-turnover", h.GetCargoTurnoverByRoute).Methods("GET")
|
|
|
|
//404 для всех остальных
|
|
r.HandleFunc("/", http.NotFound)
|
|
}
|
|
|
|
func (h *TruckCompanyHandler) GetRoutes(w http.ResponseWriter, r *http.Request) {
|
|
|
|
routes, err := h.uc.GetAllRoutes()
|
|
if err != nil {
|
|
sendErrorResponse(w, http.StatusInternalServerError, "", err.Error())
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(routes)
|
|
}
|
|
|
|
func (h *TruckCompanyHandler) GetTransportByRoute(w http.ResponseWriter, r *http.Request) {
|
|
|
|
vars := mux.Vars(r)
|
|
routeID, err := strconv.ParseInt(vars["route_id"], 10, 64)
|
|
|
|
if err != nil {
|
|
sendErrorResponse(w, http.StatusBadRequest, "Invalid route ID", err.Error())
|
|
|
|
return
|
|
}
|
|
|
|
transport, err := h.uc.GetTransportByRoute(routeID)
|
|
if err != nil {
|
|
|
|
sendErrorResponse(w, http.StatusInternalServerError, "", err.Error())
|
|
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(transport)
|
|
}
|
|
func (h *TruckCompanyHandler) GetShipmentCost(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
cargo_id, err := strconv.ParseInt(vars["cargo_id"], 10, 64)
|
|
|
|
if err != nil {
|
|
sendErrorResponse(w, http.StatusBadRequest, "Invalid cargo ID", err.Error())
|
|
return
|
|
}
|
|
|
|
final_cost, err := h.uc.GetCargoCost(cargo_id)
|
|
if err != nil {
|
|
|
|
sendErrorResponse(w, http.StatusInternalServerError, "", err.Error())
|
|
|
|
return
|
|
}
|
|
|
|
response := struct {
|
|
CargoCost float64 `json:"cargo_cost_count"`
|
|
}{
|
|
CargoCost: final_cost,
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(response)
|
|
}
|