Sync - Learn Go with tests

Search…

Sync

​You can find all the code for this chapter here​
We want to make a counter which is safe to use concurrently.
We'll start with an unsafe counter and verify its behaviour works in a single-threaded environment.
Then we'll exercise it's unsafeness with multiple goroutines trying to use it via a test and fix it.
Write the test first
We want our API to give us a method to increment the counter and then retrieve its value.
1
func TestCounter(t *testing.T) {
2
	t.Run("incrementing the counter 3 times leaves it at 3", func(t *testing.T) {
3
		counter := Counter{}
4
		counter.Inc()
5
		counter.Inc()
6
		counter.Inc()
7
​
8
		if counter.Value() != 3 {
9
			t.Errorf("got %d, want %d", counter.Value(), 3)
10
		}
11
	})
12
}
Copied!
Try to run the test
1
./sync_test.go:9:14: undefined: Counter
Copied!
Write the minimal amount of code for the test to run and check the failing test output
Let's define Counter.
1
type Counter struct {
2
}
Copied!
Try again and it fails with the following
1
./sync_test.go:14:10: counter.Inc undefined (type Counter has no field or method Inc)
2
./sync_test.go:18:13: counter.Value undefined (type Counter has no field or method Value)
Copied!
So to finally make the test run we can define those methods
1
func (c *Counter) Inc() {
2
​
3
}
4
​
5
func (c *Counter) Value() int {
6
	return 0
7
}
Copied!
It should now run and fail
1
=== RUN   TestCounter
2
=== RUN   TestCounter/incrementing_the_counter_3_times_leaves_it_at_3
3
--- FAIL: TestCounter (0.00s)
4
    --- FAIL: TestCounter/incrementing_the_counter_3_times_leaves_it_at_3 (0.00s)
5
    	sync_test.go:27: got 0, want 3
Copied!
Write enough code to make it pass
This should be trivial for Go experts like us. We need to keep some state for the counter in our datatype and then increment it on every Inc call
1
type Counter struct {
2
	value int
3
}
4
​
5
func (c *Counter) Inc() {
6
	c.value++
7
}
8
​
9
func (c *Counter) Value() int {
10
	return c.value
11
}
Copied!
Refactor
There's not a lot to refactor but given we're going to write more tests around Counter we'll write a small assertion function assertCount so the test reads a bit clearer.
1
t.Run("incrementing the counter 3 times leaves it at 3", func(t *testing.T) {
2
	counter := Counter{}
3
	counter.Inc()
4
	counter.Inc()
5
	counter.Inc()
6
​
7
	assertCounter(t, counter, 3)
8
})
Copied!
1
func assertCounter(t testing.TB, got Counter, want int) {
2
	t.Helper()
3
	if got.Value() != want {
4
		t.Errorf("got %d, want %d", got.Value(), want)
5
	}
6
}
Copied!
Next steps
That was easy enough but now we have a requirement that it must be safe to use in a concurrent environment. We will need to write a failing test to exercise this.
Write the test first
1
t.Run("it runs safely concurrently", func(t *testing.T) {
2
	wantedCount := 1000
3
	counter := Counter{}
4
​
5
	var wg sync.WaitGroup
6
	wg.Add(wantedCount)
7
​
8
	for i := 0; i < wantedCount; i++ {
9
		go func() {
10
			counter.Inc()
11
			wg.Done()
12
		}()
13
	}
14
	wg.Wait()
15
​
16
	assertCounter(t, counter, wantedCount)
17
})
Copied!
This will loop through our wantedCount and fire a goroutine to call counter.Inc().
We are using sync.WaitGroup which is a convenient way of synchronising concurrent processes.
A WaitGroup waits for a collection of goroutines to finish. The main goroutine calls Add to set the number of goroutines to wait for. Then each of the goroutines runs and calls Done when finished. At the same time, Wait can be used to block until all goroutines have finished.
By waiting for wg.Wait() to finish before making our assertions we can be sure all of our goroutines have attempted to Inc the Counter.
Try to run the test
1
=== RUN   TestCounter/it_runs_safely_in_a_concurrent_envionment
2
--- FAIL: TestCounter (0.00s)
3
    --- FAIL: TestCounter/it_runs_safely_in_a_concurrent_envionment (0.00s)
4
    	sync_test.go:26: got 939, want 1000
5
FAIL
Copied!
The test will probably fail with a different number, but nonetheless it demonstrates it does not work when multiple goroutines are trying to mutate the value of the counter at the same time.
Write enough code to make it pass
A simple solution is to add a lock to our Counter, a Mutex​
A Mutex is a mutual exclusion lock. The zero value for a Mutex is an unlocked mutex.
1
type Counter struct {
2
	mu    sync.Mutex
3
	value int
4
}
5
​
6
func (c *Counter) Inc() {
7
	c.mu.Lock()
8
	defer c.mu.Unlock()
9
	c.value++
10
}
Copied!
What this means is any goroutine calling Inc will acquire the lock on Counter if they are first. All the other goroutines will have to wait for it to be Unlocked before getting access.
If you now re-run the test it should now pass because each goroutine has to wait its turn before making a change.
I've seen other examples where the sync.Mutex is embedded into the struct.
You may see examples like this
1
type Counter struct {
2
	sync.Mutex
3
	value int
4
}
Copied!
It can be argued that it can make the code a bit more elegant.
1
func (c *Counter) Inc() {
2
	c.Lock()
3
	defer c.Unlock()
4
	c.value++
5
}
Copied!
This looks nice but while programming is a hugely subjective discipline, this is bad and wrong.
Sometimes people forget that embedding types means the methods of that type becomes part of the public interface; and you often will not want that. Remember that we should be very careful with our public APIs, the moment we make something public is the moment other code can couple themselves to it. We always want to avoid unnecessary coupling.
Exposing Lock and Unlock is at best confusing but at worst potentially very harmful to your software if callers of your type start calling these methods.
Showing how a user of this API can wrongly change the state of the lock
This seems like a really bad idea
Copying mutexes
Our test passes but our code is still a bit dangerous
If you run go vet on your code you should get an error like the following
1
sync/v2/sync_test.go:16: call of assertCounter copies lock value: v1.Counter contains sync.Mutex
2
sync/v2/sync_test.go:39: assertCounter passes lock by value: v1.Counter contains sync.Mutex
Copied!
A look at the documentation of sync.Mutex tells us why
A Mutex must not be copied after first use.
When we pass our Counter (by value) to assertCounter it will try and create a copy of the mutex.
To solve this we should pass in a pointer to our Counter instead, so change the signature of assertCounter
1
func assertCounter(t testing.TB, got *Counter, want int)
Copied!
Our tests will no longer compile because we are trying to pass in a Counter rather than a *Counter. To solve this I prefer to create a constructor which shows readers of your API that it would be better to not initialise the type yourself.
1
func NewCounter() *Counter {
2
	return &Counter{}
3
}
Copied!
Use this function in your tests when initialising Counter.
Wrapping up
We've covered a few things from the sync package​
Mutex allows us to add locks to our data
WaitGroup is a means of waiting for goroutines to finish jobs
When to use locks over channels and goroutines?
​We've previously covered goroutines in the first concurrency chapter which let us write safe concurrent code so why would you use locks? The go wiki has a page dedicated to this topic; Mutex Or Channel​
A common Go newbie mistake is to over-use channels and goroutines just because it's possible, and/or because it's fun. Don't be afraid to use a sync.Mutex if that fits your problem best. Go is pragmatic in letting you use the tools that solve your problem best and not forcing you into one style of code.
Paraphrasing:
Use channels when passing ownership of data
Use mutexes for managing state
go vet
Remember to use go vet in your build scripts as it can alert you to some subtle bugs in your code before they hit your poor users.
Don't use embedding because it's convenient
Think about the effect embedding has on your public API.
Do you really want to expose these methods and have people coupling their own code to them?
With respect to mutexes, this could be potentially disastrous in very unpredictable and weird ways, imagine some nefarious code unlocking a mutex when it shouldn't be; this would cause some very strange bugs that will be hard to track down.