可变全局变量在使用时,可能会被其他地方修改,导致程序运行时出现意外情况。
我们可以通过依赖注入的方式将依赖注入到使用方,这样便可以避免使用全局变量,进而避免了依赖的全局变量被错误修改的可能。
比如下面将依赖的 now 函数直接注入到使用方 signer struct 中。
// Bad
// sign.go
var timeNow = time.Now
func sign(msg string) string {
now := timeNow()
return signWithTime(msg, now)
}
// sign_test.go
func TestSign(t *testing.T) {
oldTimeNow := timeNow
timeNow = func() time.Time {
return someFixedTime
}
defer func() { timeNow = oldTimeNow }()
assert.Equal(t, want, sign(give))
}
// Good
// sign.go
type signer struct {
now func() time.Time
}
func newSigner() *signer {
return &signer{
now: time.Now,
}
}
func (s *signer) Sign(msg string) string {
now := s.now()
return signWithTime(msg, now)
}
// sign_test.go
func TestSigner(t *testing.T) {
s := newSigner()
s.now = func() time.Time {
return someFixedTime
}
assert.Equal(t, want, s.Sign(give))
}