41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
package controller
|
|
|
|
import (
|
|
"electricity_bill_calc/response"
|
|
"electricity_bill_calc/service"
|
|
"net/http"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
func InitializeRegionController(router *fiber.App) {
|
|
router.Get("/region/:rid", fetchRegions)
|
|
router.Get("/regions/:rid", fetchAllLeveledRegions)
|
|
}
|
|
|
|
func fetchRegions(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
requestParentId := c.Params("rid")
|
|
regions, err := service.RegionService.FetchSubRegions(requestParentId)
|
|
if err != nil {
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
if len(regions) == 0 {
|
|
return result.Json(http.StatusNotFound, "未能获取到相关的行政区划。", fiber.Map{"regions": make([]string, 0)})
|
|
}
|
|
return result.Json(http.StatusOK, "已经获取到相关的行政区划。", fiber.Map{"regions": regions})
|
|
}
|
|
|
|
func fetchAllLeveledRegions(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
requestRegionCode := c.Params("rid")
|
|
regions, err := service.RegionService.FetchAllParentRegions(requestRegionCode)
|
|
if err != nil {
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
if len(regions) == 0 {
|
|
return result.Json(http.StatusNotFound, "未能获取到相关的行政区划。", fiber.Map{"regions": make([]string, 0)})
|
|
}
|
|
return result.Json(http.StatusOK, "以及获取到相关的行政区划。", fiber.Map{"regions": regions})
|
|
}
|