答案：通过定义VoteSystem结构体并使用sync.RWMutex实现并发安全的投票系统，支持添加选项、用户投票、查看结果及防重复投票功能，结合net/http提供REST API接口，最终构建一个基于内存的简单高效投票系统。
在Go语言中开发一个投票系统并不复杂，关键在于理解需求、设计合理的数据结构和接口，并利用Golang的并发特性提升性能。下面通过实战方式带你一步步构建一个简单但功能完整的投票系统。
1. 明确系统核心功能
一个基础的投票系统通常需要支持以下功能：
- 创建投票项：允许管理员或用户添加候选选项（如A、B、C）
- 提交投票：用户对某个选项进行投票
- 查看结果：实时显示各选项得票数
- 防止重复投票：确保每个用户只能投一次
为简化实现，我们先不引入数据库，使用内存存储数据，后续可扩展为持久化存储。
2. 定义数据结构与状态管理
使用map来保存选项及其票数，用另一个map记录已投票的用户ID，避免重复投票。
立即学习“go语言免费学习笔记（深入）”；
定义结构体封装投票状态：
type VoteSystem struct {
mu sync.RWMutex
options map[string]int // 选项 -> 票数
voted map[string]bool // 用户ID -> 是否已投票
}
注意：加入sync.RWMutex是为了保证并发安全，因为多个用户可能同时投票或查看结果。
3. 实现核心方法
为VoteSystem添加方法：
func (v *VoteSystem) AddOption(option string) {
v.mu.Lock()
defer v.mu.Unlock()
if v.options == nil {
v.options = make(map[string]int)
}
v.options[option] = 0
}
func (v *VoteSystem) Vote(userID, option string) error {
v.mu.Lock()
defer v.mu.Unlock()
if v.voted == nil {
v.voted = make(map[string]bool)
}
if v.voted[userID] {
return errors.New("用户已投票")
}
if _, exists := v.options[option]; !exists {
return errors.New("无效的选项")
}
v.options[option]++
v.voted[userID] = true
return nil
}
func (v *VoteSystem) GetResults() map[string]int {
v.mu.RLock()
defer v.mu.RUnlock()
result := make(map[string]int)
for k, v := range v.options {
result[k] = v
}
return result
}
4. 使用HTTP服务暴露API
利用net/http包提供REST风格接口：
func main() {
vs := &VoteSystem{
options: make(map[string]int),
voted: make(map[string]bool),
}
http.HandleFunc("/add", func(w http.ResponseWriter, r *http.Request) {
option := r.FormValue("option")
vs.AddOption(option)
fmt.Fprintf(w, "已添加选项: %s", option)
})
http.HandleFunc("/vote", func(w http.ResponseWriter, r *http.Request) {
userID := r.FormValue("user_id")
option := r.FormValue("option")
err := vs.Vote(userID, option)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
fmt.Fprintf(w, "投票成功")
})
http.HandleFunc("/results", func(w http.ResponseWriter, r *http.Request) {
results := vs.GetResults()
json.NewEncoder(w).Encode(results)
})
fmt.Println("服务器启动在 :8080")
http.ListenAndServe(":8080", nil)
}
这样就可以通过/add?option=A、/vote?user_id=u1&option=A、/results来操作投票系统。
5. 测试与优化建议
写个简单的测试验证逻辑正确性：
func TestVoteSystem(t *testing.T) {
vs := &VoteSystem{}
vs.AddOption("Go")
err := vs.Vote("user1", "Go")
if err != nil {
t.Fatal(err)
}
if vs.GetResults()["Go"] != 1 {
t.Fail()
}
}
优化方向：
- 接入Redis或数据库实现持久化
- 使用JWT验证用户身份
- 增加过期时间控制投票周期
- 前端页面展示实时图表
