Use sync/atomic instead of Uber's implementation (#256)

Use sync/atomic instead of Uber's implementation (#256) #566