1c/v8-1c-cluster-pde
1
0
Fork 0
mirror of https://github.com/Chipazawra/v8-1c-cluster-pde.git synced 2024-12-27 01:34:09 +02:00
Code Issues Releases Activity

refactor pusher/puler

Browse Source
This commit is contained in:
Anton 2021-12-30 14:53:59 +03:00
parent 95ce08a734
commit 492a280cf6
2 changed files with 19 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
internal/app/app.go
View File

@ -96,16 +96,16 @@ func Run() error {
sigchan := make(chan os.Signal, 1)
defer close(sigchan)
errchan := make(chan error, 1)
errchan := make(chan error)
defer close(errchan)
signal.Notify(sigchan, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
switch conf.MODE {
case push:
go RunPusher_(ctx, errchan, rcli)
go RunPusher(ctx, errchan, rcli)
case pull:
go RunPuller_(ctx, errchan, rcli)
go RunPuller(ctx, errchan, rcli)
default:
{
cancel()
@ -121,37 +121,7 @@ func Run() error {
return <-errchan
}
func RunPuller(rasapi rascli.Api) error {
log.Printf("v8-1c-cluster-pde: runing in %v mode", conf.MODE)
promRegistry := prometheus.NewRegistry()
promRegistry.MustRegister(rpHostsCollector.New(rasapi))
http.Handle("/metrics",
promhttp.HandlerFor(promRegistry, promhttp.HandlerOpts{}),
)
log.Printf("v8-1c-cluster-pde: listen %v", fmt.Sprintf("%s:%s", "", conf.PULL_EXPOSE))
err := http.ListenAndServe(fmt.Sprintf("%s:%s", "", conf.PULL_EXPOSE), nil)
if err != nil {
return fmt.Errorf("app: %v", err)
}
return nil
}
func RunPusher(rasapi rascli.Api) error {
log.Printf("v8-1c-cluster-pde: runing in %v mode pushgateway %v\n",
conf.MODE, fmt.Sprintf("%s:%s", conf.PUSH_HOST, conf.PUSH_PORT))
return pusher.New(
rpHostsCollector.New(rasapi),
fmt.Sprintf("%s:%s", conf.PUSH_HOST, conf.PUSH_PORT),
pusher.WithInterval(500),
).Run(context.Background())
}
func RunPuller_(ctx context.Context, errchan chan<- error, rasapi rascli.Api) {
func RunPuller(ctx context.Context, errchan chan<- error, rasapi rascli.Api) {
log.Printf("v8-1c-cluster-pde: runing in %v mode", conf.MODE)
promRegistry := prometheus.NewRegistry()
@ -179,13 +149,13 @@ func RunPuller_(ctx context.Context, errchan chan<- error, rasapi rascli.Api) {
log.Printf("v8-1c-cluster-pde: server shutdown")
}
func RunPusher_(ctx context.Context, errchan chan<- error, rasapi rascli.Api) {
func RunPusher(ctx context.Context, errchan chan<- error, rasapi rascli.Api) {
log.Printf("v8-1c-cluster-pde: runing in %v mode pushgateway %v\n",
conf.MODE, fmt.Sprintf("%s:%s", conf.PUSH_HOST, conf.PUSH_PORT))
errchan <- pusher.New(
go pusher.New(
rpHostsCollector.New(rasapi),
fmt.Sprintf("%s:%s", conf.PUSH_HOST, conf.PUSH_PORT),
pusher.WithInterval(500),
).Run(ctx)
).Run(ctx, errchan)
}

13
internal/pusher/pusher.go
View File

@ -55,28 +55,21 @@ func New(collector prometheus.Collector, url string, opts ...PusherOption) *Push
return p
}
func (p *Pusher) Run(ctx context.Context) error {
func (p *Pusher) Run(ctx context.Context, errchan chan<- error) {
ticker := time.NewTicker(time.Duration(p.intervalMillis * int(time.Microsecond)))
done := make(chan error)
go func(done chan error) {
Loop:
for {
select {
case <-ticker.C:
err := p.pusher.Push()
if err != nil {
done <- fmt.Errorf("puser: %v", err)
errchan <- fmt.Errorf("puser: %v", err)
break Loop
}
case <-ctx.Done():
log.Println("INFO: pusher context complete")
done <- nil
log.Println("INFO: pusher context done")
break Loop
}
}
close(done)
}(done)
return <-done
}