可变全局变量在使用时,可能会被其他地方修改,导致程序运行时出现意外情况。

我们可以通过依赖注入的方式将依赖注入到使用方,这样便可以避免使用全局变量,进而避免了依赖的全局变量被错误修改的可能。

比如下面将依赖的 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))
}
powered by Gitbook该文章修订时间: 2024-08-04 07:27:05

results matching ""

    No results matching ""