答案:通过定义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是为了保证并发安全,因为多个用户可能同时投票或查看结果。
超级适合代理建设企业站点的企业源码,超方面实用!程序说明: 1.特色:简繁中文切换、产品展示系统、新闻发布系统、会员管理系统、留言本计数器、网站信息统计、强大后台操作 功能等; 2.页面包括:首页、企业介绍、滚动公告通知发布系统、企业新闻系统、产品展示系统、企业案例发布展示系 统、企业招聘信息发布系统、信息资源下载系统、在线定单系统、在线客服系统、在线留言本系统、网站调查投票系统、友情连接系统、会
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验证用户身份
- 增加过期时间控制投票周期
- 前端页面展示实时图表









