/
namespace.go
63 lines (53 loc) · 2.13 KB
/
namespace.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package servicebus
import (
"errors"
"time"
"github.com/Azure/azure-amqp-common-go/v3/auth"
servicebussdk "github.com/Azure/azure-service-bus-go"
"github.com/Azure/go-shuttle/internal/aad"
)
const (
serviceBusResourceURI = "https://servicebus.azure.net/"
//cannot be higher than 5 minutes
LockDuration = 5 * time.Minute
)
var errorOption func(h *servicebussdk.Namespace) error
func newErrorOption(err error) servicebussdk.NamespaceOption {
return func(ns *servicebussdk.Namespace) error { return err }
}
// NamespaceWithManagedIdentity is a custom NamespaceOption to instantiate a Service Bus namespace client with
// managed identity resource id
func NamespaceWithManagedIdentityResourceID(serviceBusNamespaceName, managedIdentityResourceID string) servicebussdk.NamespaceOption {
provider, err := aad.NewJWTProvider(
aad.JWTProviderWithManagedIdentityResourceID(managedIdentityResourceID, ""),
aad.JWTProviderWithResourceURI(serviceBusResourceURI),
)
if err != nil {
// TODO: make this fail at creation of the option, instead of runtime.
return newErrorOption(err)
}
return NamespaceWithTokenProvider(serviceBusNamespaceName, provider)
}
// NamespaceWithManagedIdentityClientID is a custom NamespaceOption to instantiate a Service Bus namespace client with
// managed identity client id
func NamespaceWithManagedIdentityClientID(serviceBusNamespaceName, managedIdentityClientID string) servicebussdk.NamespaceOption {
provider, err := aad.NewJWTProvider(
aad.JWTProviderWithManagedIdentityClientID(managedIdentityClientID, ""),
aad.JWTProviderWithResourceURI(serviceBusResourceURI),
)
if err != nil {
// TODO: make this fail at creation of the option, instead of runtime.
return newErrorOption(err)
}
return NamespaceWithTokenProvider(serviceBusNamespaceName, provider)
}
func NamespaceWithTokenProvider(serviceBusNamespaceName string, provider auth.TokenProvider) servicebussdk.NamespaceOption {
return func(ns *servicebussdk.Namespace) error {
if serviceBusNamespaceName == "" {
return errors.New("no Service Bus namespace name provided")
}
ns.TokenProvider = provider
ns.Name = serviceBusNamespaceName
return nil
}
}