Use round-robin strategy while assigning workers

This commit is contained in:
Murat Kabilov 2017-10-09 16:56:27 +02:00 committed by GitHub
parent d6393d46cb
commit 32aa7270e6
3 changed files with 19 additions and 4 deletions

View File

@ -30,6 +30,8 @@ type Controller struct {
stopCh chan struct{}
curWorkerID uint32 //initialized with 0
clusterWorkers map[spec.NamespacedName]uint32
clustersMu sync.RWMutex
clusters map[spec.NamespacedName]*cluster.Cluster
clusterLogs map[spec.NamespacedName]ringlog.RingLogger
@ -54,6 +56,7 @@ func NewController(controllerConfig *spec.ControllerConfig) *Controller {
config: *controllerConfig,
opConfig: &config.Config{},
logger: logger.WithField("pkg", "controller"),
clusterWorkers: make(map[spec.NamespacedName]uint32),
clusters: make(map[spec.NamespacedName]*cluster.Cluster),
clusterLogs: make(map[spec.NamespacedName]ringlog.RingLogger),
clusterHistory: make(map[spec.NamespacedName]ringlog.RingLogger),

View File

@ -2,7 +2,6 @@ package controller
import (
"fmt"
"hash/crc32"
apiextv1beta1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1beta1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@ -30,7 +29,20 @@ func (c *Controller) makeClusterConfig() cluster.Config {
}
func (c *Controller) clusterWorkerID(clusterName spec.NamespacedName) uint32 {
return crc32.ChecksumIEEE([]byte(clusterName.String())) % c.opConfig.Workers
workerId, ok := c.clusterWorkers[clusterName]
if ok {
return workerId
}
c.clusterWorkers[clusterName] = c.curWorkerID
if c.curWorkerID == c.opConfig.Workers-1 {
c.curWorkerID = 0
} else {
c.curWorkerID++
}
return c.clusterWorkers[clusterName]
}
func (c *Controller) createCRD() error {

View File

@ -95,11 +95,11 @@ func TestClusterWorkerID(t *testing.T) {
}{
{
in: spec.NamespacedName{Namespace: "foo", Name: "bar"},
expected: 2,
expected: 0,
},
{
in: spec.NamespacedName{Namespace: "default", Name: "testcluster"},
expected: 3,
expected: 1,
},
}
for _, test := range testTable {