Skip to content

Commit

Permalink
add gossip.ping
Browse files Browse the repository at this point in the history
  • Loading branch information
cryptix committed Mar 25, 2021
1 parent 7a80875 commit 991a9d8
Show file tree
Hide file tree
Showing 2 changed files with 72 additions and 1 deletion.
68 changes: 68 additions & 0 deletions muxrpc/handlers/gossip/ping.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
package gossip

import (
"context"
"encoding/json"
"io"
"time"

"go.cryptoscope.co/muxrpc/v2"
"go.mindeco.de/encodedTime"
)

func Ping(ctx context.Context, req *muxrpc.Request, peerSrc *muxrpc.ByteSource, peerSnk *muxrpc.ByteSink) error {
type arg struct {
Timeout int
}

var args []arg
err := json.Unmarshal(req.RawArgs, &args)
if err != nil {
return err
}

var timeout = time.Minute * 5
if len(args) == 1 {
timeout = time.Minute * time.Duration(args[0].Timeout/(60*1000))
}

ctx, cancel := context.WithTimeout(ctx, timeout)
defer cancel()

go func() {
peerSnk.SetEncoding(muxrpc.TypeJSON)
enc := json.NewEncoder(peerSnk)

tick := time.NewTicker(time.Second)
defer tick.Stop()

for {
select {
case <-ctx.Done():
return
case <-tick.C:
}

var pong = encodedTime.Millisecs(time.Now())
err = enc.Encode(pong)
if err != nil {
return
}
}
}()

for peerSrc.Next(ctx) {
var ping encodedTime.Millisecs
err := peerSrc.Reader(func(rd io.Reader) error {
return json.NewDecoder(rd).Decode(&ping)
})
if err != nil {
return err
}

// when := time.Time(ping)
// fmt.Printf("got ping: %s - age: %s\n", when.String(), time.Since(when))
}

return nil
}
5 changes: 4 additions & 1 deletion roomsrv/init_handlers.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,12 @@ package roomsrv

import (
kitlog "github.com/go-kit/kit/log"
"github.com/ssb-ngi-pointer/go-ssb-room/muxrpc/handlers/signinwithssb"
muxrpc "go.cryptoscope.co/muxrpc/v2"
"go.cryptoscope.co/muxrpc/v2/typemux"

"github.com/ssb-ngi-pointer/go-ssb-room/muxrpc/handlers/alias"
"github.com/ssb-ngi-pointer/go-ssb-room/muxrpc/handlers/gossip"
"github.com/ssb-ngi-pointer/go-ssb-room/muxrpc/handlers/signinwithssb"
"github.com/ssb-ngi-pointer/go-ssb-room/muxrpc/handlers/tunnel/server"
"github.com/ssb-ngi-pointer/go-ssb-room/muxrpc/handlers/whoami"
)
Expand Down Expand Up @@ -61,5 +62,7 @@ func (s *Server) initHandlers() {
mux.RegisterAsync(append(method, "invalidateAllSolutions"), typemux.AsyncFunc(siwssbHandler.InvalidateAllSolutions))
mux.RegisterAsync(append(method, "sendSolution"), typemux.AsyncFunc(siwssbHandler.SendSolution))

method = muxrpc.Method{"gossip"}
mux.RegisterDuplex(append(method, "ping"), typemux.DuplexFunc(gossip.Ping))
}
}

0 comments on commit 991a9d8

Please sign in to comment.