You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.1 KiB
87 lines
2.1 KiB
// Copyright 2017 PingCAP, Inc.
|
|
//
|
|
// 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,
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package owner
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
|
|
"github.com/pingcap/errors"
|
|
)
|
|
|
|
var _ Manager = &mockManager{}
|
|
|
|
// mockManager represents the structure which is used for electing owner.
|
|
// It's used for local store and testing.
|
|
// So this worker will always be the owner.
|
|
type mockManager struct {
|
|
owner int32
|
|
id string // id is the ID of manager.
|
|
cancel context.CancelFunc
|
|
}
|
|
|
|
// NewMockManager creates a new mock Manager.
|
|
func NewMockManager(ctx context.Context, id string) Manager {
|
|
_, cancelFunc := context.WithCancel(ctx)
|
|
return &mockManager{
|
|
id: id,
|
|
cancel: cancelFunc,
|
|
}
|
|
}
|
|
|
|
// ID implements Manager.ID interface.
|
|
func (m *mockManager) ID() string {
|
|
return m.id
|
|
}
|
|
|
|
// IsOwner implements Manager.IsOwner interface.
|
|
func (m *mockManager) IsOwner() bool {
|
|
return atomic.LoadInt32(&m.owner) == 1
|
|
}
|
|
|
|
func (m *mockManager) toBeOwner() {
|
|
atomic.StoreInt32(&m.owner, 1)
|
|
}
|
|
|
|
// RetireOwner implements Manager.RetireOwner interface.
|
|
func (m *mockManager) RetireOwner() {
|
|
atomic.StoreInt32(&m.owner, 0)
|
|
}
|
|
|
|
// Cancel implements Manager.Cancel interface.
|
|
func (m *mockManager) Cancel() {
|
|
m.cancel()
|
|
}
|
|
|
|
// GetOwnerID implements Manager.GetOwnerID interface.
|
|
func (m *mockManager) GetOwnerID(ctx context.Context) (string, error) {
|
|
if m.IsOwner() {
|
|
return m.ID(), nil
|
|
}
|
|
return "", errors.New("no owner")
|
|
}
|
|
|
|
// CampaignOwner implements Manager.CampaignOwner interface.
|
|
func (m *mockManager) CampaignOwner() error {
|
|
m.toBeOwner()
|
|
return nil
|
|
}
|
|
|
|
// ResignOwner lets the owner start a new election.
|
|
func (m *mockManager) ResignOwner(ctx context.Context) error {
|
|
if m.IsOwner() {
|
|
m.RetireOwner()
|
|
}
|
|
return nil
|
|
}
|
|
|