astra
/
plc-mirror
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: s/LastSuccess/LastCompletion/ and use it as orignally intended

main^2
Max Ignatenko 2024-11-14 12:44:42 +00:00
parent 04ab6167c7
commit bad22e6422
2 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
cmd/plc-mirror/mirror.go
View File

@ -36,9 +36,9 @@ type Mirror struct {
upstream *url.URL
limiter *rate.Limiter
mu sync.RWMutex
lastSuccessTimestamp time.Time
lastRecordTimestamp time.Time
mu sync.RWMutex
lastCompletionTimestamp time.Time
lastRecordTimestamp time.Time
}
func NewMirror(ctx context.Context, upstream string, db *gorm.DB) (*Mirror, error) {
@ -78,7 +78,7 @@ func (m *Mirror) run(ctx context.Context) {
} else {
now := time.Now()
m.mu.Lock()
m.lastSuccessTimestamp = now
m.lastCompletionTimestamp = now
m.mu.Unlock()
}
time.Sleep(10 * time.Second)
@ -86,10 +86,10 @@ func (m *Mirror) run(ctx context.Context) {
}
}
func (m *Mirror) LastSuccess() time.Time {
func (m *Mirror) LastCompletion() time.Time {
m.mu.RLock()
defer m.mu.RUnlock()
return m.lastSuccessTimestamp
return m.lastCompletionTimestamp
}
func (m *Mirror) LastRecordTimestamp(ctx context.Context) (time.Time, error) {
@ -201,12 +201,11 @@ func (m *Mirror) runOnce(ctx context.Context) error {
return fmt.Errorf("inserting log entry into database: %w", err)
}
m.mu.Lock()
m.lastSuccessTimestamp = time.Now()
if !lastTimestamp.IsZero() {
m.mu.Lock()
m.lastRecordTimestamp = lastTimestamp
m.mu.Unlock()
}
m.mu.Unlock()
log.Info().Msgf("Got %d log entries. New cursor: %q", len(newEntries), cursor)
}

11
cmd/plc-mirror/serve.go
View File

@ -66,14 +66,21 @@ func (s *Server) serve(ctx context.Context, req *http.Request) convreq.HttpRespo
requestLatency.WithLabelValues(fmt.Sprint(c)).Observe(float64(time.Now().Sub(start)) / float64(time.Millisecond))
}
// Check if the mirror is up to date.
ts, err := s.mirror.LastRecordTimestamp(ctx)
if err != nil {
return respond.InternalServerError(err.Error())
}
delay := time.Since(ts)
if delay > s.MaxDelay {
updateMetrics(http.StatusServiceUnavailable)
return respond.ServiceUnavailable(fmt.Sprintf("mirror is %s behind", delay))
// Check LastCompletion and if it's recent enough - that means
// that we're actually caught up and there simply aren't any recent
// PLC operations.
completionDelay := time.Since(s.mirror.LastCompletion())
if completionDelay > s.MaxDelay {
updateMetrics(http.StatusServiceUnavailable)
return respond.ServiceUnavailable(fmt.Sprintf("mirror is %s behind", delay))
}
}
log := zerolog.Ctx(ctx)