用Go语言标准库实现Web服务之创建路由

// application 一个管理API服务的结构体
type application struct {
    cfg   config.AppConfig
    store dbrepo.Repository
}

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

func newApplication() *application {
	// 解析命令行参数
	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)
	}
        // 格式化输出一下配置，仅开发模式下
	cfg.Println()

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

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

	// ping 确定能连接上数据库
	if err = db.Ping(); err != nil {
		log.Fatal(err)
	}

	// 设置最大链接数、最大空闲数和最大空闲时间
	db.SetMaxOpenConns(cfg.MaxOpenConns)
	db.SetMaxIdleConns(cfg.MaxIdleConns)
	db.SetConnMaxIdleTime(cfg.MaxIdleTimeToDuration())

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

	return a
}

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

env := os.Getenv("APP_ENV")

- 1. 在代码中设置 os.Setenv(key string, value string)
- 2. 启动程序通过命令设置如：
    APP_ENV=dev go run main.go

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

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

    // 创建 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()
}

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   创建一个图书
	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)
}

package main

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

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