@@ -3,29 +3,35 @@ package grpcclient
3
3
import (
4
4
"context"
5
5
"crypto/x509"
6
+ "github.com/hduhelp/api-consulting/config"
7
+ "google.golang.org/grpc/credentials"
6
8
"log"
7
9
8
10
"go.uber.org/zap"
9
11
"google.golang.org/grpc"
10
- "google.golang.org/grpc/credentials"
11
12
"google.golang.org/grpc/metadata"
12
13
)
13
14
14
- var defaultEndpoint = "gapi .hduhelp.com:443 "
15
+ var defaultEndpoint = "gateway-service .hduhelp:9101 "
15
16
16
- func Conn (ctx context.Context , endpoints ... string ) * grpc.ClientConn {
17
+ func Conn (ctx context.Context , endpoints ... string ) ( conn * grpc.ClientConn ) {
17
18
var endpoint string
18
19
if len (endpoints ) != 0 {
19
20
endpoint = endpoints [0 ]
20
21
} else {
21
22
endpoint = defaultEndpoint
22
23
}
23
- certPool , err := x509 .SystemCertPool ()
24
- if err != nil {
25
- log .Fatalf ("failed to load credentials: %v" , err )
24
+ var err error
25
+ if ! config .IsProd () {
26
+ certPool , err := x509 .SystemCertPool ()
27
+ if err != nil {
28
+ log .Fatalf ("failed to load credentials: %v" , err )
29
+ }
30
+ creds := credentials .NewClientTLSFromCert (certPool , "127.0.0.1" )
31
+ conn , err = grpc .DialContext (ctx , endpoint , grpc .WithTransportCredentials (creds ))
32
+ } else {
33
+ conn , err = grpc .DialContext (ctx , endpoint , grpc .WithInsecure ())
26
34
}
27
- creds := credentials .NewClientTLSFromCert (certPool , "gapi.hduhelp.com" )
28
- conn , err := grpc .DialContext (ctx , endpoint , grpc .WithTransportCredentials (creds ))
29
35
if err != nil || conn == nil {
30
36
log .Fatal ("grpc client did not connect" , zap .Error (err ))
31
37
}
0 commit comments