mirror of https://github.com/k3s-io/k3s
41 lines
966 B
Go
41 lines
966 B
Go
|
package dqlite
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/canonical/go-dqlite/client"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (d *DQLite) Reset(ctx context.Context) error {
|
||
|
|
dqClient, err := client.New(ctx, d.getBindAddress(), client.WithLogFunc(log()))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
current, err := dqClient.Cluster(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// There's a chance our ID and the ID the server has doesn't match so find the ID
|
||
|
|
var surviving []client.NodeInfo
|
||
|
|
for _, testNode := range current {
|
||
|
|
if testNode.Address == d.NodeInfo.Address && testNode.ID == d.NodeInfo.ID {
|
||
|
|
surviving = append(surviving, testNode)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if err := dqClient.Remove(ctx, testNode.ID); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(surviving) != 1 {
|
||
|
|
return fmt.Errorf("failed to find %s in the current node, can not reset", d.NodeInfo.Address)
|
||
|
|
}
|
||
|
|
|
||
|
|
logrus.Infof("Resetting cluster to single master, please rejoin members")
|
||
|
|
return d.node.Recover(surviving)
|
||
|
|
}
|