cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

Merge pull request #10149 from FlattestWhite/richard/make-user-agent-dependency-injectable

Browse Source 
feat: Allow libp2p user agent to be overriden
This commit is contained in:
Łukasz Magiera 2023-02-01 19:12:08 +01:00 committed by GitHub
commit 7060150bdc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
node/builder.go
View File

@ -67,9 +67,10 @@ var (
ConnectionManagerKey = special{9} // Libp2p option ConnectionManagerKey = special{9} // Libp2p option
AutoNATSvcKey = special{10} // Libp2p option AutoNATSvcKey = special{10} // Libp2p option
BandwidthReporterKey = special{11} // Libp2p option BandwidthReporterKey = special{11} // Libp2p option
ConnGaterKey = special{12} // libp2p option ConnGaterKey = special{12} // Libp2p option
DAGStoreKey = special{13} // constructor returns multiple values DAGStoreKey = special{13} // constructor returns multiple values
ResourceManagerKey = special{14} // Libp2p option ResourceManagerKey = special{14} // Libp2p option
UserAgentKey = special{15} // Libp2p option
) )
type invoke int type invoke int

7
node/modules/lp2p/host.go
View File

@ -69,6 +69,13 @@ func Host(mctx helpers.MetricsCtx, lc fx.Lifecycle, params P2PHostIn) (RawHost,
return h, nil return h, nil
} }
func UserAgentOption(agent string) func() (opts Libp2pOpts, err error) {
return func() (opts Libp2pOpts, err error) {
opts.Opts = append(opts.Opts, libp2p.UserAgent(agent))
return
}
}
func MockHost(mn mocknet.Mocknet, id peer.ID, ps peerstore.Peerstore) (RawHost, error) { func MockHost(mn mocknet.Mocknet, id peer.ID, ps peerstore.Peerstore) (RawHost, error) {
return mn.AddPeerWithPeerstore(id, ps) return mn.AddPeerWithPeerstore(id, ps)
} }