module_compare_check_test.go

// 测试 RoundRobin 中通过取余和通过判断大小的两种方法的性能
package main

import "testing"

type roundrobin struct {
	i       int      // 服务地址索引
	servers []string // 可用的服务地址
}

const count = 10

func (r *roundrobin) module() string {
	s := r.servers[r.i]
	r.i++
	r.i %= len(r.servers)
	return s
}
func (r *roundrobin) check() string {
	s := r.servers[r.i]
	r.i++
	if r.i == len(r.servers) {
		r.i = 0
	}
	return s
}

func BenchmarkModule(b *testing.B) {
	var r = roundrobin{
		servers: make([]string, count),
	}
	for i := 0; i < b.N; i++ {
		r.module()
	}
}
func BenchmarkCheck(b *testing.B) {
	var r = roundrobin{
		servers: make([]string, count),
	}
	for i := 0; i < b.N; i++ {
		r.check()
	}
}