tidb schedule_manager 源码
tidb schedule_manager 代码
文件路径：/domain/infosync/schedule_manager.go
// Copyright 2022 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,
// 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 infosync
import (
"bytes"
"context"
"encoding/json"
"path"
"sync"
"github.com/pingcap/errors"
"github.com/pingcap/tidb/util/pdapi"
clientv3 "go.etcd.io/etcd/client/v3"
)
// ScheduleManager manages schedule configs
type ScheduleManager interface {
GetPDScheduleConfig(ctx context.Context) (map[string]interface{}, error)
SetPDScheduleConfig(ctx context.Context, config map[string]interface{}) error
}
// PDScheduleManager manages schedule with pd
type PDScheduleManager struct {
etcdCli *clientv3.Client
}
// GetPDScheduleConfig get schedule config from pd
func (sm *PDScheduleManager) GetPDScheduleConfig(ctx context.Context) (map[string]interface{}, error) {
ret, err := doRequest(ctx, "GetPDSchedule", sm.etcdCli.Endpoints(), path.Join(pdapi.Config, "schedule"), "GET", nil)
if err != nil {
return nil, errors.Trace(err)
}
var schedule map[string]interface{}
if err = json.Unmarshal(ret, &schedule); err != nil {
return nil, errors.Trace(err)
}
return schedule, nil
}
// SetPDScheduleConfig set schedule config to pd
func (sm *PDScheduleManager) SetPDScheduleConfig(ctx context.Context, config map[string]interface{}) error {
configJSON, err := json.Marshal(config)
if err != nil {
return err
}
_, err = doRequest(ctx, "SetPDSchedule", sm.etcdCli.Endpoints(), path.Join(pdapi.Config, "schedule"), "POST", bytes.NewReader(configJSON))
if err != nil {
return errors.Trace(err)
}
return nil
}
type mockScheduleManager struct {
sync.RWMutex
schedules map[string]interface{}
}
// GetPDScheduleConfig get schedule config from schedules map
func (mm *mockScheduleManager) GetPDScheduleConfig(ctx context.Context) (map[string]interface{}, error) {
mm.Lock()
schedules := make(map[string]interface{})
for key, values := range mm.schedules {
schedules[key] = values
}
mm.Unlock()
return schedules, nil
}
// SetPDScheduleConfig set schedule config to schedules map
func (mm *mockScheduleManager) SetPDScheduleConfig(ctx context.Context, config map[string]interface{}) error {
mm.Lock()
if mm.schedules == nil {
mm.schedules = make(map[string]interface{})
}
for key, value := range config {
mm.schedules[key] = value
}
mm.Unlock()
return nil
}
相关信息
相关文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
6、 golang
-
8、 openharmony