Go 言語でリバースプロキシ

Go 言語の標準ライブラリ net/http/httputil には、リバースプロキシを作るための NewSingleHostReverseProxy 関数が用意されています。この関数を使うことで、Nginx で行うようなリバースプロキシを Go 言語のプログラム上で行うことができます。

下記コードを実行して http://localhost:10000/ にアクセスすると、hello, world が返ってきます。ポート番号 10000〜10003 が使われていると正常に動作しないので注意が必要です。

package main

import (
	"net/http"
	"net/http/httputil"
	"net/url"
	"time"

	"github.com/gin-gonic/gin"
)

func main() {
	// サーバーを作成する
	s10000 := gin.Default()
	s10001 := gin.Default()
	s10002 := gin.Default()
	s10003 := gin.Default()

	// s10000 へのルートパスのリクエストは s10001 と s10002 を経由して s10003 に渡る
	s10000.GET("/", makeReverseProxy("10001"))
	s10001.GET("/", makeReverseProxy("10002"))
	s10002.GET("/", makeReverseProxy("10003"))
	s10003.GET("/", func(c *gin.Context) {
		c.String(http.StatusOK, "hello, world")
	})

	// 非同期にサーバーを立ち上げる
	go s10000.Run(":10000")
	go s10001.Run(":10001")
	go s10002.Run(":10002")
	go s10003.Run(":10003")

	// プログラムが終わらないようにする
	time.Sleep(time.Hour)
}

func makeReverseProxy(port string) func(*gin.Context) {
	return func(c *gin.Context) {
		remote, _ := url.Parse("http://localhost:" + port)
		proxy := httputil.NewSingleHostReverseProxy(remote)
		proxy.ServeHTTP(c.Writer, c.Request)
	}
}

この仕組を使うことで、モノリシックな Web サーバーを複数の Web サーバーに分割できます。将来は複数のサーバーで運用したいけれど、とりあえずはひとつのサーバーで動かしたいといったときに使えるのではと思っています。