Add listener to engine

This commit is contained in:
херетик
2023-05-11 08:49:44 +01:00
parent 1d22a2aa20
commit cba46e7298
3 changed files with 18 additions and 9 deletions

View File

@@ -23,6 +23,7 @@ func TestEngine_Dispatcher(t *testing.T) {
var listeners []*transport.Listener
var keys []*crypto.Keys
var nodes []*node.Node
var engines []*Engine
var seed string
for i := 0; i < nTotal; i++ {
var k *crypto.Keys
@@ -57,14 +58,20 @@ func TestEngine_Dispatcher(t *testing.T) {
if addr, e = netip.ParseAddrPort(ip + ":" + port); fails(e) {
t.FailNow()
}
log.D.Ln(transport.GetHostOnlyAddress(l.Host))
var nod *node.Node
if nod, _ = node.NewNode(&addr, k, nil, 50000); fails(e) {
t.FailNow()
}
log.D.S("node", nod)
nodes = append(nodes, nod)
var eng *Engine
if eng, e = NewEngine(Params{
Listener: l,
ID: k,
Node: nod,
}); fails(e) {
t.FailNow()
}
engines = append(engines, eng)
}
time.Sleep(time.Second * 2)
cancel()

View File

@@ -3,6 +3,7 @@ package engine
import (
"go.uber.org/atomic"
"git-indra.lan/indra-labs/indra/pkg/engine/transport"
"git-indra.lan/indra-labs/indra/pkg/util/qu"
"git-indra.lan/indra-labs/indra/pkg/util/splice"
@@ -25,6 +26,7 @@ import (
type Engine struct {
Responses *responses.Pending
Manager *sess.Manager
Listener *transport.Listener
h *onions.Hidden
KeySet *crypto.KeySet
Load atomic.Uint32
@@ -34,6 +36,7 @@ type Engine struct {
type Params struct {
tpt.Transport
Listener *transport.Listener
ID *crypto.Keys
Node *node.Node
Nodes []*node.Node

View File

@@ -38,12 +38,11 @@ func createNMockCircuits(inclSessions bool, nCircuits int,
addr := slice.GenerateRandomAddrPortIPv4()
nodes[i], _ = node.NewNode(addr, id, tpts[i], 50000)
if cl[i], e = NewEngine(Params{
tpts[i],
id,
nodes[i],
nil,
nReturnSessions},
); fails(e) {
Transport: tpts[i],
ID: id,
Node: nodes[i],
NReturnSessions: nReturnSessions,
}); fails(e) {
return
}
cl[i].Manager.SetLocalNodeAddress(nodes[i].AddrPort)