使用 kickstart.go 创建轻量级 HTTP 服务器

kickstart.go 是一个简洁的 Go 语言单文件 HTTP 服务器模板，非常适合快速启动小型 Web 项目或 API 服务。下面我将详细介绍如何使用它。

基本使用方法

首先，获取 kickstart.go 文件（假设你已经有了这个模板文件），然后可以这样使用：

package main

import (
	"fmt"
	"net/http"
)

func main() {
	// 设置路由
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "欢迎使用 kickstart.go HTTP 服务器!")
	})

	// 启动服务器
	fmt.Println("服务器启动在 :8080...")
	http.ListenAndServe(":8080", nil)
}

增强功能模板

下面是一个更完整的 kickstart.go 模板，包含常用功能：

package main

import (
	"encoding/json"
	"fmt"
	"log"
	"net/http"
	"time"
)

// 配置结构体
type Config struct {
	Port    string `json:"port"`
	Env     string `json:"env"`
	Version string `json:"version"`
}

// 应用结构体
type Application struct {
	Config Config
	Logger *log.Logger
}

func main() {
	// 初始化配置
	cfg := Config{
		Port:    ":8080",
		Env:     "development",
		Version: "1.0.0",
	}

	// 初始化应用
	app := &Application{
		Config: cfg,
		Logger: log.New(log.Writer(), "kickstart: ", log.LstdFlags|log.Lshortfile),
	}

	// 设置路由
	mux := http.NewServeMux()
	mux.HandleFunc("/", app.homeHandler)
	mux.HandleFunc("/api/data", app.dataHandler)
	mux.HandleFunc("/health", app.healthHandler)

	// 创建自定义服务器
	srv := &http.Server{
		Addr:         cfg.Port,
		Handler:      app.logRequest(mux),
		IdleTimeout:  time.Minute,
		ReadTimeout:  10 * time.Second,
		WriteTimeout: 30 * time.Second,
	}

	// 启动服务器
	app.Logger.Printf("启动 %s 服务器在 %s", cfg.Env, cfg.Port)
	err := srv.ListenAndServe()
	app.Logger.Fatal(err)
}

// 处理器函数
func (app *Application) homeHandler(w http.ResponseWriter, r *http.Request) {
	if r.URL.Path != "/" {
		http.NotFound(w, r)
		return
	}
	fmt.Fprintf(w, "欢迎使用 kickstart.go (v%s)", app.Config.Version)
}

func (app *Application) dataHandler(w http.ResponseWriter, r *http.Request) {
	data := map[string]interface{}{
		"status":  "success",
		"message": "这是API数据",
		"version": app.Config.Version,
	}

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(data)
}

func (app *Application) healthHandler(w http.ResponseWriter, r *http.Request) {
	w.Write([]byte("OK"))
}

// 中间件：记录请求
func (app *Application) logRequest(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		start := time.Now()
		next.ServeHTTP(w, r)
		app.Logger.Printf("%s %s %s %v", r.Method, r.URL.Path, r.RemoteAddr, time.Since(start))
	})
}

功能特点

1. 简洁结构：单文件包含所有必要组件
2. 配置管理：使用结构体管理配置
3. 日志记录：内置请求日志中间件
4. 优雅超时：设置读写和空闲超时
5. 路由处理：支持多个路由处理器
6. JSON API：内置JSON响应支持

如何扩展

你可以根据需要添加以下功能：

1. 添加数据库支持：

import "database/sql"
import _ "github.com/lib/pq"

// 在Application结构体中添加
type Application struct {
	Config Config
	Logger *log.Logger
	DB     *sql.DB
}

// 初始化数据库
db, err := sql.Open("postgres", "your-connection-string")
if err != nil {
    log.Fatal(err)
}
app.DB = db

1. 添加环境变量支持：

import "os"

// 在配置中
Port:    getEnv("PORT", "8080"),
Env:     getEnv("ENV", "development"),

// 辅助函数
func getEnv(key, defaultValue string) string {
    if value, exists := os.LookupEnv(key); exists {
        return value
    }
    return defaultValue
}

1. 添加静态文件服务：

// 在路由设置中添加
fs := http.FileServer(http.Dir("./static"))
mux.Handle("/static/", http.StripPrefix("/static", fs))

这个 kickstart.go 模板提供了足够的基础功能，同时保持简洁，你可以根据项目需求轻松扩展它。