Golang构建投票系统教程

本文深入浅出地讲解了如何使用原生 Go（net/http）构建一个健壮、可维护的投票系统，强调“轻量不等于简陋”：从按 HTTP 方法清晰拆分路由（POST /vote 接收投票、GET /results 返回统计），到合理选择持久化方案（sync.Map 适用于低并发演示，SQLite 提供真正持久化与生产就绪能力），再到直击实战痛点——防重复提交、严格内容类型校验、超时控制、原子计票、日志埋点与边界错误处理，每一步都给出可直接复用的代码范式和避坑指南，帮你避开新手常踩的并发 panic、静默失败、状态混乱等深坑，真正写出既简单又可靠的服务。

用 net/http 启动服务并处理投票请求

Go 的 net/http 足够轻量，不需要框架也能快速响应投票动作。关键不是写得多，而是路由清晰、状态明确。

常见错误是把所有逻辑塞进一个 http.HandleFunc，导致无法测试、难加日志、难控制并发。应该按 HTTP 方法拆分：用 POST /vote 接收投票，用 GET /results 返回统计。

• http.HandleFunc 里只做参数解析、校验和简单转发，业务逻辑抽到独立函数
• 务必检查 r.Method，避免 GET 请求意外触发写操作
• 用 r.ParseForm() 后读 r.FormValue("option")，别直接读 r.Body —— 否则后续中间件或日志拿不到原始数据
• 返回时统一用 http.StatusCreated（201）表示成功投票，http.StatusBadRequest（400）表示选项非法

func voteHandler(w http.ResponseWriter, r *http.Request) {
    if r.Method != http.MethodPost {
        http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
        return
    }
    if err := r.ParseForm(); err != nil {
        http.Error(w, "bad request", http.StatusBadRequest)
        return
    }
    option := r.FormValue("option")
    if option != "a" && option != "b" {
        http.Error(w, "invalid option", http.StatusBadRequest)
        return
    }
    if err := recordVote(option); err != nil {
        http.Error(w, "internal error", http.StatusInternalServerError)
        return
    }
    w.WriteHeader(http.StatusCreated)
    w.Write([]byte("ok"))
}

用 sync.Map 存临时计票结果

如果只是演示或低并发场景（比如课堂作业、内部工具），用内存存储比立刻上数据库更直接。但别用普通 map —— Go 的 map 并发读写会 panic。

sync.Map 是标准库提供的线程安全映射，适合“读多写少”的计票场景。它不支持遍历，所以统计时得配合额外变量或封装一层。

• 声明为包级变量即可，无需初始化：var votes sync.Map
• 用 votes.LoadOrStore("a", int64(0)) 初始化键值，再用 votes.Load("a") + 类型断言取值
• 递增必须自己加锁或用原子操作：推荐用 atomic.AddInt64 配合指针存值，而不是反复 Load-Modify-Store
• 注意 sync.Map 的零值是可用的，但它的迭代性能差，不要在 /results 中直接 range 它

var voteCounts sync.Map // key: string, value: *int64
func recordVote(option string) error {
ptr, _ := voteCounts.LoadOrStore(option, new(int64))
atomic.AddInt64(ptr.(*int64), 1)
return nil
}
func getResults() map[string]int64 {
result := make(map[string]int64)
voteCounts.Range(func(key, value interface{}) bool {
result[key.(string)] = atomic.LoadInt64(value.(*int64))
return true
})
return result
}

切换到 SQLite 存储需改三处核心代码

当需要持久化、重启不丢票、或多人协作测试时，SQLite 是最平滑的落地选择。它不需要单独服务进程，单文件即数据库，Go 生态有成熟驱动 mattn/go-sqlite3。

切换重点不在 SQL 多复杂，而在连接管理、事务控制和错误传播。很多人卡在打开失败没报错、或忘记 db.Close() 导致文件被锁。

• 用 sql.Open("sqlite3", "votes.db")，不是 sql.Open("sqlite", ...) —— 驱动名写错会静默失败
• 建表语句必须包含 PRIMARY KEY 和 NOT NULL，否则插入空值可能成功但查不出来
• 每次投票用 db.Exec("INSERT INTO votes(option) VALUES(?)", option)，别用 QueryRow —— 写操作不需要返回行
• 统计时用 db.Query("SELECT option, COUNT(*) FROM votes GROUP BY option")，手动 scan 到结构体或 map

var db *sql.DB
func initDB() error {
var err error
db, err = sql.Open("sqlite3", "votes.db")
if err != nil {
return err
}
_, err = db.Exec(CREATE TABLE IF NOT EXISTS votes ( id INTEGER PRIMARY KEY, option TEXT NOT NULL, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ))
return err
}

HTTP 处理中容易忽略的边界问题

投票系统看似简单，但真实部署时最常崩在边界：重复提交、跨站请求、超长参数、时区混乱。这些问题不会在本地 curl 测试时暴露。

• 前端没禁用按钮？后端要防重复：用 r.RemoteAddr + option 做简易去重（存在 Redis 更好，但至少先加一层）
• 没设 Content-Type: application/x-www-form-urlencoded？r.ParseForm() 会失败，返回空值 —— 加个 if r.Header.Get("Content-Type") == "" 提前拦截
• SQLite 时间字段用 CURRENT_TIMESTAMP 是本地时区，但 Go 的 time.Now() 默认也是本地时区，只要不跨时区服务器就一致；若需 UTC，建表时用 CURRENT_TIMESTAMP AT TIME ZONE 'UTC'（SQLite 3.38+）
• 生产环境必须加 http.Server{ReadTimeout: 5 * time.Second}，否则慢连接耗尽 goroutine

真正麻烦的从来不是“怎么存”，而是“谁在什么时候以什么方式改了它”。哪怕只是 demo，也建议在 recordVote 开头打一行日志：log.Printf("vote from %s for %s", r.RemoteAddr, option) —— 出问题时这行比任何文档都管用。

今天关于《Golang构建投票系统教程》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！