run stabilization

This commit is contained in:
Branden J Brown 2025-03-13 22:07:12 -04:00
parent 1e77429a91
commit 7a04a22f76
2 changed files with 33 additions and 13 deletions

View File

@ -24,8 +24,9 @@ type Node struct {
// New creates an instance of a Chord network that responds on HTTP.
// The listener must be the same one used for the HTTP server that routes to
// [Node.ServeHTTP]. It must be bound to a single interface and port.
func New(l net.Listener, cl chord.Client) (*Node, error) {
// [Node.ServeHTTP] and its address must match self.
// It must be bound to a single interface and port.
func New(l net.Listener, cl chord.Client, self *chord.Node) (*Node, error) {
addr, err := netip.ParseAddrPort(l.Addr().String())
if err != nil {
return nil, err
@ -33,12 +34,8 @@ func New(l net.Listener, cl chord.Client) (*Node, error) {
if addr.Addr().IsUnspecified() || addr.Port() == 0 {
return nil, errors.New("listener must be bound to a single interface and port")
}
n, err := chord.Create(addr)
if err != nil {
return nil, err
}
r := &Node{
self: n,
self: self,
client: cl,
}
return r, nil

35
main.go
View File

@ -14,6 +14,7 @@ import (
"github.com/urfave/cli/v3"
"git.sunturtle.xyz/zephyr/chord/chord"
"git.sunturtle.xyz/zephyr/chord/chord/httpnode"
)
@ -116,21 +117,21 @@ var (
}
)
func addrflag(ip string, p uint64) (string, error) {
func addrflag(ip string, p uint64) (netip.AddrPort, error) {
ap, err := netip.ParseAddrPort(ip)
if err == nil {
// note inverted error check
if p != 0 {
ap = netip.AddrPortFrom(ap.Addr(), uint16(p))
}
return ap.String(), nil
return ap, nil
}
a, err := netip.ParseAddr(ip)
if err == nil {
// note inverted error check
return netip.AddrPortFrom(a, uint16(p)).String(), nil
return netip.AddrPortFrom(a, uint16(p)), nil
}
return "", fmt.Errorf("couldn't parse %q as ip:port or ip", ip)
return netip.AddrPort{}, fmt.Errorf("couldn't parse %q as ip:port or ip", ip)
}
func cliJoin(ctx context.Context, cmd *cli.Command) error {
@ -138,13 +139,26 @@ func cliJoin(ctx context.Context, cmd *cli.Command) error {
if err != nil {
return err
}
l, err := net.Listen("tcp", addr)
l, err := net.Listen("tcp", addr.String())
if err != nil {
return err
}
defer l.Close()
cl := &httpnode.Client{HTTP: http.Client{Timeout: 5 * time.Second}}
s, err := httpnode.New(l, cl)
var node *chord.Node
if peer := cmd.String("c"); peer != "" {
p, err := netip.ParseAddrPort(peer)
if err != nil {
return err
}
node, err = chord.Join(ctx, cl, addr, chord.Address(p))
} else {
node, err = chord.Create(addr)
}
if err != nil {
return err
}
s, err := httpnode.New(l, cl, node)
if err != nil {
return err
}
@ -161,7 +175,16 @@ func cliJoin(ctx context.Context, cmd *cli.Command) error {
}
slog.ErrorContext(ctx, "HTTP API server closed", slog.Any("err", err))
}()
t := time.NewTicker(time.Second)
go func() {
for range t.C {
if err := chord.Stabilize(ctx, cl, node); err != nil {
slog.ErrorContext(ctx, "stabilize", slog.Any("err", err))
}
}
}()
<-ctx.Done()
t.Stop()
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
return srv.Shutdown(ctx)