ZiHangQin/electricity_bill_calc_service
1
0
Fork 0
forked from free-lancers/electricity_bill_calc_service
Code Pull Requests Activity

refactor(session):使用统一的用户会话获取方法。

Browse Source
This commit is contained in:
徐涛 2022-08-16 06:11:25 +08:00
parent edabcea58d
commit 272847446d
1 changed files with 21 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
controller/park.go
View File

@ -35,14 +35,9 @@ func InitializeParkController(router *gin.Engine) {
func listAllParksUnderSessionUser(c *gin.Context) { func listAllParksUnderSessionUser(c *gin.Context) {
result := response.NewResult(c) result := response.NewResult(c)
session, exists := c.Get("session") userSession, err := _retreiveSession(c)
if !exists { if err != nil {
result.Error(http.StatusUnauthorized, "用户会话无效。") result.Failure(http.StatusInternalServerError, err.Error())
return
}
userSession, ok := session.(*model.Session)
if !ok {
result.Failure(http.StatusInternalServerError, "内部缓存错误，需要重新登录。")
return return
} }
parks, err := service.ParkService.ListAllParkBelongsTo(userSession.Uid) parks, err := service.ParkService.ListAllParkBelongsTo(userSession.Uid)
@ -79,14 +74,9 @@ type _ParkInfoFormData struct {
func createNewPark(c *gin.Context) { func createNewPark(c *gin.Context) {
result := response.NewResult(c) result := response.NewResult(c)
session, exists := c.Get("session") userSession, err := _retreiveSession(c)
if !exists { if err != nil {
result.Error(http.StatusUnauthorized, "用户会话无效。") result.Failure(http.StatusInternalServerError, err.Error())
return
}
userSession, ok := session.(*model.Session)
if !ok {
result.Failure(http.StatusInternalServerError, "内部缓存错误，需要重新登录。")
return return
} }
formData := new(_ParkInfoFormData) formData := new(_ParkInfoFormData)
@ -98,7 +88,7 @@ func createNewPark(c *gin.Context) {
nameAbbr := utils.PinyinAbbr(newPark.Name) nameAbbr := utils.PinyinAbbr(newPark.Name)
newPark.Abbr = &nameAbbr newPark.Abbr = &nameAbbr
newPark.Enabled = true newPark.Enabled = true
err := service.ParkService.SaveNewPark(*newPark) err = service.ParkService.SaveNewPark(*newPark)
if err != nil { if err != nil {
result.Error(http.StatusInternalServerError, err.Error()) result.Error(http.StatusInternalServerError, err.Error())
return return
@ -108,14 +98,9 @@ func createNewPark(c *gin.Context) {
func modifyPark(c *gin.Context) { func modifyPark(c *gin.Context) {
result := response.NewResult(c) result := response.NewResult(c)
session, exists := c.Get("session") userSession, err := _retreiveSession(c)
if !exists { if err != nil {
result.Error(http.StatusUnauthorized, "用户会话无效。") result.Failure(http.StatusInternalServerError, err.Error())
return
}
userSession, ok := session.(*model.Session)
if !ok {
result.Failure(http.StatusInternalServerError, "内部缓存错误，需要重新登录。")
return return
} }
requestParkId := c.Param("pid") requestParkId := c.Param("pid")
@ -143,14 +128,9 @@ func modifyPark(c *gin.Context) {
func fetchParkDetail(c *gin.Context) { func fetchParkDetail(c *gin.Context) {
result := response.NewResult(c) result := response.NewResult(c)
session, exists := c.Get("session") userSession, err := _retreiveSession(c)
if !exists { if err != nil {
result.Error(http.StatusUnauthorized, "用户会话无效。") result.Failure(http.StatusInternalServerError, err.Error())
return
}
userSession, ok := session.(*model.Session)
if !ok {
result.Failure(http.StatusInternalServerError, "内部缓存错误，需要重新登录。")
return return
} }
requestParkId := c.Param("pid") requestParkId := c.Param("pid")
@ -172,20 +152,15 @@ type _ParkStateFormData struct {
func changeParkEnableState(c *gin.Context) { func changeParkEnableState(c *gin.Context) {
result := response.NewResult(c) result := response.NewResult(c)
session, exists := c.Get("session") userSession, err := _retreiveSession(c)
if !exists { if err != nil {
result.Error(http.StatusUnauthorized, "用户会话无效。") result.Failure(http.StatusInternalServerError, err.Error())
return
}
userSession, ok := session.(*model.Session)
if !ok {
result.Failure(http.StatusInternalServerError, "内部缓存错误，需要重新登录。")
return return
} }
requestParkId := c.Param("pid") requestParkId := c.Param("pid")
formData := new(_ParkStateFormData) formData := new(_ParkStateFormData)
c.BindJSON(formData) c.BindJSON(formData)
err := service.ParkService.ChangeParkState(userSession.Uid, requestParkId, formData.Enabled) err = service.ParkService.ChangeParkState(userSession.Uid, requestParkId, formData.Enabled)
if err != nil { if err != nil {
result.Error(http.StatusInternalServerError, err.Error()) result.Error(http.StatusInternalServerError, err.Error())
return return
@ -195,18 +170,13 @@ func changeParkEnableState(c *gin.Context) {
func deleteSpecificPark(c *gin.Context) { func deleteSpecificPark(c *gin.Context) {
result := response.NewResult(c) result := response.NewResult(c)
session, exists := c.Get("session") userSession, err := _retreiveSession(c)
if !exists { if err != nil {
result.Error(http.StatusUnauthorized, "用户会话无效。") result.Failure(http.StatusInternalServerError, err.Error())
return
}
userSession, ok := session.(*model.Session)
if !ok {
result.Failure(http.StatusInternalServerError, "内部缓存错误，需要重新登录。")
return return
} }
requestParkId := c.Param("pid") requestParkId := c.Param("pid")
err := service.ParkService.DeletePark(userSession.Uid, requestParkId) err = service.ParkService.DeletePark(userSession.Uid, requestParkId)
if err != nil { if err != nil {
result.Error(http.StatusInternalServerError, err.Error()) result.Error(http.StatusInternalServerError, err.Error())
} }