eosswedenorg/thalos
1
0
Fork 0
mirror of https://github.com/eosswedenorg/thalos synced 2026-06-16 04:24:56 +02:00
Code Releases Activity

transport/publisher.go: rename to writer.go

Browse source
This commit is contained in:
Henrik Hautakoski 2023-03-07 11:41:22 +01:00
parent b3f773655d
commit f9fc88b0fb
4 changed files with 20 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

24
app/ship_processor.go
View file

@ -27,18 +27,18 @@ func logDecoratedEncoder(encoder message.Encoder) message.Encoder {
}
type ShipProcessor struct {
abi *abi.AbiManager
publisher transport.Publisher
shClient *shipclient.Client
encode message.Encoder
abi *abi.AbiManager
writer transport.Writer
shClient *shipclient.Client
encode message.Encoder
}
func SpawnProccessor(shClient *shipclient.Client, publisher transport.Publisher, abi *abi.AbiManager) {
func SpawnProccessor(shClient *shipclient.Client, writer transport.Writer, abi *abi.AbiManager) {
processor := &ShipProcessor{
abi: abi,
publisher: publisher,
shClient: shClient,
encode: logDecoratedEncoder(json.Marshal),
abi: abi,
writer: writer,
shClient: shClient,
encode: logDecoratedEncoder(json.Marshal),
}
// Attach handlers
@ -47,7 +47,7 @@ func SpawnProccessor(shClient *shipclient.Client, publisher transport.Publisher,
}
func (processor *ShipProcessor) queueMessage(channel transport.ChannelInterface, payload []byte) bool {
err := processor.publisher.Publish(channel, payload)
err := processor.writer.Write(channel, payload)
if err != nil {
log.WithError(err).Errorf("Failed to post to channel '%s'", channel)
return false
@ -76,7 +76,7 @@ func (processor *ShipProcessor) processBlock(block *ship.GetBlocksResultV0) {
processor.encodeQueue(transport.HeartbeatChannel, hb)
err := processor.publisher.Flush()
err := processor.writer.Flush()
if err != nil {
log.WithError(err).Error("Failed to send messages")
}
@ -129,7 +129,7 @@ func (processor *ShipProcessor) processTraces(traces []*ship.TransactionTraceV0)
}
}
err := processor.publisher.Flush()
err := processor.writer.Flush()
if err != nil {
log.WithError(err).Error("Failed to send messages")
}

2
transport/redis_pubsub/publisher.go
View file

@ -22,7 +22,7 @@ func NewPublisher(client *redis.Client, ns Namespace) *Publisher {
}
}
func (r *Publisher) Publish(channel transport.ChannelInterface, payload []byte) error {
func (r *Publisher) Write(channel transport.ChannelInterface, payload []byte) error {
return r.pipeline.Publish(r.ctx, r.ns.NewKey(channel).String(), payload).Err()
}

4
transport/redis_pubsub/publisher_test.go
View file

@ -18,8 +18,8 @@ func TestPublisher_Publish(t *testing.T) {
mock.ExpectPublish("ship::id::test", []byte("some string")).SetVal(0)
mock.ExpectPublish("ship::id::test2", []byte("some other string")).SetVal(0)
assert.NoError(t, pub.Publish(transport.Channel{"test"}, []byte("some string")))
assert.NoError(t, pub.Publish(transport.Channel{"test2"}, []byte("some other string")))
assert.NoError(t, pub.Write(transport.Channel{"test"}, []byte("some string")))
assert.NoError(t, pub.Write(transport.Channel{"test2"}, []byte("some other string")))
assert.NoError(t, pub.Flush())
assert.NoError(t, mock.ExpectationsWereMet())

10
transport/publisher.go → transport/writer.go
View file

@ -1,11 +1,11 @@
package transport
// Publisher interface defines the required methods
// to send messages over channel.
type Publisher interface {
// Publish a message to a channel.
// Writer interface defines the required methods
// to send messages over an channel.
type Writer interface {
// Write writes a message over a channel.
// The message may or may not be buffered depending on the implementation.
Publish(channel ChannelInterface, payload []byte) error
Write(channel ChannelInterface, payload []byte) error
// Flush writes any buffered messages to the channel.
// If the implementation does not support buffering. this is a noop.