我正在嘗試使用 golang、redis 和 mognodb 創建一個 url 縮短器，但面臨實際資料庫連接的問題。
在我的應用程式中，我有幾個包：
base62
|
|_base62.go
config
|
|__redis.go
models
|
|__url.go
request
|
|__shorten_request.go
response
|
|__shorten_response.go
routes
|
|__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)
}
現在我想以縮短和決議檔案可以訪問資料庫的方式創建資料庫連接。
我獲得的所有資源都有一個用于連接的檔案，并且所有路由和控制器都在其中定義。
幫助將不勝感激
源代碼可以在這里找到：github
uj5u.com熱心網友回復：
根據您的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")
}
轉載請註明出處，本文鏈接：https://www.uj5u.com/caozuo/522796.html
標籤：去雷迪斯
上一篇：goudp請求（包）丟失