95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package event
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha512"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
aws_sns "github.com/aws/aws-sdk-go-v2/service/sns"
|
|
"github.com/aws/aws-sdk-go-v2/service/sns/types"
|
|
"github.com/wormhole-foundation/wormhole-explorer/fly/internal/track"
|
|
)
|
|
|
|
type SnsEventDispatcher struct {
|
|
api *aws_sns.Client
|
|
url string
|
|
}
|
|
|
|
func NewSnsEventDispatcher(awsConfig aws.Config, url string) (*SnsEventDispatcher, error) {
|
|
return &SnsEventDispatcher{
|
|
api: aws_sns.NewFromConfig(awsConfig),
|
|
url: url,
|
|
}, nil
|
|
}
|
|
|
|
func (s *SnsEventDispatcher) NewDuplicateVaa(ctx context.Context, e DuplicateVaa) error {
|
|
attrs := map[string]types.MessageAttributeValue{
|
|
"messageType": {
|
|
DataType: aws.String("String"),
|
|
StringValue: aws.String("duplicated-vaa"),
|
|
},
|
|
}
|
|
body, err := json.Marshal(event{
|
|
TrackID: track.GetTrackIDForDuplicatedVAA(e.VaaID),
|
|
Type: "duplicated-vaa",
|
|
Source: "fly",
|
|
Data: e,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
groupID := createDeduplicationIDForDuplicateVaa(e)
|
|
_, err = s.api.Publish(ctx,
|
|
&aws_sns.PublishInput{
|
|
MessageGroupId: aws.String(groupID),
|
|
MessageDeduplicationId: aws.String(groupID),
|
|
Message: aws.String(string(body)),
|
|
TopicArn: aws.String(s.url),
|
|
MessageAttributes: attrs,
|
|
})
|
|
return err
|
|
}
|
|
|
|
func createDeduplicationIDForDuplicateVaa(e DuplicateVaa) string {
|
|
id := fmt.Sprintf("%s%s", e.Digest, e.VaaID)
|
|
h := sha512.New()
|
|
io.WriteString(h, id)
|
|
deduplicationID := base64.StdEncoding.EncodeToString(h.Sum(nil))
|
|
if len(deduplicationID) > 127 {
|
|
return deduplicationID[:127]
|
|
}
|
|
return deduplicationID
|
|
}
|
|
|
|
func (s *SnsEventDispatcher) NewGovernorStatus(ctx context.Context, e GovernorStatus) error {
|
|
attrs := map[string]types.MessageAttributeValue{
|
|
"messageType": {
|
|
DataType: aws.String("String"),
|
|
StringValue: aws.String("governor"),
|
|
},
|
|
}
|
|
body, err := json.Marshal(event{
|
|
TrackID: track.GetTrackIDForGovernorStatus(e.NodeName, e.Timestamp),
|
|
Type: "governor-status",
|
|
Source: "fly",
|
|
Data: e,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
groupID := fmt.Sprintf("%s-%v", e.NodeAddress, e.Timestamp)
|
|
_, err = s.api.Publish(ctx,
|
|
&aws_sns.PublishInput{
|
|
MessageGroupId: aws.String(groupID),
|
|
MessageDeduplicationId: aws.String(groupID),
|
|
Message: aws.String(string(body)),
|
|
TopicArn: aws.String(s.url),
|
|
MessageAttributes: attrs,
|
|
})
|
|
return err
|
|
}
|