libs5-go/protocol/message.go

51 lines
1.3 KiB
Go
Raw Normal View History

2024-01-06 11:33:46 +00:00
package protocol
import (
"git.lumeweb.com/LumeWeb/libs5-go/types"
)
var (
2024-01-30 03:25:21 +00:00
messageTypes map[int]func() IncomingMessage
2024-01-06 11:33:46 +00:00
)
func RegisterProtocols() {
2024-01-30 03:25:21 +00:00
messageTypes = make(map[int]func() IncomingMessage)
2024-01-06 11:33:46 +00:00
// Register factory functions instead of instances
2024-01-30 03:25:21 +00:00
RegisterMessageType(int(types.ProtocolMethodHandshakeOpen), func() IncomingMessage {
2024-01-06 11:33:46 +00:00
return NewHandshakeOpen([]byte{}, "")
})
2024-01-30 03:25:21 +00:00
RegisterMessageType(int(types.ProtocolMethodHashQuery), func() IncomingMessage {
2024-01-08 17:42:02 +00:00
return NewHashQuery()
})
2024-01-30 03:25:21 +00:00
RegisterMessageType(int(types.RecordTypeStorageLocation), func() IncomingMessage {
return NewStorageLocation()
})
2024-01-30 03:25:21 +00:00
RegisterMessageType(int(types.RecordTypeRegistryEntry), func() IncomingMessage {
2024-01-10 11:21:03 +00:00
return NewEmptyRegistryEntryRequest()
})
2024-01-30 03:25:21 +00:00
RegisterMessageType(int(types.ProtocolMethodRegistryQuery), func() IncomingMessage {
2024-01-10 11:21:03 +00:00
return NewEmptyRegistryQuery()
})
2024-01-30 03:25:21 +00:00
RegisterMessageType(int(types.ProtocolMethodSignedMessage), func() IncomingMessage {
return NewSignedMessage()
2024-01-06 11:33:46 +00:00
})
2024-01-08 17:42:02 +00:00
2024-01-06 11:33:46 +00:00
}
2024-01-30 03:25:21 +00:00
func RegisterMessageType(messageType int, factoryFunc func() IncomingMessage) {
2024-01-06 11:33:46 +00:00
if factoryFunc == nil {
panic("factoryFunc cannot be nil")
}
2024-01-09 18:57:35 +00:00
messageTypes[messageType] = factoryFunc
2024-01-06 11:33:46 +00:00
}
2024-01-30 03:25:21 +00:00
func GetMessageType(kind int) (IncomingMessage, bool) {
2024-01-09 18:57:35 +00:00
value, ok := messageTypes[kind]
2024-01-06 11:33:46 +00:00
if !ok {
return nil, false
}
2024-01-09 18:57:35 +00:00
return value(), true
2024-01-06 11:33:46 +00:00
}