用Go語言標準庫實現(xiàn)Web服務之創(chuàng)建路由

更新時間：2023年05月21日 10:27:49 作者：LightSaid

在上一節(jié)中創(chuàng)建了項目，這篇文章主要介紹如何用Go語言標準庫創(chuàng)建路由，文中有詳細的代碼示例，對大家的學習或工作有一定的幫助，感興趣的同學可以參考下

在上一節(jié)中創(chuàng)建了項目；現(xiàn)在打開internal/app/app.go中創(chuàng)建一個application結(jié)構(gòu)體來管理api服務：

// application 一個管理API服務的結(jié)構(gòu)體
type application struct {
    cfg   config.AppConfig
    store dbrepo.Repository
}

在上面其中 dbrepo.Repository 數(shù)據(jù)庫操作層，先創(chuàng)建一個結(jié)構(gòu)體,后續(xù)實現(xiàn)

internal/dbrepo/repository.go

type Repository struct {
}
func NewRepository(db *sql.DB) Repository {
    return Repository{
        Book: NewBookRepo(db),
    }
}

在這里我希望application實例管理整個api服務，包括啟動服務、關閉服務、路由注冊、路由處理、中間件等功能。

因此有一個newApplication函數(shù)，做初始化服務工作，首先要初始化就是加載配置文件，在系統(tǒng)中配置文件是一個關鍵信息，要最先初始化，從而保證后面用到配置的對象能夠正常創(chuàng)建。

func newApplication() *application {
	// 解析命令行參數(shù)
	var cfgPath string
	flag.StringVar(&cfgPath, "path", "configs/app.json", "config 配置文件路徑")
	flag.Parse()

	// 加載app配置
	cfg, err := config.LoadAppConfig(cfgPath)
	if err != nil {
		log.Fatal(err)
	}
        // 格式化輸出一下配置，僅開發(fā)模式下
	cfg.Println()

	// 設置全局日志
	logger.SetGlobalLogger()

	// 連接mysql
	db, err := sql.Open("mysql", cfg.DSN)
	if err != nil {
		log.Fatal(err)
	}

	// ping 確定能連接上數(shù)據(jù)庫
	if err = db.Ping(); err != nil {
		log.Fatal(err)
	}

	// 設置最大鏈接數(shù)、最大空閑數(shù)和最大空閑時間
	db.SetMaxOpenConns(cfg.MaxOpenConns)
	db.SetMaxIdleConns(cfg.MaxIdleConns)
	db.SetConnMaxIdleTime(cfg.MaxIdleTimeToDuration())

	a := &application{
		cfg:   cfg,
		store: dbrepo.NewRepository(db),
	}

	return a
}

上面函數(shù)一開始使用flag庫來解析命令行參數(shù)，函數(shù)解析：flag.StringVar(解析后賦值給誰,參數(shù)名字,參數(shù)默認值,參數(shù)提示語)，最后一定要用flag.Parse()解析參數(shù)。那么這個命令參數(shù)是怎么用的呢？簡單，在啟動的時候如下：

go run main.go -path=configs/app.jso

命令行參數(shù)使用介紹完畢，在這里順便另一個場景，例如在別人代碼中經(jīng)?？吹竭@樣一個用法,如：

env := os.Getenv("APP_ENV")

那么這個APP_ENV環(huán)境變量在哪里配置的呢？這里介紹兩種情況設置:

- 1. 在代碼中設置 os.Setenv(key string, value string)
- 2. 啟動程序通過命令設置如：
    APP_ENV=dev go run main.go

另外在這里我修改了maxOpenConns、maxIdleConns、maxIdleTime 配置，目前來講對一個中小型項目來說剛運營時，這個配置是最合適的。后續(xù)可以根據(jù)服務流量調(diào)整。

{
    ...
    "maxOpenConns": 25,
    "maxIdleConns": 25,
    "maxIdleTime": "5m"
}

接下來就是創(chuàng)建一個http.Server 實例，使用這個實例來啟動API服務。

func (a *application) serve() error {
    var address = fmt.Sprintf("0.0.0.0:%d", a.cfg.Port)

    // 創(chuàng)建 http.Server 用來啟動 Web Server
    srv := http.Server{
            Addr:         address,
            Handler:      a.routes(),
            IdleTimeout:  time.Minute,
            ReadTimeout:  15 * time.Second,
            WriteTimeout: 30 * time.Second,
    }

    logger.InfoLog.Println("Starting API Server on: " + address)
    return srv.ListenAndServe()
}

func Serve() error {
    return newApplication().serve()
}

首先在這里func (a *application) serve() 用于內(nèi)部創(chuàng)建一個 http.Server 和啟動server; 而 func Serve() 是提供外部的main.go 中main函數(shù)調(diào)用。很多開源項目都是這個做法，對導出去函數(shù)簡化。

然后使用http.NewServeMux()創(chuàng)建路由：

internal/app/routes.go

package app

import "net/http"

func (a *application) routes() http.Handler {
	mux := http.NewServeMux()

	mux.HandleFunc("/v1/book", a.tmp)       // GET    獲取一個圖書 book?id=1
	mux.HandleFunc("/v1/book/post", a.tmp)  // POST   創(chuàng)建一個圖書
	mux.HandleFunc("/v1/book/put", a.tmp)   // PUT    更新一個圖書
	mux.HandleFunc("/v1/book/del", a.tmp)   // DELETE 刪除一個圖書
	mux.HandleFunc("/v1/book/list", a.tmp) // GET    獲取圖書列表 list?page_idx=1&pageNum=10

	return mux
}

func (a *application) tmp(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "%s:%s", r.Method, r.RequestURI)
}

最后就是啟動程序～

main.go

package main

import (
    "log"
    "github.com/lightsaid/ebook/internal/app"
)

func main() {
    if err := app.Serve(); err != nil {
            log.Fatal(err)
    }
}

人生苦短 Let's Go