86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package shardctrler
|
|
|
|
//
|
|
// Shardctrler with InitConfig, Query, and ChangeConfigTo methods
|
|
//
|
|
|
|
import (
|
|
|
|
"sync/atomic"
|
|
|
|
"6.5840/kvsrv1"
|
|
"6.5840/kvsrv1/rpc"
|
|
"6.5840/kvtest1"
|
|
"6.5840/shardkv1/shardcfg"
|
|
"6.5840/tester1"
|
|
)
|
|
|
|
|
|
// ShardCtrler for the controller and kv clerk.
|
|
type ShardCtrler struct {
|
|
clnt *tester.Clnt
|
|
kvtest.IKVClerk
|
|
|
|
killed int32 // set by Kill()
|
|
leases bool
|
|
|
|
// Your data here.
|
|
}
|
|
|
|
// Make a ShardCltler, which stores its state in a kvsrv.
|
|
func MakeShardCtrler(clnt *tester.Clnt, leases bool) *ShardCtrler {
|
|
sck := &ShardCtrler{clnt: clnt, leases: leases}
|
|
srv := tester.ServerName(tester.GRP0, 0)
|
|
sck.IKVClerk = kvsrv.MakeClerk(clnt, srv)
|
|
// Your code here.
|
|
return sck
|
|
}
|
|
|
|
// The tester calls InitController() before starting a new
|
|
// controller. In part A, this method doesn't need to do anything. In
|
|
// B and C, this method implements recovery (part B) and uses a lock
|
|
// to become leader (part C).
|
|
func (sck *ShardCtrler) InitController() {
|
|
}
|
|
|
|
// The tester calls ExitController to exit a controller. In part B and
|
|
// C, release lock.
|
|
func (sck *ShardCtrler) ExitController() {
|
|
}
|
|
|
|
// Called once by the tester to supply the first configuration. You
|
|
// can marshal ShardConfig into a string using shardcfg.String(), and
|
|
// then Put it in the kvsrv for the controller at version 0. You can
|
|
// pick the key to name the configuration.
|
|
func (sck *ShardCtrler) InitConfig(cfg *shardcfg.ShardConfig) {
|
|
// Your code here
|
|
}
|
|
|
|
// Called by the tester to ask the controller to change the
|
|
// configuration from the current one to new. While the controller
|
|
// changes the configuration it may be superseded by another
|
|
// controller.
|
|
func (sck *ShardCtrler) ChangeConfigTo(new *shardcfg.ShardConfig) {
|
|
return
|
|
}
|
|
|
|
// Tester "kills" shardctrler by calling Kill(). For your
|
|
// convenience, we also supply isKilled() method to test killed in
|
|
// loops.
|
|
func (sck *ShardCtrler) Kill() {
|
|
atomic.StoreInt32(&sck.killed, 1)
|
|
}
|
|
|
|
func (sck *ShardCtrler) isKilled() bool {
|
|
z := atomic.LoadInt32(&sck.killed)
|
|
return z == 1
|
|
}
|
|
|
|
|
|
// Return the current configuration and its version number
|
|
func (sck *ShardCtrler) Query() (*shardcfg.ShardConfig, rpc.Tversion) {
|
|
// Your code here.
|
|
return nil, 0
|
|
}
|
|
|