1
0
mirror of https://github.com/golang/go synced 2024-11-22 10:54:46 -07:00

sync: add sync.Once, a proposed replacement for package once.

This implementation is simpler and more useful, since it can work
with methods.

R=rsc
CC=golang-dev
https://golang.org/cl/1864051
This commit is contained in:
Rob Pike 2010-08-05 13:31:10 +10:00
parent 9265bc2484
commit 895c5db6df
4 changed files with 74 additions and 3 deletions

View File

@ -7,6 +7,7 @@ include ../../Make.$(GOARCH)
TARG=sync TARG=sync
GOFILES=\ GOFILES=\
mutex.go\ mutex.go\
once.go \
rwmutex.go\ rwmutex.go\
# 386-specific object files # 386-specific object files

View File

@ -3,9 +3,10 @@
// license that can be found in the LICENSE file. // license that can be found in the LICENSE file.
// The sync package provides basic synchronization primitives // The sync package provides basic synchronization primitives
// such as mutual exclusion locks. These are intended for use // such as mutual exclusion locks. Other than the Once type,
// by low-level library routines. Higher-level synchronization // most are intended for use by low-level library routines.
// is better done via channels and communication. // Higher-level synchronization is better done via channels
// and communication.
package sync package sync
import "runtime" import "runtime"

32
src/pkg/sync/once.go Normal file
View File

@ -0,0 +1,32 @@
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package sync
// Once is an object that will perform exactly one action.
type Once struct {
m Mutex
done bool
}
// Do calls the function f if and only if the method is being called for the
// first time with this receiver. In other words, given
// var once Once
// if once.Do(f) is called multiple times, only the first call will invoke f,
// even if f has a different value in each invocation. A new instance of
// Once is required for each function to execute.
//
// Do is intended for initialization that must be run exactly once. Since f
// is niladic, it may be necessary to use a function literal to capture the
// arguments to a function to be invoked by Do:
// config.once.Do(func() { config.init(filename) })
//
func (o *Once) Do(f func()) {
o.m.Lock()
defer o.m.Unlock()
if !o.done {
o.done = true
f()
}
}

37
src/pkg/sync/once_test.go Normal file
View File

@ -0,0 +1,37 @@
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package sync_test
import (
. "sync"
"testing"
)
type one int
func (o *one) Increment() {
*o++
}
func run(once *Once, o *one, c chan bool) {
once.Do(func() { o.Increment() })
c <- true
}
func TestOnce(t *testing.T) {
o := new(one)
once := new(Once)
c := make(chan bool)
const N = 10
for i := 0; i < N; i++ {
go run(once, o, c)
}
for i := 0; i < N; i++ {
<-c
}
if *o != 1 {
t.Errorf("once failed: %d is not 1", *o)
}
}