local.go

package membership

import (
	"context"
	"sync/atomic"
	"time"

	"github.com/rakoo/raft/internal/raftpb"
	"github.com/rakoo/raft/raftlog"
	etcdraftpb "go.etcd.io/etcd/raft/v3/raftpb"
)

func newLocal(cfg Config, m raftpb.Member) (Member, error) {
	l := &local{
		r:      cfg.Reporter(),
		active: time.Now(),
	}
	_ = l.Update(m)
	return l, nil
}

// local represents the current cluster member.
type local struct {
	r      Reporter
	active time.Time
	raw    atomic.Value
}

func (l *local) ID() uint64 {
	return l.Raw().ID
}

func (l *local) Address() string {
	return l.Raw().Address
}

func (l *local) ActiveSince() time.Time {
	return l.active
}

func (l *local) IsActive() bool {
	return !l.active.IsZero()
}

func (l *local) Type() raftpb.MemberType {
	return l.Raw().Type
}

func (l *local) Update(m raftpb.Member) (err error) {
	l.raw.Store(m)
	return
}

func (l *local) Close() error {
	l.r.ReportShutdown(l.ID())
	return nil
}

func (l *local) Send(etcdraftpb.Message) error {
	raftlog.Panic("raft.membership: attempted to send msg to local member; should never happen")
	return nil
}

func (l *local) Raw() raftpb.Member {
	return l.raw.Load().(raftpb.Member)
}

func (l *local) TearDown(ctx context.Context) error {
	return l.Close()
}