用net/http启动投票服务需按HTTP方法拆分路由,POST /vote处理投票,GET /results返回统计;用sync.Map或SQLite持久化计票结果;注意防重复提交、内容类型校验、超时控制及日志记录。

用 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 是标准库提供的线程安全映射,适合“读多写少”的计票场景。它不支持遍历,所以统计时得配合额外变量或封装一层。
立即学习“go语言免费学习笔记(深入)”;
- 声明为包级变量即可,无需初始化:
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: *int64func 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.DBfunc 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) —— 出问题时这行比任何文档都管用。










