我正在嘗試使用 golang、redis 和 mognodb 創建一個 url shortener，但面臨實際數據庫連接的問題。在我的應用程序中，我有幾個包：base62||_base62.goconfig||__redis.gomodels||__url.gorequest||__shorten_request.goresponse||__shorten_response.goroutes||__shorten.go main.go在我的 shorten.go 文件中：func Shorten(context *gin.Context){ var request request.ShortenURLRequest if err := context.BindJSON(&request); err != nil { return } if !IsUrl(request.URL){ context.IndentedJSON(http.StatusBadRequest, "Error") return } // encode the long url short_url := base62.Encode(request.URL) // URL model var url models.URL url.ID = 1234567890 url.LONG_URL = request.URL url.SHORT_URL = short_url // insert the data in redis db // create the response to send back to client var response response.ShortenURLResponse response.URL = request.URL response.SHORT_URL = short_url response.CREATED_AT = time.Now() context.IndentedJSON(http.StatusOK, response)}現在我想以縮短和解析文件可以訪問數據庫的方式創建數據庫連接。我得到的所有資源都有一個用于連接的文件，所有路由和控制器都在其中定義。幫助將不勝感激
1 回答
千萬里不及你
TA貢獻1784條經驗 獲得超9個贊
根據您的main.go文件，我認為您可以這樣做：
func main() {
rdb, err := redis.NewClient()
if err != nil {
log.Fatal(err)
}
router := gin.Default()
router.POST("/api/v1/shorten", routes.Shorten(rdb))
router.GET("/api/v1/:url", routes.Resolve(rdb))
router.Run("localhost:9000")
}
func Shorten(rdb *redis.Client) gin.HandlerFunc {
return func(context *gin.Context) {
ctx := context.Request.Context()
...
if err := rdb.Set(ctx, key, value); err != nil {
context.IndentedJSON(http.StatusInternalServerError, "Error")
return
}
....
}
}
或者更好的是，您可以創建一個包含 redis 客戶端并充當請求路由器的結構，如下所示：
type Router struct {
rdb *redis.Client
}
func NewRouter(rdb *redis.Client) *Router {
return &Router{rdb: rdb}
}
func (r *Router) Shorten() gin.HandlerFunc {
return func(context *gin.Context) {
...
r.rdb.Set()
...
}
}
func (r *Router) Resolve() gin.HandlerFunc {
return func(context *gin.Context) {
...
r.rdb.Get()
...
}
}
func main() {
rdb, err := redis.NewClient()
if err != nil {
log.Fatal(err)
}
r := NewRouter(rdb)
router := gin.Default()
router.POST("/api/v1/shorten", r.Shorten)
router.GET("/api/v1/:url", r.Resolve)
router.Run("localhost:9000")
}
- 1 回答
- 0 關注
- 100 瀏覽
添加回答
舉報
0/150
提交
取消