count_time.go 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179
  1. // package count_time -- счётчик обратного времени в мсек
  2. //
  3. // Похоже, нигде не используется
  4. package count_time
  5. import (
  6. "context"
  7. "fmt"
  8. // "log"
  9. "sync"
  10. "time"
  11. "wartank/pkg/components/parsetime"
  12. "wartank/pkg/components/safebool"
  13. "wartank/pkg/components/safeint"
  14. "wartank/pkg/types"
  15. )
  16. const (
  17. sleepInterval = time.Millisecond * 100 // Малый интервал сна в 100 мсек
  18. )
  19. // CountTime -- счётчик обратного времени для игровой зоны (анга, база, битва и т.п.)
  20. type CountTime struct {
  21. zone types.IZone
  22. val *safeint.SafeInt
  23. parser *parsetime.ParseTime
  24. chTick chan int // Канал секундных интервалов сна
  25. chCall chan int // Канал для отправки сигналов
  26. isWork *safebool.SafeBool // Признак работы
  27. timeTarget *safeint.SafeInt // Целевое время срабатывания
  28. ctx context.Context // Контекст для счётчика времени
  29. fnCancel func() // Функция отмены контекста для счётчика времени
  30. block sync.RWMutex
  31. }
  32. // NewCountTime -- возвращает новый *CountTime
  33. func NewCountTime(zone types.IZone) *CountTime {
  34. if zone == nil {
  35. panic("NewCountTime(): IZone == nil")
  36. }
  37. ctx, fnCancel := context.WithCancel(zone.Ctx())
  38. sf := &CountTime{
  39. zone: zone,
  40. val: safeint.NewSafeInt(),
  41. chTick: make(chan int, 3),
  42. chCall: make(chan int, 2),
  43. isWork: safebool.NewSafeBool(),
  44. parser: parsetime.NewParseTime(),
  45. timeTarget: safeint.NewSafeInt(),
  46. ctx: ctx,
  47. fnCancel: fnCancel,
  48. }
  49. sf.isWork.Set()
  50. go sf.makeTick()
  51. go sf.run()
  52. return sf
  53. }
  54. // Запускает тикер для секундных интервалов
  55. func (sf *CountTime) makeTick() {
  56. defer func() {
  57. if !sf.isWork.Get() {
  58. return
  59. }
  60. sf.fnCancel()
  61. sf.isWork.Reset()
  62. close(sf.chTick)
  63. // log._rintf("CountTime.makeTick(): работа завершена")
  64. }()
  65. countSleep := 0 // Счётчик сна (10 периодов -- 1 секунда)
  66. for {
  67. select {
  68. case <-sf.ctx.Done(): // Отмена контекста бота
  69. // log._rintf("CountTime.makeTick(): отмена контекста бота\n")
  70. return
  71. default:
  72. if !sf.isWork.Get() {
  73. return
  74. }
  75. if countSleep >= 10 {
  76. sf.chTick <- 1
  77. countSleep = 0
  78. }
  79. countSleep++
  80. time.Sleep(sleepInterval)
  81. }
  82. }
  83. }
  84. // Главный цикл обратного отсчёта
  85. func (sf *CountTime) run() {
  86. for range sf.chTick {
  87. timeNow := time.Now().UTC().Unix()
  88. if sf.timeTarget.Get() > int(timeNow) {
  89. val := sf.timeTarget.Get() - int(timeNow)
  90. val -= 1
  91. sf.Set(val)
  92. continue
  93. }
  94. close(sf.chCall)
  95. sf.fnCancel()
  96. return
  97. }
  98. }
  99. // Stop -- останавливает работу cчётчика
  100. func (sf *CountTime) Stop() {
  101. sf.isWork.Reset()
  102. }
  103. // Get -- возвращает число оставшихся сек
  104. func (sf *CountTime) Get() int {
  105. return sf.val.Get()
  106. }
  107. // Parse -- устанавливает число оставшихся сек
  108. func (sf *CountTime) Parse(val string) error {
  109. sf.block.Lock()
  110. defer sf.block.Unlock()
  111. if val == "" {
  112. return fmt.Errorf("CountTime.Set(): val is empty")
  113. }
  114. sf.parser.Parse(val)
  115. _val := sf.parser.Hour().Get()*3600 + sf.parser.Min().Get()*60 + sf.parser.Min().Get()
  116. sf.val.Set(_val)
  117. _val = int(time.Now().UTC().Unix()) + sf.val.Get()
  118. sf.timeTarget.Set(_val)
  119. return nil
  120. }
  121. // Set -- устанавливает число оставшихся сек
  122. func (sf *CountTime) Set(val int) error {
  123. sf.block.Lock()
  124. defer sf.block.Unlock()
  125. if val < 0 {
  126. return fmt.Errorf("CountTime.Set(): val(%v)<0", val)
  127. }
  128. sf.val.Set(val)
  129. { // Обновить локальные счётчики
  130. if val < 60 {
  131. sf.parser.Hour().Reset()
  132. sf.parser.Min().Reset()
  133. sf.parser.Sec().Set(val)
  134. return nil
  135. }
  136. if 60 < val && val < 3600 {
  137. sf.parser.Hour().Reset()
  138. iMin := val / 60
  139. sf.parser.Min().Set(iMin)
  140. val -= iMin * 60
  141. sf.parser.Sec().Set(val)
  142. return nil
  143. }
  144. sf.parser.Hour().Set(val / 3600)
  145. val -= sf.parser.Hour().Get() * 3600
  146. sf.parser.Min().Set(val / 60)
  147. val -= sf.parser.Min().Get() * 60
  148. sf.parser.Sec().Set(val)
  149. val = int(time.Now().UTC().Unix()) + sf.val.Get()
  150. sf.timeTarget.Set(val)
  151. }
  152. return nil
  153. }
  154. // String -- возвращает строковое представление оставшихся сек
  155. func (sf *CountTime) String() string {
  156. sf.block.RLock()
  157. defer sf.block.RUnlock()
  158. return sf.parser.String()
  159. }
  160. // ChanSig -- возвращает канал чтения тиков
  161. func (sf *CountTime) ChanSig() <-chan int {
  162. return sf.chCall
  163. }