You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.7 KiB
62 lines
1.7 KiB
package identitychain
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
|
|
"github.com/simple-rules/harmony-benchmark/p2p"
|
|
"github.com/simple-rules/harmony-benchmark/proto"
|
|
proto_identity "github.com/simple-rules/harmony-benchmark/proto/identity"
|
|
"github.com/simple-rules/harmony-benchmark/waitnode"
|
|
)
|
|
|
|
//IdentityChainHandler handles registration of new Identities
|
|
// This could have been its seperate package like consensus, but am avoiding creating a lot of packages.
|
|
func (IDC *IdentityChain) IdentityChainHandler(conn net.Conn) {
|
|
content, err := p2p.ReadMessageContent(conn)
|
|
if err != nil {
|
|
IDC.log.Error("Read p2p data failed")
|
|
return
|
|
}
|
|
msgCategory, err := proto.GetMessageCategory(content)
|
|
if err != nil {
|
|
IDC.log.Error("Read message category failed", "err", err)
|
|
return
|
|
}
|
|
if msgCategory != proto.IDENTITY {
|
|
IDC.log.Error("Identity Chain Recieved incorrect protocol message")
|
|
os.Exit(1)
|
|
}
|
|
msgType, err := proto.GetMessageType(content)
|
|
if err != nil {
|
|
IDC.log.Error("Read action type failed")
|
|
return
|
|
}
|
|
msgPayload, err := proto.GetMessagePayload(content)
|
|
if err != nil {
|
|
IDC.log.Error("Read message payload failed")
|
|
return
|
|
}
|
|
switch msgCategory {
|
|
case proto.IDENTITY:
|
|
actionType := proto_identity.IdentityMessageType(msgType)
|
|
switch actionType {
|
|
case proto_identity.IDENTITY:
|
|
IDC.registerIdentity(msgPayload)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func (IDC *IdentityChain) registerIdentity(msgPayload []byte) {
|
|
identityPayload, err := proto_identity.GetIdentityMessagePayload(msgPayload)
|
|
if err != nil {
|
|
IDC.log.Error("identity payload not read")
|
|
} else {
|
|
fmt.Println("identity payload read")
|
|
}
|
|
NewWaitNode := waitnode.DeserializeWaitNode(identityPayload)
|
|
IDC.PendingIdentities = append(IDC.PendingIdentities, NewWaitNode)
|
|
fmt.Println(len(IDC.PendingIdentities))
|
|
}
|
|
|