youtubebeat/vendor/github.com/elastic/beats/filebeat/beater/channels.go

89 lines
2.2 KiB
Go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package beater
import (
"sync"
"github.com/elastic/beats/filebeat/input/file"
"github.com/elastic/beats/filebeat/registrar"
"github.com/elastic/beats/libbeat/monitoring"
)
type registrarLogger struct {
done chan struct{}
ch chan<- []file.State
}
type finishedLogger struct {
wg *eventCounter
}
type eventCounter struct {
added *monitoring.Uint
done *monitoring.Uint
count *monitoring.Int
wg sync.WaitGroup
}
func newRegistrarLogger(reg *registrar.Registrar) *registrarLogger {
return &registrarLogger{
done: make(chan struct{}),
ch: reg.Channel,
}
}
func (l *registrarLogger) Close() { close(l.done) }
func (l *registrarLogger) Published(states []file.State) {
select {
case <-l.done:
// set ch to nil, so no more events will be send after channel close signal
// has been processed the first time.
// Note: nil channels will block, so only done channel will be actively
// report 'closed'.
l.ch = nil
case l.ch <- states:
}
}
func newFinishedLogger(wg *eventCounter) *finishedLogger {
return &finishedLogger{wg}
}
func (l *finishedLogger) Published(n int) bool {
for i := 0; i < n; i++ {
l.wg.Done()
}
return true
}
func (c *eventCounter) Add(delta int) {
c.count.Add(int64(delta))
c.added.Add(uint64(delta))
c.wg.Add(delta)
}
func (c *eventCounter) Done() {
c.count.Dec()
c.done.Inc()
c.wg.Done()
}
func (c *eventCounter) Wait() {
c.wg.Wait()
}