status_pool.go 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. // Copyright 2016 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package sync
  5. import (
  6. "sync"
  7. )
  8. // StatusTable is a table maintains true/false values.
  9. //
  10. // This table is particularly useful for un/marking and checking values
  11. // in different goroutines.
  12. type StatusTable struct {
  13. lock sync.RWMutex
  14. pool map[string]bool
  15. }
  16. // NewStatusTable initializes and returns a new StatusTable object.
  17. func NewStatusTable() *StatusTable {
  18. return &StatusTable{
  19. pool: make(map[string]bool),
  20. }
  21. }
  22. // Start sets value of given name to true in the pool.
  23. func (p *StatusTable) Start(name string) {
  24. p.lock.Lock()
  25. defer p.lock.Unlock()
  26. p.pool[name] = true
  27. }
  28. // Stop sets value of given name to false in the pool.
  29. func (p *StatusTable) Stop(name string) {
  30. p.lock.Lock()
  31. defer p.lock.Unlock()
  32. p.pool[name] = false
  33. }
  34. // IsRunning checks if value of given name is set to true in the pool.
  35. func (p *StatusTable) IsRunning(name string) bool {
  36. p.lock.RLock()
  37. defer p.lock.RUnlock()
  38. return p.pool[name]
  39. }