mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-31 15:25:57 +00:00
Move Reconciler into reconciler.go
This commit is contained in:
Dr. Stefan Schimanski
parent
2c00037883
commit
eea8accfd5
@ -731,160 +731,6 @@ func (k *MesosScheduler) explicitlyReconcileTasks(driver bindings.SchedulerDrive
|
||||
return nil
|
||||
}
|
||||
|
||||
var (
|
||||
reconciliationCancelledErr = fmt.Errorf("explicit task reconciliation cancelled")
|
||||
)
|
||||
|
||||
type ReconcilerAction func(driver bindings.SchedulerDriver, cancel <-chan struct{}) <-chan error
|
||||
|
||||
type Reconciler struct {
|
||||
proc.Doer
|
||||
Action ReconcilerAction
|
||||
explicit chan struct{} // send an empty struct to trigger explicit reconciliation
|
||||
implicit chan struct{} // send an empty struct to trigger implicit reconciliation
|
||||
done <-chan struct{} // close this when you want the reconciler to exit
|
||||
cooldown time.Duration
|
||||
explicitReconciliationAbortTimeout time.Duration
|
||||
}
|
||||
|
||||
func newReconciler(doer proc.Doer, action ReconcilerAction,
|
||||
cooldown, explicitReconciliationAbortTimeout time.Duration, done <-chan struct{}) *Reconciler {
|
||||
return &Reconciler{
|
||||
Doer: doer,
|
||||
explicit: make(chan struct{}, 1),
|
||||
implicit: make(chan struct{}, 1),
|
||||
cooldown: cooldown,
|
||||
explicitReconciliationAbortTimeout: explicitReconciliationAbortTimeout,
|
||||
done: done,
|
||||
Action: func(driver bindings.SchedulerDriver, cancel <-chan struct{}) <-chan error {
|
||||
// trigged the reconciler action in the doer's execution context,
|
||||
// but it could take a while and the scheduler needs to be able to
|
||||
// process updates, the callbacks for which ALSO execute in the SAME
|
||||
// deferred execution context -- so the action MUST be executed async.
|
||||
errOnce := proc.NewErrorOnce(cancel)
|
||||
return errOnce.Send(doer.Do(func() {
|
||||
// only triggers the action if we're the currently elected,
|
||||
// registered master and runs the action async.
|
||||
go func() {
|
||||
var err <-chan error
|
||||
defer errOnce.Send(err)
|
||||
err = action(driver, cancel)
|
||||
}()
|
||||
})).Err()
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Reconciler) RequestExplicit() {
|
||||
select {
|
||||
case r.explicit <- struct{}{}: // noop
|
||||
default: // request queue full; noop
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Reconciler) RequestImplicit() {
|
||||
select {
|
||||
case r.implicit <- struct{}{}: // noop
|
||||
default: // request queue full; noop
|
||||
}
|
||||
}
|
||||
|
||||
// execute task reconciliation, returns when r.done is closed. intended to run as a goroutine.
|
||||
// if reconciliation is requested while another is in progress, the in-progress operation will be
|
||||
// cancelled before the new reconciliation operation begins.
|
||||
func (r *Reconciler) Run(driver bindings.SchedulerDriver) {
|
||||
var cancel, finished chan struct{}
|
||||
requestLoop:
|
||||
for {
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
default: // proceed
|
||||
}
|
||||
select {
|
||||
case <-r.implicit:
|
||||
metrics.ReconciliationRequested.WithLabelValues("implicit").Inc()
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
case <-r.explicit:
|
||||
break // give preference to a pending request for explicit
|
||||
default: // continue
|
||||
// don't run implicit reconciliation while explicit is ongoing
|
||||
if finished != nil {
|
||||
select {
|
||||
case <-finished: // continue w/ implicit
|
||||
default:
|
||||
log.Infoln("skipping implicit reconcile because explicit reconcile is ongoing")
|
||||
continue requestLoop
|
||||
}
|
||||
}
|
||||
errOnce := proc.NewErrorOnce(r.done)
|
||||
errCh := r.Do(func() {
|
||||
var err error
|
||||
defer errOnce.Report(err)
|
||||
log.Infoln("implicit reconcile tasks")
|
||||
metrics.ReconciliationExecuted.WithLabelValues("implicit").Inc()
|
||||
if _, err = driver.ReconcileTasks([]*mesos.TaskStatus{}); err != nil {
|
||||
log.V(1).Infof("failed to request implicit reconciliation from mesos: %v", err)
|
||||
}
|
||||
})
|
||||
proc.OnError(errOnce.Send(errCh).Err(), func(err error) {
|
||||
log.Errorf("failed to run implicit reconciliation: %v", err)
|
||||
}, r.done)
|
||||
goto slowdown
|
||||
}
|
||||
case <-r.done:
|
||||
return
|
||||
case <-r.explicit: // continue
|
||||
metrics.ReconciliationRequested.WithLabelValues("explicit").Inc()
|
||||
}
|
||||
|
||||
if cancel != nil {
|
||||
close(cancel)
|
||||
cancel = nil
|
||||
|
||||
// play nice and wait for the prior operation to finish, complain
|
||||
// if it doesn't
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
case <-finished: // noop, expected
|
||||
case <-time.After(r.explicitReconciliationAbortTimeout): // very unexpected
|
||||
log.Error("reconciler action failed to stop upon cancellation")
|
||||
}
|
||||
}
|
||||
// copy 'finished' to 'fin' here in case we end up with simultaneous go-routines,
|
||||
// if cancellation takes too long or fails - we don't want to close the same chan
|
||||
// more than once
|
||||
cancel = make(chan struct{})
|
||||
finished = make(chan struct{})
|
||||
go func(fin chan struct{}) {
|
||||
startedAt := time.Now()
|
||||
defer func() {
|
||||
metrics.ReconciliationLatency.Observe(metrics.InMicroseconds(time.Since(startedAt)))
|
||||
}()
|
||||
|
||||
metrics.ReconciliationExecuted.WithLabelValues("explicit").Inc()
|
||||
defer close(fin)
|
||||
err := <-r.Action(driver, cancel)
|
||||
if err == reconciliationCancelledErr {
|
||||
metrics.ReconciliationCancelled.WithLabelValues("explicit").Inc()
|
||||
log.Infoln(err.Error())
|
||||
} else if err != nil {
|
||||
log.Errorf("reconciler action failed: %v", err)
|
||||
}
|
||||
}(finished)
|
||||
slowdown:
|
||||
// don't allow reconciliation to run very frequently, either explicit or implicit
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
case <-time.After(r.cooldown): // noop
|
||||
}
|
||||
} // for
|
||||
}
|
||||
|
||||
func (ks *MesosScheduler) recoverTasks() error {
|
||||
podList, err := ks.client.Pods(api.NamespaceAll).List(labels.Everything(), fields.Everything())
|
||||
if err != nil {
|
||||
|
||||
@ -13,6 +13,7 @@ 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 scheduler
|
||||
|
||||
import (
|
||||
|
||||
182
contrib/mesos/pkg/scheduler/reconciler.go
Normal file
182
contrib/mesos/pkg/scheduler/reconciler.go
Normal file
@ -0,0 +1,182 @@
|
||||
/*
|
||||
Copyright 2015 The Kubernetes Authors All rights reserved.
|
||||
|
||||
Licensed 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 scheduler
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
log "github.com/golang/glog"
|
||||
mesos "github.com/mesos/mesos-go/mesosproto"
|
||||
bindings "github.com/mesos/mesos-go/scheduler"
|
||||
"k8s.io/kubernetes/contrib/mesos/pkg/proc"
|
||||
"k8s.io/kubernetes/contrib/mesos/pkg/scheduler/metrics"
|
||||
)
|
||||
|
||||
var (
|
||||
reconciliationCancelledErr = fmt.Errorf("explicit task reconciliation cancelled")
|
||||
)
|
||||
|
||||
type ReconcilerAction func(driver bindings.SchedulerDriver, cancel <-chan struct{}) <-chan error
|
||||
|
||||
type Reconciler struct {
|
||||
proc.Doer
|
||||
Action ReconcilerAction
|
||||
explicit chan struct{} // send an empty struct to trigger explicit reconciliation
|
||||
implicit chan struct{} // send an empty struct to trigger implicit reconciliation
|
||||
done <-chan struct{} // close this when you want the reconciler to exit
|
||||
cooldown time.Duration
|
||||
explicitReconciliationAbortTimeout time.Duration
|
||||
}
|
||||
|
||||
func newReconciler(doer proc.Doer, action ReconcilerAction,
|
||||
cooldown, explicitReconciliationAbortTimeout time.Duration, done <-chan struct{}) *Reconciler {
|
||||
return &Reconciler{
|
||||
Doer: doer,
|
||||
explicit: make(chan struct{}, 1),
|
||||
implicit: make(chan struct{}, 1),
|
||||
cooldown: cooldown,
|
||||
explicitReconciliationAbortTimeout: explicitReconciliationAbortTimeout,
|
||||
done: done,
|
||||
Action: func(driver bindings.SchedulerDriver, cancel <-chan struct{}) <-chan error {
|
||||
// trigged the reconciler action in the doer's execution context,
|
||||
// but it could take a while and the scheduler needs to be able to
|
||||
// process updates, the callbacks for which ALSO execute in the SAME
|
||||
// deferred execution context -- so the action MUST be executed async.
|
||||
errOnce := proc.NewErrorOnce(cancel)
|
||||
return errOnce.Send(doer.Do(func() {
|
||||
// only triggers the action if we're the currently elected,
|
||||
// registered master and runs the action async.
|
||||
go func() {
|
||||
var err <-chan error
|
||||
defer errOnce.Send(err)
|
||||
err = action(driver, cancel)
|
||||
}()
|
||||
})).Err()
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Reconciler) RequestExplicit() {
|
||||
select {
|
||||
case r.explicit <- struct{}{}: // noop
|
||||
default: // request queue full; noop
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Reconciler) RequestImplicit() {
|
||||
select {
|
||||
case r.implicit <- struct{}{}: // noop
|
||||
default: // request queue full; noop
|
||||
}
|
||||
}
|
||||
|
||||
// execute task reconciliation, returns when r.done is closed. intended to run as a goroutine.
|
||||
// if reconciliation is requested while another is in progress, the in-progress operation will be
|
||||
// cancelled before the new reconciliation operation begins.
|
||||
func (r *Reconciler) Run(driver bindings.SchedulerDriver) {
|
||||
var cancel, finished chan struct{}
|
||||
requestLoop:
|
||||
for {
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
default: // proceed
|
||||
}
|
||||
select {
|
||||
case <-r.implicit:
|
||||
metrics.ReconciliationRequested.WithLabelValues("implicit").Inc()
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
case <-r.explicit:
|
||||
break // give preference to a pending request for explicit
|
||||
default: // continue
|
||||
// don't run implicit reconciliation while explicit is ongoing
|
||||
if finished != nil {
|
||||
select {
|
||||
case <-finished: // continue w/ implicit
|
||||
default:
|
||||
log.Infoln("skipping implicit reconcile because explicit reconcile is ongoing")
|
||||
continue requestLoop
|
||||
}
|
||||
}
|
||||
errOnce := proc.NewErrorOnce(r.done)
|
||||
errCh := r.Do(func() {
|
||||
var err error
|
||||
defer errOnce.Report(err)
|
||||
log.Infoln("implicit reconcile tasks")
|
||||
metrics.ReconciliationExecuted.WithLabelValues("implicit").Inc()
|
||||
if _, err = driver.ReconcileTasks([]*mesos.TaskStatus{}); err != nil {
|
||||
log.V(1).Infof("failed to request implicit reconciliation from mesos: %v", err)
|
||||
}
|
||||
})
|
||||
proc.OnError(errOnce.Send(errCh).Err(), func(err error) {
|
||||
log.Errorf("failed to run implicit reconciliation: %v", err)
|
||||
}, r.done)
|
||||
goto slowdown
|
||||
}
|
||||
case <-r.done:
|
||||
return
|
||||
case <-r.explicit: // continue
|
||||
metrics.ReconciliationRequested.WithLabelValues("explicit").Inc()
|
||||
}
|
||||
|
||||
if cancel != nil {
|
||||
close(cancel)
|
||||
cancel = nil
|
||||
|
||||
// play nice and wait for the prior operation to finish, complain
|
||||
// if it doesn't
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
case <-finished: // noop, expected
|
||||
case <-time.After(r.explicitReconciliationAbortTimeout): // very unexpected
|
||||
log.Error("reconciler action failed to stop upon cancellation")
|
||||
}
|
||||
}
|
||||
// copy 'finished' to 'fin' here in case we end up with simultaneous go-routines,
|
||||
// if cancellation takes too long or fails - we don't want to close the same chan
|
||||
// more than once
|
||||
cancel = make(chan struct{})
|
||||
finished = make(chan struct{})
|
||||
go func(fin chan struct{}) {
|
||||
startedAt := time.Now()
|
||||
defer func() {
|
||||
metrics.ReconciliationLatency.Observe(metrics.InMicroseconds(time.Since(startedAt)))
|
||||
}()
|
||||
|
||||
metrics.ReconciliationExecuted.WithLabelValues("explicit").Inc()
|
||||
defer close(fin)
|
||||
err := <-r.Action(driver, cancel)
|
||||
if err == reconciliationCancelledErr {
|
||||
metrics.ReconciliationCancelled.WithLabelValues("explicit").Inc()
|
||||
log.Infoln(err.Error())
|
||||
} else if err != nil {
|
||||
log.Errorf("reconciler action failed: %v", err)
|
||||
}
|
||||
}(finished)
|
||||
slowdown:
|
||||
// don't allow reconciliation to run very frequently, either explicit or implicit
|
||||
select {
|
||||
case <-r.done:
|
||||
return
|
||||
case <-time.After(r.cooldown): // noop
|
||||
}
|
||||
} // for
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user