diff options
author | Unknwon <u@gogs.io> | 2016-08-14 17:44:20 -0700 |
---|---|---|
committer | Unknwon <u@gogs.io> | 2016-08-14 23:52:24 -0700 |
commit | 660e7a178a9d72a03d13f704aca61726b4672232 (patch) | |
tree | 0668bcf438c586df14c204625b3829f5a4647267 /modules/sync | |
parent | 15845cb28763c1542556ad61d2aa9735541dbf45 (diff) | |
download | gitea-660e7a178a9d72a03d13f704aca61726b4672232.tar.gz gitea-660e7a178a9d72a03d13f704aca61726b4672232.zip |
modules/sync: move sync objects to independent module
Diffstat (limited to 'modules/sync')
-rw-r--r-- | modules/sync/single_instance_pool.go | 67 | ||||
-rw-r--r-- | modules/sync/status_pool.go | 49 |
2 files changed, 116 insertions, 0 deletions
diff --git a/modules/sync/single_instance_pool.go b/modules/sync/single_instance_pool.go new file mode 100644 index 0000000000..4a00d5283d --- /dev/null +++ b/modules/sync/single_instance_pool.go @@ -0,0 +1,67 @@ +// Copyright 2016 The Gogs Authors. All rights reserved. +// Use of this source code is governed by a MIT-style +// license that can be found in the LICENSE file. + +package sync + +import ( + "sync" +) + +// SingleInstancePool is a pool of non-identical instances +// that only one instance with same identity is in the pool at a time. +// In other words, only instances with different identities can exist +// at the same time. +// +// This pool is particularly useful for performing tasks on same resource +// on the file system in different goroutines. +type SingleInstancePool struct { + lock sync.Mutex + + // pool maintains locks for each instance in the pool. + pool map[string]*sync.Mutex + + // count maintains the number of times an instance with same identity checks in + // to the pool, and should be reduced to 0 (removed from map) by checking out + // with same number of times. + count map[string]int +} + +// NewSingleInstancePool initializes and returns a new SingleInstancePool object. +func NewSingleInstancePool() *SingleInstancePool { + return &SingleInstancePool{ + pool: make(map[string]*sync.Mutex), + count: make(map[string]int), + } +} + +// CheckIn checks in an instance to the pool and hangs while instance +// with same indentity is using the lock. +func (p *SingleInstancePool) CheckIn(identity string) { + p.lock.Lock() + + lock, has := p.pool[identity] + if !has { + lock = &sync.Mutex{} + p.pool[identity] = lock + } + p.count[identity]++ + + p.lock.Unlock() + lock.Lock() +} + +// CheckOut checks out an instance from the pool and releases the lock +// to let other instances with same identity to grab the lock. +func (p *SingleInstancePool) CheckOut(identity string) { + p.lock.Lock() + defer p.lock.Unlock() + + p.pool[identity].Unlock() + if p.count[identity] == 1 { + delete(p.pool, identity) + delete(p.count, identity) + } else { + p.count[identity]-- + } +} diff --git a/modules/sync/status_pool.go b/modules/sync/status_pool.go new file mode 100644 index 0000000000..f6a7f9495c --- /dev/null +++ b/modules/sync/status_pool.go @@ -0,0 +1,49 @@ +// Copyright 2016 The Gogs Authors. All rights reserved. +// Use of this source code is governed by a MIT-style +// license that can be found in the LICENSE file. + +package sync + +import ( + "sync" +) + +// StatusTable is a table maintains true/false values. +// +// This table is particularly useful for un/marking and checking values +// in different goroutines. +type StatusTable struct { + lock sync.RWMutex + pool map[string]bool +} + +// NewStatusTable initializes and returns a new StatusTable object. +func NewStatusTable() *StatusTable { + return &StatusTable{ + pool: make(map[string]bool), + } +} + +// Start sets value of given name to true in the pool. +func (p *StatusTable) Start(name string) { + p.lock.Lock() + defer p.lock.Unlock() + + p.pool[name] = true +} + +// Stop sets value of given name to false in the pool. +func (p *StatusTable) Stop(name string) { + p.lock.Lock() + defer p.lock.Unlock() + + p.pool[name] = false +} + +// IsRunning checks if value of given name is set to true in the pool. +func (p *StatusTable) IsRunning(name string) bool { + p.lock.RLock() + defer p.lock.RUnlock() + + return p.pool[name] +} |