You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.7 KiB
74 lines
1.7 KiB
package database
|
|
|
|
import (
|
|
"errors"
|
|
"io/fs"
|
|
"os"
|
|
"time"
|
|
|
|
"gordenko.dev/dima/diploma"
|
|
"gordenko.dev/dima/diploma/proto"
|
|
)
|
|
|
|
func timeBoundsOfAggregation(since, until proto.TimeBound, groupBy diploma.GroupBy, firstHourOfDay int) (s time.Time, u time.Time) {
|
|
switch groupBy {
|
|
case diploma.GroupByHour, diploma.GroupByDay:
|
|
s = time.Date(since.Year, since.Month, since.Day, 0, 0, 0, 0, time.Local)
|
|
u = time.Date(until.Year, until.Month, until.Day, 0, 0, 0, 0, time.Local)
|
|
|
|
case diploma.GroupByMonth:
|
|
s = time.Date(since.Year, since.Month, 1, 0, 0, 0, 0, time.Local)
|
|
u = time.Date(until.Year, until.Month, 1, 0, 0, 0, 0, time.Local)
|
|
}
|
|
|
|
if firstHourOfDay > 0 {
|
|
duration := time.Duration(firstHourOfDay) * time.Hour
|
|
s = s.Add(duration)
|
|
u = u.Add(duration)
|
|
}
|
|
|
|
u = u.Add(-1 * time.Second)
|
|
return
|
|
}
|
|
|
|
func isFileExist(fileName string) (bool, error) {
|
|
_, err := os.Stat(fileName)
|
|
if err != nil {
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
return false, nil
|
|
} else {
|
|
return false, err
|
|
}
|
|
} else {
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
func (s *Database) appendJobToWorkerQueue(job any) {
|
|
s.mutex.Lock()
|
|
s.workerQueue = append(s.workerQueue, job)
|
|
s.mutex.Unlock()
|
|
|
|
select {
|
|
case s.workerSignalCh <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
|
|
func (s *Database) metricRUnlock(metricID uint32) {
|
|
s.mutex.Lock()
|
|
s.rLocksToRelease = append(s.rLocksToRelease, metricID)
|
|
s.mutex.Unlock()
|
|
|
|
select {
|
|
case s.workerSignalCh <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
|
|
func correctToFHD(since, until uint32, firstHourOfDay int) (uint32, uint32) {
|
|
duration := time.Duration(firstHourOfDay) * time.Hour
|
|
since = uint32(time.Unix(int64(since), 0).Add(duration).Unix())
|
|
until = uint32(time.Unix(int64(until), 0).Add(duration).Unix())
|
|
return since, until
|
|
}
|
|
|