Golang实现简单投票统计方法

type VoteCounter struct {

    votes map[string]int

    mu sync.Mutex

}

func NewVoteCounter() *VoteCounter {

    return &VoteCounter{

        votes: make(map[string]int),

    }

}

func (vc *VoteCounter) Vote(candidate string) {

    vc.mu.Lock()

    defer vc.mu.Unlock()

    vc.votes[candidate]++

}

func (vc *VoteCounter) GetResults() map[string]int {

    vc.mu.Lock()

    defer vc.mu.Unlock()

    results := make(map[string]int)

    for k, v := range vc.votes {

        results[k] = v

    }

    return results

}

func main() {

    counter := NewVoteCounter()

    counter.Vote("Alice")

    counter.Vote("Bob")

    counter.Vote("Alice")

    results := counter.GetResults()

    for name, count := range results {

        fmt.Printf("%s: %d票\n", name, count)

    }

}

Alice: 2票

Bob: 1票