logos-messaging/go-multiaddr
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-multiaddr.git synced 2026-01-04 05:53:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

unexport ProtocolsByName and add protocolsByCode

Browse Source
This commit is contained in:
Steven Allen 2018-06-22 18:39:50 -07:00
parent 47e3ab13b2
commit 6688376452
1 changed files with 21 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
protocols.go
View File

@ -81,16 +81,18 @@ var (
protoUNIX = Protocol{P_UNIX, LengthPrefixedVarSize, "unix", CodeToVarint(P_UNIX), true, TranscoderUnix} protoUNIX = Protocol{P_UNIX, LengthPrefixedVarSize, "unix", CodeToVarint(P_UNIX), true, TranscoderUnix}
) )
var ProtocolsByName = map[string]Protocol{} var protocolsByName = map[string]Protocol{}
var protocolsByCode = map[int]Protocol{}
func init() { func init() {
for _, p := range Protocols { for _, p := range Protocols {
ProtocolsByName[p.Name] = p protocolsByName[p.Name] = p
protocolsByCode[p.Code] = p
} }
// explicitly set both of these // explicitly set both of these
ProtocolsByName["p2p"] = protoP2P protocolsByName["p2p"] = protoP2P
ProtocolsByName["ipfs"] = protoP2P protocolsByName["ipfs"] = protoP2P
} }
// SwapToP2pMultiaddrs is a function to make the transition from /ipfs/... // SwapToP2pMultiaddrs is a function to make the transition from /ipfs/...
@ -115,44 +117,41 @@ func SwapToP2pMultiaddrs() {
protoP2P.Name = "p2p" protoP2P.Name = "p2p"
ProtocolsByName["ipfs"] = protoP2P protocolsByName["ipfs"] = protoP2P
ProtocolsByName["p2p"] = protoP2P protocolsByName["p2p"] = protoP2P
protocolsByCode[protoP2P.Code] = protoP2P
} }
func AddProtocol(p Protocol) error { func AddProtocol(p Protocol) error {
if _, ok := protocolsByName[p.Name]; ok {
return fmt.Errorf("protocol by the name %q already exists", p.Name)
}
if _, ok := protocolsByCode[p.Code]; ok {
return fmt.Errorf("protocol code %d already taken by %q", p.Code, p.Code)
}
if p.Size != 0 && p.Transcoder == nil { if p.Size != 0 && p.Transcoder == nil {
return fmt.Errorf("protocols with arguments must define transcoders") return fmt.Errorf("protocols with arguments must define transcoders")
} }
if p.Path && p.Size >= 0 { if p.Path && p.Size >= 0 {
return fmt.Errorf("path protocols must have variable-length sizes") return fmt.Errorf("path protocols must have variable-length sizes")
} }
for _, pt := range Protocols {
if pt.Code == p.Code {
return fmt.Errorf("protocol code %d already taken by %q", p.Code, pt.Name)
}
if pt.Name == p.Name {
return fmt.Errorf("protocol by the name %q already exists", p.Name)
}
}
Protocols = append(Protocols, p) Protocols = append(Protocols, p)
ProtocolsByName[p.Name] = p protocolsByName[p.Name] = p
protocolsByCode[p.Code] = p
return nil return nil
} }
// ProtocolWithName returns the Protocol description with given string name. // ProtocolWithName returns the Protocol description with given string name.
func ProtocolWithName(s string) Protocol { func ProtocolWithName(s string) Protocol {
return ProtocolsByName[s] return protocolsByName[s]
} }
// ProtocolWithCode returns the Protocol description with given protocol code. // ProtocolWithCode returns the Protocol description with given protocol code.
func ProtocolWithCode(c int) Protocol { func ProtocolWithCode(c int) Protocol {
for _, p := range Protocols { return protocolsByCode[c]
if p.Code == c {
return p
}
}
return Protocol{}
} }
// ProtocolsWithString returns a slice of protocols matching given string. // ProtocolsWithString returns a slice of protocols matching given string.