/
server_factory.go
151 lines (139 loc) · 7.03 KB
/
server_factory.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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
//go:build go1.18
// +build go1.18
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.
// Code generated by Microsoft (R) AutoRest Code Generator. DO NOT EDIT.
// Changes may cause incorrect behavior and will be lost if the code is regenerated.
package fake
import (
"errors"
"fmt"
"github.com/Azure/azure-sdk-for-go/sdk/azcore/runtime"
"net/http"
"strings"
"sync"
)
// ServerFactory is a fake server for instances of the armpostgresqlflexibleservers.ClientFactory type.
type ServerFactory struct {
AdministratorsServer AdministratorsServer
BackupsServer BackupsServer
CheckNameAvailabilityServer CheckNameAvailabilityServer
CheckNameAvailabilityWithLocationServer CheckNameAvailabilityWithLocationServer
ConfigurationsServer ConfigurationsServer
DatabasesServer DatabasesServer
FirewallRulesServer FirewallRulesServer
GetPrivateDNSZoneSuffixServer GetPrivateDNSZoneSuffixServer
LocationBasedCapabilitiesServer LocationBasedCapabilitiesServer
OperationsServer OperationsServer
ReplicasServer ReplicasServer
ServersServer ServersServer
VirtualNetworkSubnetUsageServer VirtualNetworkSubnetUsageServer
}
// NewServerFactoryTransport creates a new instance of ServerFactoryTransport with the provided implementation.
// The returned ServerFactoryTransport instance is connected to an instance of armpostgresqlflexibleservers.ClientFactory via the
// azcore.ClientOptions.Transporter field in the client's constructor parameters.
func NewServerFactoryTransport(srv *ServerFactory) *ServerFactoryTransport {
return &ServerFactoryTransport{
srv: srv,
}
}
// ServerFactoryTransport connects instances of armpostgresqlflexibleservers.ClientFactory to instances of ServerFactory.
// Don't use this type directly, use NewServerFactoryTransport instead.
type ServerFactoryTransport struct {
srv *ServerFactory
trMu sync.Mutex
trAdministratorsServer *AdministratorsServerTransport
trBackupsServer *BackupsServerTransport
trCheckNameAvailabilityServer *CheckNameAvailabilityServerTransport
trCheckNameAvailabilityWithLocationServer *CheckNameAvailabilityWithLocationServerTransport
trConfigurationsServer *ConfigurationsServerTransport
trDatabasesServer *DatabasesServerTransport
trFirewallRulesServer *FirewallRulesServerTransport
trGetPrivateDNSZoneSuffixServer *GetPrivateDNSZoneSuffixServerTransport
trLocationBasedCapabilitiesServer *LocationBasedCapabilitiesServerTransport
trOperationsServer *OperationsServerTransport
trReplicasServer *ReplicasServerTransport
trServersServer *ServersServerTransport
trVirtualNetworkSubnetUsageServer *VirtualNetworkSubnetUsageServerTransport
}
// Do implements the policy.Transporter interface for ServerFactoryTransport.
func (s *ServerFactoryTransport) Do(req *http.Request) (*http.Response, error) {
rawMethod := req.Context().Value(runtime.CtxAPINameKey{})
method, ok := rawMethod.(string)
if !ok {
return nil, nonRetriableError{errors.New("unable to dispatch request, missing value for CtxAPINameKey")}
}
client := method[:strings.Index(method, ".")]
var resp *http.Response
var err error
switch client {
case "AdministratorsClient":
initServer(s, &s.trAdministratorsServer, func() *AdministratorsServerTransport {
return NewAdministratorsServerTransport(&s.srv.AdministratorsServer)
})
resp, err = s.trAdministratorsServer.Do(req)
case "BackupsClient":
initServer(s, &s.trBackupsServer, func() *BackupsServerTransport { return NewBackupsServerTransport(&s.srv.BackupsServer) })
resp, err = s.trBackupsServer.Do(req)
case "CheckNameAvailabilityClient":
initServer(s, &s.trCheckNameAvailabilityServer, func() *CheckNameAvailabilityServerTransport {
return NewCheckNameAvailabilityServerTransport(&s.srv.CheckNameAvailabilityServer)
})
resp, err = s.trCheckNameAvailabilityServer.Do(req)
case "CheckNameAvailabilityWithLocationClient":
initServer(s, &s.trCheckNameAvailabilityWithLocationServer, func() *CheckNameAvailabilityWithLocationServerTransport {
return NewCheckNameAvailabilityWithLocationServerTransport(&s.srv.CheckNameAvailabilityWithLocationServer)
})
resp, err = s.trCheckNameAvailabilityWithLocationServer.Do(req)
case "ConfigurationsClient":
initServer(s, &s.trConfigurationsServer, func() *ConfigurationsServerTransport {
return NewConfigurationsServerTransport(&s.srv.ConfigurationsServer)
})
resp, err = s.trConfigurationsServer.Do(req)
case "DatabasesClient":
initServer(s, &s.trDatabasesServer, func() *DatabasesServerTransport { return NewDatabasesServerTransport(&s.srv.DatabasesServer) })
resp, err = s.trDatabasesServer.Do(req)
case "FirewallRulesClient":
initServer(s, &s.trFirewallRulesServer, func() *FirewallRulesServerTransport {
return NewFirewallRulesServerTransport(&s.srv.FirewallRulesServer)
})
resp, err = s.trFirewallRulesServer.Do(req)
case "GetPrivateDNSZoneSuffixClient":
initServer(s, &s.trGetPrivateDNSZoneSuffixServer, func() *GetPrivateDNSZoneSuffixServerTransport {
return NewGetPrivateDNSZoneSuffixServerTransport(&s.srv.GetPrivateDNSZoneSuffixServer)
})
resp, err = s.trGetPrivateDNSZoneSuffixServer.Do(req)
case "LocationBasedCapabilitiesClient":
initServer(s, &s.trLocationBasedCapabilitiesServer, func() *LocationBasedCapabilitiesServerTransport {
return NewLocationBasedCapabilitiesServerTransport(&s.srv.LocationBasedCapabilitiesServer)
})
resp, err = s.trLocationBasedCapabilitiesServer.Do(req)
case "OperationsClient":
initServer(s, &s.trOperationsServer, func() *OperationsServerTransport { return NewOperationsServerTransport(&s.srv.OperationsServer) })
resp, err = s.trOperationsServer.Do(req)
case "ReplicasClient":
initServer(s, &s.trReplicasServer, func() *ReplicasServerTransport { return NewReplicasServerTransport(&s.srv.ReplicasServer) })
resp, err = s.trReplicasServer.Do(req)
case "ServersClient":
initServer(s, &s.trServersServer, func() *ServersServerTransport { return NewServersServerTransport(&s.srv.ServersServer) })
resp, err = s.trServersServer.Do(req)
case "VirtualNetworkSubnetUsageClient":
initServer(s, &s.trVirtualNetworkSubnetUsageServer, func() *VirtualNetworkSubnetUsageServerTransport {
return NewVirtualNetworkSubnetUsageServerTransport(&s.srv.VirtualNetworkSubnetUsageServer)
})
resp, err = s.trVirtualNetworkSubnetUsageServer.Do(req)
default:
err = fmt.Errorf("unhandled client %s", client)
}
if err != nil {
return nil, err
}
return resp, nil
}
func initServer[T any](s *ServerFactoryTransport, dst **T, src func() *T) {
s.trMu.Lock()
if *dst == nil {
*dst = src()
}
s.trMu.Unlock()
}