credentials.*Config.SetConnectTimeout - Code Metrics - Inspection of "add set func to config" - aliyun/credentials-go - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Pull Request — master (#25)

by zuochao

created 2020-09-05 01:35 UTC

credentials.*Config.SetConnectTimeout A

↳ Parent: credentials/credential.go

Complexity

Conditions

Size

Total Lines	3
Code Lines	3

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	1
eloc	3
nop	1
dl	0
loc	3
rs	10
c	0
b	0
f	0

package credentials

import (
	"bufio"
	"errors"
	"fmt"
	"net/http"
	"net/url"
	"os"
	"strings"
	"time"

	"github.com/alibabacloud-go/debug/debug"
	"github.com/alibabacloud-go/tea/tea"
	"github.com/aliyun/credentials-go/credentials/request"
	"github.com/aliyun/credentials-go/credentials/response"
	"github.com/aliyun/credentials-go/credentials/utils"
)

var debuglog = debug.Init("credential")

var hookParse = func(err error) error {
	return err
}

// Credential is an interface for getting actual credential
type Credential interface {
	GetAccessKeyId() (*string, error)
	GetAccessKeySecret() (*string, error)
	GetSecurityToken() (*string, error)
	GetBearerToken() *string
	GetType() *string
}

// Config is important when call NewCredential
type Config struct {
	Type                  *string `json:"type"`
	AccessKeyId           *string `json:"access_key_id"`
	AccessKeySecret       *string `json:"access_key_secret"`
	RoleArn               *string `json:"role_arn"`
	RoleSessionName       *string `json:"role_session_name"`
	PublicKeyId           *string `json:"public_key_id"`
	RoleName              *string `json:"role_name"`
	SessionExpiration     *int    `json:"session_expiration"`
	PrivateKeyFile        *string `json:"private_key_file"`
	BearerToken           *string `json:"bearer_token"`
	SecurityToken         *string `json:"security_token"`
	RoleSessionExpiration *int    `json:"role_session_expiratioon"`
	Policy                *string `json:"policy"`
	Host                  *string `json:"host"`
	Timeout               *int    `json:"timeout"`
	ConnectTimeout        *int    `json:"connect_timeout"`
	Proxy                 *string `json:"proxy"`
}

func (s Config) String() string {
	return tea.Prettify(s)
}

func (s Config) GoString() string {
	return s.String()
}

func (s *Config) SetAccessKeyId(v string) *Config {
	s.AccessKeyId = &v
	return s
}

func (s *Config) SetAccessKeySecret(v string) *Config {
	s.AccessKeySecret = &v
	return s
}

func (s *Config) SetSecurityToken(v string) *Config {
	s.SecurityToken = &v
	return s
}

func (s *Config) SetRoleArn(v string) *Config {
	s.RoleArn = &v
	return s
}

func (s *Config) SetRoleSessionName(v string) *Config {
	s.RoleSessionName = &v
	return s
}

func (s *Config) SetPublicKeyId(v string) *Config {
	s.PublicKeyId = &v
	return s
}

func (s *Config) SetRoleName(v string) *Config {
	s.RoleName = &v
	return s
}

func (s *Config) SetSessionExpiration(v int) *Config {
	s.SessionExpiration = &v
	return s
}

func (s *Config) SetPrivateKeyFile(v string) *Config {
	s.PrivateKeyFile = &v
	return s
}

func (s *Config) SetBearerToken(v string) *Config {
	s.BearerToken = &v
	return s
}

func (s *Config) SetRoleSessionExpiration(v int) *Config {
	s.RoleSessionExpiration = &v
	return s
}

func (s *Config) SetPolicy(v string) *Config {
	s.Policy = &v
	return s
}

func (s *Config) SetHost(v string) *Config {
	s.Host = &v
	return s
}

func (s *Config) SetTimeout(v int) *Config {
	s.Timeout = &v
	return s
}

func (s *Config) SetConnectTimeout(v int) *Config {
	s.ConnectTimeout = &v
	return s
}

func (s *Config) SetProxy(v string) *Config {
	s.Proxy = &v
	return s
}

func (s *Config) SetType(v string) *Config {
	s.Type = &v
	return s
}

// NewCredential return a credential according to the type in config.
// if config is nil, the function will use default provider chain to get credential.
// please see README.md for detail.
func NewCredential(config *Config) (credential Credential, err error) {
	if config == nil {
		config, err = defaultChain.resolve()
		if err != nil {
			return
		}
		return NewCredential(config)
	}
	switch tea.StringValue(config.Type) {
	case "access_key":
		err = checkAccessKey(config)
		if err != nil {
			return
		}
		credential = newAccessKeyCredential(tea.StringValue(config.AccessKeyId), tea.StringValue(config.AccessKeySecret))
	case "sts":
		err = checkSTS(config)
		if err != nil {
			return
		}
		credential = newStsTokenCredential(tea.StringValue(config.AccessKeyId), tea.StringValue(config.AccessKeySecret), tea.StringValue(config.SecurityToken))
	case "ecs_ram_role":
		checkEcsRAMRole(config)
		runtime := &utils.Runtime{
			Host:           tea.StringValue(config.Host),
			Proxy:          tea.StringValue(config.Proxy),
			ReadTimeout:    tea.IntValue(config.Timeout),
			ConnectTimeout: tea.IntValue(config.ConnectTimeout),
		}
		credential = newEcsRAMRoleCredential(tea.StringValue(config.RoleName), runtime)
	case "ram_role_arn":
		err = checkRAMRoleArn(config)
		if err != nil {
			return
		}
		runtime := &utils.Runtime{
			Host:           tea.StringValue(config.Host),
			Proxy:          tea.StringValue(config.Proxy),
			ReadTimeout:    tea.IntValue(config.Timeout),
			ConnectTimeout: tea.IntValue(config.ConnectTimeout),
		}
		credential = newRAMRoleArnCredential(tea.StringValue(config.AccessKeyId), tea.StringValue(config.AccessKeySecret), tea.StringValue(config.RoleArn), tea.StringValue(config.RoleSessionName), tea.StringValue(config.Policy), tea.IntValue(config.RoleSessionExpiration), runtime)
	case "rsa_key_pair":
		err = checkRSAKeyPair(config)
		if err != nil {
			return
		}
		file, err1 := os.Open(tea.StringValue(config.PrivateKeyFile))
		if err1 != nil {
			err = fmt.Errorf("InvalidPath: Can not open PrivateKeyFile, err is %s", err1.Error())
			return
		}
		defer file.Close()
		var privateKey string
		scan := bufio.NewScanner(file)
		for scan.Scan() {
			if strings.HasPrefix(scan.Text(), "----") {
				continue
			}
			privateKey += scan.Text() + "\n"
		}
		runtime := &utils.Runtime{
			Host:           tea.StringValue(config.Host),
			Proxy:          tea.StringValue(config.Proxy),
			ReadTimeout:    tea.IntValue(config.Timeout),
			ConnectTimeout: tea.IntValue(config.ConnectTimeout),
		}
		credential = newRsaKeyPairCredential(privateKey, tea.StringValue(config.PublicKeyId), tea.IntValue(config.SessionExpiration), runtime)
	case "bearer":
		if tea.StringValue(config.BearerToken) == "" {
			err = errors.New("BearerToken cannot be empty")
			return
		}
		credential = newBearerTokenCredential(tea.StringValue(config.BearerToken))
	default:
		err = errors.New("Invalid type option, support: access_key, sts, ecs_ram_role, ram_role_arn, rsa_key_pair")
		return
	}
	return credential, nil
}

func checkRSAKeyPair(config *Config) (err error) {
	if tea.StringValue(config.PrivateKeyFile) == "" {
		err = errors.New("PrivateKeyFile cannot be empty")
		return
	}
	if tea.StringValue(config.PublicKeyId) == "" {
		err = errors.New("PublicKeyId cannot be empty")
		return
	}
	return
}

func checkRAMRoleArn(config *Config) (err error) {
	if tea.StringValue(config.AccessKeySecret) == "" {
		err = errors.New("AccessKeySecret cannot be empty")
		return
	}
	if tea.StringValue(config.RoleArn) == "" {
		err = errors.New("RoleArn cannot be empty")
		return
	}
	if tea.StringValue(config.RoleSessionName) == "" {
		err = errors.New("RoleSessionName cannot be empty")
		return
	}
	if tea.StringValue(config.AccessKeyId) == "" {
		err = errors.New("AccessKeyId cannot be empty")
		return
	}
	return
}

func checkEcsRAMRole(config *Config) (err error) {
	return
}

func checkSTS(config *Config) (err error) {
	if tea.StringValue(config.AccessKeyId) == "" {
		err = errors.New("AccessKeyId cannot be empty")
		return
	}
	if tea.StringValue(config.AccessKeySecret) == "" {
		err = errors.New("AccessKeySecret cannot be empty")
		return
	}
	if tea.StringValue(config.SecurityToken) == "" {
		err = errors.New("SecurityToken cannot be empty")
		return
	}
	return
}

func checkAccessKey(config *Config) (err error) {
	if tea.StringValue(config.AccessKeyId) == "" {
		err = errors.New("AccessKeyId cannot be empty")
		return
	}
	if tea.StringValue(config.AccessKeySecret) == "" {
		err = errors.New("AccessKeySecret cannot be empty")
		return
	}
	return
}

func doAction(request *request.CommonRequest, runtime *utils.Runtime) (content []byte, err error) {
	httpRequest, err := http.NewRequest(request.Method, request.URL, strings.NewReader(""))
	if err != nil {
		return
	}
	httpRequest.Proto = "HTTP/1.1"
	httpRequest.Host = request.Domain
	debuglog("> %s %s %s", httpRequest.Method, httpRequest.URL.RequestURI(), httpRequest.Proto)
	debuglog("> Host: %s", httpRequest.Host)
	for key, value := range request.Headers {
		if value != "" {
			debuglog("> %s: %s", key, value)
			httpRequest.Header[key] = []string{value}
		}
	}
	debuglog(">")
	httpClient := &http.Client{}
	httpClient.Timeout = time.Duration(runtime.ReadTimeout) * time.Second
	proxy := &url.URL{}
	if runtime.Proxy != "" {
		proxy, err = url.Parse(runtime.Proxy)
		if err != nil {
			return
		}
	}
	trans := &http.Transport{}
	if proxy != nil && runtime.Proxy != "" {
		trans.Proxy = http.ProxyURL(proxy)
	}
	trans.DialContext = utils.Timeout(time.Duration(runtime.ConnectTimeout) * time.Second)
	httpClient.Transport = trans
	httpResponse, err := hookDo(httpClient.Do)(httpRequest)
	if err != nil {
		return
	}
	debuglog("< %s %s", httpResponse.Proto, httpResponse.Status)
	for key, value := range httpResponse.Header {
		debuglog("< %s: %v", key, strings.Join(value, ""))
	}
	debuglog("<")

	resp := &response.CommonResponse{}
	err = hookParse(resp.ParseFromHTTPResponse(httpResponse))
	if err != nil {
		return
	}
	debuglog("%s", resp.GetHTTPContentString())
	if resp.GetHTTPStatus() != http.StatusOK {
		err = fmt.Errorf("httpStatus: %d, message = %s", resp.GetHTTPStatus(), resp.GetHTTPContentString())
		return
	}
	return resp.GetHTTPContentBytes(), nil
}


1			package credentials
2
3			import (
4			"bufio"
5			"errors"
6			"fmt"
7			"net/http"
8			"net/url"
9			"os"
10			"strings"
11			"time"
12
13			"github.com/alibabacloud-go/debug/debug"
14			"github.com/alibabacloud-go/tea/tea"
15			"github.com/aliyun/credentials-go/credentials/request"
16			"github.com/aliyun/credentials-go/credentials/response"
17			"github.com/aliyun/credentials-go/credentials/utils"
18			)
19
20			var debuglog = debug.Init("credential")
21
22			var hookParse = func(err error) error {
23			return err
24			}
25
26			// Credential is an interface for getting actual credential
27			type Credential interface {
28			GetAccessKeyId() (*string, error)
29			GetAccessKeySecret() (*string, error)
30			GetSecurityToken() (*string, error)
31			GetBearerToken() *string
32			GetType() *string
33			}
34
35			// Config is important when call NewCredential
36			type Config struct {
37			Type *string `json:"type"`
38			AccessKeyId *string `json:"access_key_id"`
39			AccessKeySecret *string `json:"access_key_secret"`
40			RoleArn *string `json:"role_arn"`
41			RoleSessionName *string `json:"role_session_name"`
42			PublicKeyId *string `json:"public_key_id"`
43			RoleName *string `json:"role_name"`
44			SessionExpiration *int `json:"session_expiration"`
45			PrivateKeyFile *string `json:"private_key_file"`
46			BearerToken *string `json:"bearer_token"`
47			SecurityToken *string `json:"security_token"`
48			RoleSessionExpiration *int `json:"role_session_expiratioon"`
49			Policy *string `json:"policy"`
50			Host *string `json:"host"`
51			Timeout *int `json:"timeout"`
52			ConnectTimeout *int `json:"connect_timeout"`
53			Proxy *string `json:"proxy"`
54			}
55
56			func (s Config) String() string {
57			return tea.Prettify(s)
58			}
59
60			func (s Config) GoString() string {
61			return s.String()
62			}
63
64			func (s Config) SetAccessKeyId(v string) Config {
65			s.AccessKeyId = &v
66			return s
67			}
68
69			func (s Config) SetAccessKeySecret(v string) Config {
70			s.AccessKeySecret = &v
71			return s
72			}
73
74			func (s Config) SetSecurityToken(v string) Config {
75			s.SecurityToken = &v
76			return s
77			}
78
79			func (s Config) SetRoleArn(v string) Config {
80			s.RoleArn = &v
81			return s
82			}
83
84			func (s Config) SetRoleSessionName(v string) Config {
85			s.RoleSessionName = &v
86			return s
87			}
88
89			func (s Config) SetPublicKeyId(v string) Config {
90			s.PublicKeyId = &v
91			return s
92			}
93
94			func (s Config) SetRoleName(v string) Config {
95			s.RoleName = &v
96			return s
97			}
98
99			func (s Config) SetSessionExpiration(v int) Config {
100			s.SessionExpiration = &v
101			return s
102			}
103
104			func (s Config) SetPrivateKeyFile(v string) Config {
105			s.PrivateKeyFile = &v
106			return s
107			}
108
109			func (s Config) SetBearerToken(v string) Config {
110			s.BearerToken = &v
111			return s
112			}
113
114			func (s Config) SetRoleSessionExpiration(v int) Config {
115			s.RoleSessionExpiration = &v
116			return s
117			}
118
119			func (s Config) SetPolicy(v string) Config {
120			s.Policy = &v
121			return s
122			}
123
124			func (s Config) SetHost(v string) Config {
125			s.Host = &v
126			return s
127			}
128
129			func (s Config) SetTimeout(v int) Config {
130			s.Timeout = &v
131			return s
132			}
133
134			func (s Config) SetConnectTimeout(v int) Config {
135			s.ConnectTimeout = &v
136			return s
137			}
138
139			func (s Config) SetProxy(v string) Config {
140			s.Proxy = &v
141			return s
142			}
143
144			func (s Config) SetType(v string) Config {
145			s.Type = &v
146			return s
147			}
148
149			// NewCredential return a credential according to the type in config.
150			// if config is nil, the function will use default provider chain to get credential.
151			// please see README.md for detail.
152			func NewCredential(config *Config) (credential Credential, err error) {
153			if config == nil {
154			config, err = defaultChain.resolve()
155			if err != nil {
156			return
157			}
158			return NewCredential(config)
159			}
160			switch tea.StringValue(config.Type) {
161			case "access_key":
162			err = checkAccessKey(config)
163			if err != nil {
164			return
165			}
166			credential = newAccessKeyCredential(tea.StringValue(config.AccessKeyId), tea.StringValue(config.AccessKeySecret))
167			case "sts":
168			err = checkSTS(config)
169			if err != nil {
170			return
171			}
172			credential = newStsTokenCredential(tea.StringValue(config.AccessKeyId), tea.StringValue(config.AccessKeySecret), tea.StringValue(config.SecurityToken))
173			case "ecs_ram_role":
174			checkEcsRAMRole(config)
175			runtime := &utils.Runtime{
176			Host: tea.StringValue(config.Host),
177			Proxy: tea.StringValue(config.Proxy),
178			ReadTimeout: tea.IntValue(config.Timeout),
179			ConnectTimeout: tea.IntValue(config.ConnectTimeout),
180			}
181			credential = newEcsRAMRoleCredential(tea.StringValue(config.RoleName), runtime)
182			case "ram_role_arn":
183			err = checkRAMRoleArn(config)
184			if err != nil {
185			return
186			}
187			runtime := &utils.Runtime{
188			Host: tea.StringValue(config.Host),
189			Proxy: tea.StringValue(config.Proxy),
190			ReadTimeout: tea.IntValue(config.Timeout),
191			ConnectTimeout: tea.IntValue(config.ConnectTimeout),
192			}
193			credential = newRAMRoleArnCredential(tea.StringValue(config.AccessKeyId), tea.StringValue(config.AccessKeySecret), tea.StringValue(config.RoleArn), tea.StringValue(config.RoleSessionName), tea.StringValue(config.Policy), tea.IntValue(config.RoleSessionExpiration), runtime)
194			case "rsa_key_pair":
195			err = checkRSAKeyPair(config)
196			if err != nil {
197			return
198			}
199			file, err1 := os.Open(tea.StringValue(config.PrivateKeyFile))
200			if err1 != nil {
201			err = fmt.Errorf("InvalidPath: Can not open PrivateKeyFile, err is %s", err1.Error())
202			return
203			}
204			defer file.Close()
205			var privateKey string
206			scan := bufio.NewScanner(file)
207			for scan.Scan() {
208			if strings.HasPrefix(scan.Text(), "----") {
209			continue
210			}
211			privateKey += scan.Text() + "\n"
212			}
213			runtime := &utils.Runtime{
214			Host: tea.StringValue(config.Host),
215			Proxy: tea.StringValue(config.Proxy),
216			ReadTimeout: tea.IntValue(config.Timeout),
217			ConnectTimeout: tea.IntValue(config.ConnectTimeout),
218			}
219			credential = newRsaKeyPairCredential(privateKey, tea.StringValue(config.PublicKeyId), tea.IntValue(config.SessionExpiration), runtime)
220			case "bearer":
221			if tea.StringValue(config.BearerToken) == "" {
222			err = errors.New("BearerToken cannot be empty")
223			return
224			}
225			credential = newBearerTokenCredential(tea.StringValue(config.BearerToken))
226			default:
227			err = errors.New("Invalid type option, support: access_key, sts, ecs_ram_role, ram_role_arn, rsa_key_pair")
228			return
229			}
230			return credential, nil
231			}
232
233			func checkRSAKeyPair(config *Config) (err error) {
234			if tea.StringValue(config.PrivateKeyFile) == "" {
235			err = errors.New("PrivateKeyFile cannot be empty")
236			return
237			}
238			if tea.StringValue(config.PublicKeyId) == "" {
239			err = errors.New("PublicKeyId cannot be empty")
240			return
241			}
242			return
243			}
244
245			func checkRAMRoleArn(config *Config) (err error) {
246			if tea.StringValue(config.AccessKeySecret) == "" {
247			err = errors.New("AccessKeySecret cannot be empty")
248			return
249			}
250			if tea.StringValue(config.RoleArn) == "" {
251			err = errors.New("RoleArn cannot be empty")
252			return
253			}
254			if tea.StringValue(config.RoleSessionName) == "" {
255			err = errors.New("RoleSessionName cannot be empty")
256			return
257			}
258			if tea.StringValue(config.AccessKeyId) == "" {
259			err = errors.New("AccessKeyId cannot be empty")
260			return
261			}
262			return
263			}
264
265			func checkEcsRAMRole(config *Config) (err error) {
266			return
267			}
268
269			func checkSTS(config *Config) (err error) {
270			if tea.StringValue(config.AccessKeyId) == "" {
271			err = errors.New("AccessKeyId cannot be empty")
272			return
273			}
274			if tea.StringValue(config.AccessKeySecret) == "" {
275			err = errors.New("AccessKeySecret cannot be empty")
276			return
277			}
278			if tea.StringValue(config.SecurityToken) == "" {
279			err = errors.New("SecurityToken cannot be empty")
280			return
281			}
282			return
283			}
284
285			func checkAccessKey(config *Config) (err error) {
286			if tea.StringValue(config.AccessKeyId) == "" {
287			err = errors.New("AccessKeyId cannot be empty")
288			return
289			}
290			if tea.StringValue(config.AccessKeySecret) == "" {
291			err = errors.New("AccessKeySecret cannot be empty")
292			return
293			}
294			return
295			}
296
297			func doAction(request request.CommonRequest, runtime utils.Runtime) (content []byte, err error) {
298			httpRequest, err := http.NewRequest(request.Method, request.URL, strings.NewReader(""))
299			if err != nil {
300			return
301			}
302			httpRequest.Proto = "HTTP/1.1"
303			httpRequest.Host = request.Domain
304			debuglog("> %s %s %s", httpRequest.Method, httpRequest.URL.RequestURI(), httpRequest.Proto)
305			debuglog("> Host: %s", httpRequest.Host)
306			for key, value := range request.Headers {
307			if value != "" {
308			debuglog("> %s: %s", key, value)
309			httpRequest.Header[key] = []string{value}
310			}
311			}
312			debuglog(">")
313			httpClient := &http.Client{}
314			httpClient.Timeout = time.Duration(runtime.ReadTimeout) * time.Second
315			proxy := &url.URL{}
316			if runtime.Proxy != "" {
317			proxy, err = url.Parse(runtime.Proxy)
318			if err != nil {
319			return
320			}
321			}
322			trans := &http.Transport{}
323			if proxy != nil && runtime.Proxy != "" {
324			trans.Proxy = http.ProxyURL(proxy)
325			}
326			trans.DialContext = utils.Timeout(time.Duration(runtime.ConnectTimeout) * time.Second)
327			httpClient.Transport = trans
328			httpResponse, err := hookDo(httpClient.Do)(httpRequest)
329			if err != nil {
330			return
331			}
332			debuglog("< %s %s", httpResponse.Proto, httpResponse.Status)
333			for key, value := range httpResponse.Header {
334			debuglog("< %s: %v", key, strings.Join(value, ""))
335			}
336			debuglog("<")
337
338			resp := &response.CommonResponse{}
339			err = hookParse(resp.ParseFromHTTPResponse(httpResponse))
340			if err != nil {
341			return
342			}
343			debuglog("%s", resp.GetHTTPContentString())
344			if resp.GetHTTPStatus() != http.StatusOK {
345			err = fmt.Errorf("httpStatus: %d, message = %s", resp.GetHTTPStatus(), resp.GetHTTPContentString())
346			return
347			}
348			return resp.GetHTTPContentBytes(), nil
349			}
350

aliyun / credentials-go

GitHub Access Token became invalid

Pull Request — master (#25)

credentials.*Config.SetConnectTimeout A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like