Golang-HTTP动态路由

最新推荐文章于 2025-05-09 14:56:44 发布

原创最新推荐文章于 2025-05-09 14:56:44 发布 · 2.7k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#go #golang #restful

Golang 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了如何利用Gorilla/Mux库在Go语言中实现RESTful API的不同请求方法处理、请求转发、HTTPS与HTTP请求差异化处理以及子路由转发。通过示例代码详细展示了如何配置路由，包括创建、读取、更新和删除操作，请求转发到不同服务器，以及根据请求协议和路径前缀匹配不同处理函数。

在Restful服务中，我们可能期望实现这样的需求。

接口路径相同，但是根据不同的请求方式，进行不同的处理请求
将某个路径的请求转发到另一条主机
将 https 和 http 请求交给不同的handler处理
根据路径前缀匹配不同的子路由，转发给不同handler处理

强大的gorilla/mux即可实现这些需求

1. 根据请求方式不同进行路由

package main

import (
	"fmt"
	"net/http"

	"github.com/gorilla/mux"
)

func main() {
	r := mux.NewRouter()
	
	r.HandleFunc("/books/{title}", CreateBook).Methods("POST")
	r.HandleFunc("/books/{title}", ReadBook).Methods("GET")
	r.HandleFunc("/books/{title}", UpdateBook).Methods("PUT")
	r.HandleFunc("/books/{title}", DeleteBook).Methods("DELETE")
	
	http.ListenAndServe(":8080", r)
}

func CreateBook(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	w.WriteHeader(http.StatusOK)
	fmt.Fprintf(w, "CreateBook: %v\n", vars["title"])
}

func ReadBook(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	w.WriteHeader(http.StatusOK)
	fmt.Fprintf(w, "ReadBook: %v\n", vars["title"])
}

func UpdateBook(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	w.WriteHeader(http.StatusOK)
	fmt.Fprintf(w, "UpdateBook: %v\n", vars["title"])
}

func DeleteBook(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	w.WriteHeader(http.StatusOK)
	fmt.Fprintf(w, "DeleteBook: %v\n", vars["title"])
}

2. 请求转发

假设我们启动两个server，其中server1监听8081端口，server2监听8080端口。
server1：

func main() {
	http.HandleFunc("/", func (w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, "Welcome to my website!")
	})
	http.ListenAndServe(":8081", nil)
}

server2:

func main() {
	r := mux.NewRouter()
	r.HandleFunc("/books/{title}", BookHandler).Host("localhost:8081")
	http.ListenAndServe(":8080", r)
}

由于server2中做了请求转发，对于/books/{title}的请求会转发到server1，所以访问http://localhost:8081/books/红楼梦才是有效的。

3. 对https和http请求分别处理

可以通过Schemes函数来区分请求是https还是http然后交给不同的handler处理。

func main() {
	r := mux.NewRouter()
	r.HandleFunc("/secure", SecureHandler).Schemes("https")
	r.HandleFunc("/insecure", InsecureHandler).Schemes("http")
	http.ListenAndServe(":8080", r)
}

4. 子路由转发

按照特定前缀匹配请求，对请求做子路由转发。如下，当请求为http://localhost:8080/books/，调用AllBooks函数。当请求为http://localhost:8080/books/红楼梦，调用GetBook函数。

func main() {
	r := mux.NewRouter()
	bookrouter := r.PathPrefix("/books").Subrouter()
	bookrouter.HandleFunc("/", AllBooks)
	bookrouter.HandleFunc("/{title}", GetBook)
	http.ListenAndServe(":8080", r)
}