Add transport override in mns (#18)

Author: sandeepvinayak

client.go

@@ -56,7 +56,7 @@ const (
 type MNSClient interface {
 	Send(method Method, headers map[string]string, message interface{}, resource string) (*fasthttp.Response, error)
 	SetProxy(url string)
-
+	SetTransport(transport fasthttp.RoundTripper)
 	getAccountID() (accountId string)
 	getRegion() (region string)
 }
@@ -175,6 +175,13 @@ func (p *aliMNSClient) initFastHttpClient() {
 	p.client = &fasthttp.Client{ReadTimeout: timeout, WriteTimeout: timeout, Name: ClientName}
 }
 
+func (p *aliMNSClient) SetTransport(transport fasthttp.RoundTripper) {
+	p.client.ConfigureClient = func(hc *fasthttp.HostClient) error {
+		hc.Transport = transport
+		return nil
+	}
+}
+
 func (p *aliMNSClient) proxy(req *http.Request) (*neturl.URL, error) {
 	if p.proxyURL != "" {
 		return neturl.Parse(p.proxyURL)

go.mod

@@ -0,0 +1,18 @@
+module git.soma.salesforce.com/chameleon/aliyun-mns-go-sdk
+
+go 1.20
+
+require (
+	github.com/aliyun/aliyun-mns-go-sdk v1.0.2
+	github.com/gogap/errors v0.0.0-20210818113853-edfbba0ddea9
+	github.com/gogap/logs v0.0.0-20150329044033-31c6d1e28b2c
+	github.com/valyala/fasthttp v1.52.0
+)
+
+require (
+	github.com/andybalholm/brotli v1.1.0 // indirect
+	github.com/facebookgo/stack v0.0.0-20160209184415-751773369052 // indirect
+	github.com/gogap/stack v0.0.0-20150131034635-fef68dddd4f8 // indirect
+	github.com/klauspost/compress v1.17.6 // indirect
+	github.com/valyala/bytebufferpool v1.0.0 // indirect
+)

go.sum

@@ -0,0 +1,18 @@
+github.com/aliyun/aliyun-mns-go-sdk v1.0.2 h1:dq2AwayUe1QrMXVEGTBhaoQ61UI3cHju+p2Lq3Q+HCc=
+github.com/aliyun/aliyun-mns-go-sdk v1.0.2/go.mod h1:eD/mEH7SwtLSwI9p8fP9VTH2cYM3wFSY1WNaxEdLIFU=
+github.com/andybalholm/brotli v1.1.0 h1:eLKJA0d02Lf0mVpIDgYnqXcUn0GqVmEFny3VuID1U3M=
+github.com/andybalholm/brotli v1.1.0/go.mod h1:sms7XGricyQI9K10gOSf56VKKWS4oLer58Q+mhRPtnY=
+github.com/facebookgo/stack v0.0.0-20160209184415-751773369052 h1:JWuenKqqX8nojtoVVWjGfOF9635RETekkoH6Cc9SX0A=
+github.com/facebookgo/stack v0.0.0-20160209184415-751773369052/go.mod h1:UbMTZqLaRiH3MsBH8va0n7s1pQYcu3uTb8G4tygF4Zg=
+github.com/gogap/errors v0.0.0-20210818113853-edfbba0ddea9 h1:qvGIRaCYFKkyFK9SgRXJCc/lmQCeeg2cl3mwBKQd5W0=
+github.com/gogap/errors v0.0.0-20210818113853-edfbba0ddea9/go.mod h1:tbRYYYC7g/H7QlCeX0Z2zaThWKowF4QQCFIsGgAsqRo=
+github.com/gogap/logs v0.0.0-20150329044033-31c6d1e28b2c h1:Tv0OlEQfruT8rH/BoW2RGrC/RVVzE5n0SMbKBr1spHM=
+github.com/gogap/logs v0.0.0-20150329044033-31c6d1e28b2c/go.mod h1:QJ6r1fyvyIC0lwOdQo9Z/AVi9tlemaE7HMGdMbzMQRQ=
+github.com/gogap/stack v0.0.0-20150131034635-fef68dddd4f8 h1:AuxION6c7in+AsPmFjQTUKT6/o1suT8XEEpfU0pWsHA=
+github.com/gogap/stack v0.0.0-20150131034635-fef68dddd4f8/go.mod h1:6q1WEv2BiAO4FSdwLQTJbWQYAn1/qDNJHUGJNXCj9kM=
+github.com/klauspost/compress v1.17.6 h1:60eq2E/jlfwQXtvZEeBUYADs+BwKBWURIY+Gj2eRGjI=
+github.com/klauspost/compress v1.17.6/go.mod h1:/dCuZOvVtNoHsyb+cuJD3itjs3NbnF6KH9zAO4BDxPM=
+github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
+github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
+github.com/valyala/fasthttp v1.52.0 h1:wqBQpxH71XW0e2g+Og4dzQM8pk34aFYlA1Ga8db7gU0=
+github.com/valyala/fasthttp v1.52.0/go.mod h1:hf5C4QnVMkNXMspnsUlfM3WitlgYflyhHYoKol/szxQ=

utils.go

@@ -43,10 +43,14 @@ func send(client MNSClient, decoder MNSDecoder, method Method, headers map[strin
 			}
 
 			if baseResponder, ok := v.(BaseResponder); ok {
+				hostId := ""
+				if resp.RemoteAddr() != nil {
+					hostId = resp.RemoteAddr().String()
+				}
 				baseResponder.SetBaseResponse(BaseResponse{
 					RequestId: string(resp.Header.Peek("x-mns-request-id")),
 					Code:      strconv.Itoa(resp.StatusCode()),
-					HostId:    resp.RemoteAddr().String(),
+					HostId:    hostId,
 				})
 			}
 		}