mirror of
https://dev.azure.com/schwarzit/schwarzit.stackit-public/_git/audit-go
synced 2026-02-13 19:37:25 +00:00
5742604629
So far the SDK provided a messaging API that was not thread-safe (i.e. goroutine-safe). Additionally the SDK provided a MutexAPI which made it thread-safe at the cost of removed concurrency possibilities. The changes implemented in this commit replace both implementations with a thread-safe connection pool based solution. The api gateway is a SDK user that requires reliable high performance send capabilities with a limit amount of amqp connections. These changes in the PR try address their requirements by moving the responsibility of connection management into the SDK. From this change other SDK users will benefit as well. Security-concept-update-needed: false. JIRA Work Item: STACKITALO-62
78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
package messaging
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/Azure/go-amqp"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type amqpSender interface {
|
|
Send(ctx context.Context, msg *amqp.Message, opts *amqp.SendOptions) error
|
|
Close(ctx context.Context) error
|
|
}
|
|
|
|
type AmqpSenderSession struct {
|
|
session amqpSession
|
|
sender amqpSender
|
|
}
|
|
|
|
func (s *AmqpSenderSession) Send(
|
|
topic string,
|
|
data [][]byte,
|
|
contentType string,
|
|
applicationProperties map[string]any,
|
|
) error {
|
|
// check topic name
|
|
if !strings.HasPrefix(topic, AmqpTopicPrefix) {
|
|
return fmt.Errorf(
|
|
"topic %q name lacks mandatory prefix %q",
|
|
topic,
|
|
AmqpTopicPrefix,
|
|
)
|
|
}
|
|
|
|
if contentType == "" {
|
|
return errors.New("content-type is required")
|
|
}
|
|
|
|
// prepare the amqp message
|
|
message := amqp.Message{
|
|
Header: &amqp.MessageHeader{
|
|
Durable: true,
|
|
},
|
|
Properties: &amqp.MessageProperties{
|
|
To: &topic,
|
|
ContentType: &contentType,
|
|
},
|
|
ApplicationProperties: applicationProperties,
|
|
Data: data,
|
|
}
|
|
|
|
// send
|
|
ctx, cancelFn := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancelFn()
|
|
return s.sender.Send(ctx, &message, nil)
|
|
}
|
|
|
|
func (s *AmqpSenderSession) Close() error {
|
|
ctx, cancelFn := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancelFn()
|
|
|
|
var closeErrors []error
|
|
senderErr := s.sender.Close(ctx)
|
|
if senderErr != nil {
|
|
closeErrors = append(closeErrors, senderErr)
|
|
}
|
|
sessionErr := s.session.Close(ctx)
|
|
if sessionErr != nil {
|
|
closeErrors = append(closeErrors, sessionErr)
|
|
}
|
|
|
|
if len(closeErrors) > 0 {
|
|
return errors.Join(closeErrors...)
|
|
}
|
|
return nil
|
|
}
|