New web-sockets connector
#5170
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,41 @@ | ||
| package factory | ||
|
|
||
| import ( | ||
| "github.com/multiversx/mx-chain-communication-go/websocket/data" | ||
| "github.com/multiversx/mx-chain-communication-go/websocket/factory" | ||
| "github.com/multiversx/mx-chain-core-go/marshal" | ||
| "github.com/multiversx/mx-chain-go/config" | ||
| "github.com/multiversx/mx-chain-go/outport" | ||
| "github.com/multiversx/mx-chain-go/outport/host" | ||
| logger "github.com/multiversx/mx-chain-logger-go" | ||
| ) | ||
|
|
||
| type ArgsHostDriverFactory struct { | ||
| HostConfig config.HostDriverConfig | ||
| Marshaller marshal.Marshalizer | ||
| } | ||
|
|
||
| var log = logger.GetOrCreate("outport/factory/hostdriver") | ||
|
|
||
| func CreateHostDriver(args ArgsHostDriverFactory) (outport.Driver, error) { | ||
|
There was a problem hiding this comment. missing comments in the whole file |
||
| wsHost, err := factory.CreateWebSocketHost(factory.ArgsWebSocketHost{ | ||
| WebSocketConfig: data.WebSocketConfig{ | ||
| URL: args.HostConfig.URL, | ||
| WithAcknowledge: args.HostConfig.WithAcknowledge, | ||
| IsServer: args.HostConfig.IsServer, | ||
| RetryDurationInSec: args.HostConfig.RetryDurationInSec, | ||
| BlockingAckOnError: args.HostConfig.BlockingAckOnError, | ||
| }, | ||
| Marshaller: args.Marshaller, | ||
| Log: log, | ||
| }) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return host.NewHostDriver(host.ArgsHostDriver{ | ||
| Marshaller: args.Marshaller, | ||
| SenderHost: wsHost, | ||
| Log: log, | ||
| }) | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| package factory | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,114 @@ | ||
| package host | ||
|
|
||
| import ( | ||
| "fmt" | ||
|
|
||
| "github.com/multiversx/mx-chain-core-go/core" | ||
| "github.com/multiversx/mx-chain-core-go/core/atomic" | ||
| "github.com/multiversx/mx-chain-core-go/core/check" | ||
| "github.com/multiversx/mx-chain-core-go/data/outport" | ||
| "github.com/multiversx/mx-chain-core-go/marshal" | ||
| ) | ||
|
|
||
| // ArgsHostDriver holds the arguments needed for creating a new hostDriver | ||
| type ArgsHostDriver struct { | ||
| Marshaller marshal.Marshalizer | ||
| SenderHost SenderHost | ||
| Log core.Logger | ||
| } | ||
|
|
||
| type hostDriver struct { | ||
| marshaller marshal.Marshalizer | ||
| senderHost SenderHost | ||
| isClosed atomic.Flag | ||
| log core.Logger | ||
| } | ||
|
|
||
| func NewHostDriver(args ArgsHostDriver) (*hostDriver, error) { | ||
|
|
||
| if check.IfNil(args.SenderHost) { | ||
| return nil, ErrNilHost | ||
| } | ||
| if check.IfNil(args.Marshaller) { | ||
| return nil, core.ErrNilMarshalizer | ||
| } | ||
| if check.IfNil(args.Log) { | ||
| return nil, core.ErrNilLogger | ||
| } | ||
|
|
||
| return &hostDriver{ | ||
| marshaller: args.Marshaller, | ||
| senderHost: args.SenderHost, | ||
| log: args.Log, | ||
| isClosed: atomic.Flag{}, | ||
| }, nil | ||
| } | ||
|
|
||
| func (o *hostDriver) SaveBlock(outportBlock *outport.OutportBlock) error { | ||
|
|
||
| return o.handleAction(outportBlock, outport.TopicSaveBlock) | ||
| } | ||
|
|
||
| // RevertIndexedBlock will handle the action of reverting the indexed block | ||
| func (o *hostDriver) RevertIndexedBlock(blockData *outport.BlockData) error { | ||
| return o.handleAction(blockData, outport.TopicRevertIndexedBlock) | ||
| } | ||
|
|
||
| // SaveRoundsInfo will handle the saving of rounds | ||
| func (o *hostDriver) SaveRoundsInfo(roundsInfos *outport.RoundsInfo) error { | ||
| return o.handleAction(roundsInfos, outport.TopicSaveRoundsInfo) | ||
| } | ||
|
|
||
| // SaveValidatorsPubKeys will handle the saving of the validators' public keys | ||
| func (o *hostDriver) SaveValidatorsPubKeys(validatorsPubKeys *outport.ValidatorsPubKeys) error { | ||
| return o.handleAction(validatorsPubKeys, outport.TopicSaveValidatorsPubKeys) | ||
| } | ||
|
|
||
| // SaveValidatorsRating will handle the saving of the validators' rating | ||
| func (o *hostDriver) SaveValidatorsRating(validatorsRating *outport.ValidatorsRating) error { | ||
| return o.handleAction(validatorsRating, outport.TopicSaveValidatorsRating) | ||
| } | ||
|
|
||
| // SaveAccounts will handle the accounts' saving | ||
| func (o *hostDriver) SaveAccounts(accounts *outport.Accounts) error { | ||
| return o.handleAction(accounts, outport.TopicSaveAccounts) | ||
| } | ||
|
|
||
| // FinalizedBlock will handle the finalized block | ||
| func (o *hostDriver) FinalizedBlock(finalizedBlock *outport.FinalizedBlock) error { | ||
| return o.handleAction(finalizedBlock, outport.TopicFinalizedBlock) | ||
| } | ||
|
|
||
| // GetMarshaller returns the internal marshaller | ||
| func (o *hostDriver) GetMarshaller() marshal.Marshalizer { | ||
| return o.marshaller | ||
| } | ||
|
|
||
| func (o *hostDriver) handleAction(args interface{}, topic string) error { | ||
| if o.isClosed.IsSet() { | ||
| return ErrHostIsClosed | ||
| } | ||
|
|
||
| marshalledPayload, err := o.marshaller.Marshal(args) | ||
| if err != nil { | ||
| o.log.Error("cannot marshal block", "topic", topic, "error", err) | ||
| return fmt.Errorf("%w while marshaling block for topic %s", err, topic) | ||
| } | ||
|
|
||
| err = o.senderHost.Send(marshalledPayload, topic) | ||
| if err != nil { | ||
| o.log.Error("cannot send on route", "topic", topic, "error", err) | ||
|
There was a problem hiding this comment. Since this still can happen while closing the node, I would filter out log.Error prints only if they are not of type closed error. |
||
| return fmt.Errorf("%w while sending data on route for topic %s", err, topic) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // Close will handle the closing of the outport driver web socket sender | ||
| func (o *hostDriver) Close() error { | ||
| o.isClosed.SetValue(true) | ||
| return o.senderHost.Close() | ||
| } | ||
|
|
||
| // IsInterfaceNil returns true if there is no value under the interface | ||
| func (o *hostDriver) IsInterfaceNil() bool { | ||
| return o == nil | ||
| } | ||
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe "ServerMode = false/true" ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
or
Mode = "server"There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
changed in
Mode