支持HTTPS的API节点

internal/rpc/rpc_client.go

@@ -2,6 +2,7 @@ package rpc
 
 import (
 	"context"
+	"crypto/tls"
 	"encoding/base64"
 	"errors"
 	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
@@ -12,7 +13,9 @@ import (
 	"github.com/iwind/TeaGo/maps"
 	"github.com/iwind/TeaGo/rands"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/metadata"
+	"net/url"
 	"time"
 )
 
@@ -28,7 +31,20 @@ func NewRPCClient(apiConfig *configs.APIConfig) (*RPCClient, error) {
 
 	conns := []*grpc.ClientConn{}
 	for _, endpoint := range apiConfig.RPC.Endpoints {
-		conn, err := grpc.Dial(endpoint, grpc.WithInsecure())
+		u, err := url.Parse(endpoint)
+		if err != nil {
+			return nil, errors.New("parse endpoint failed: " + err.Error())
+		}
+		var conn *grpc.ClientConn
+		if u.Scheme == "http" {
+			conn, err = grpc.Dial(u.Host, grpc.WithInsecure())
+		} else if u.Scheme == "https" {
+			conn, err = grpc.Dial(u.Host, grpc.WithTransportCredentials(credentials.NewTLS(&tls.Config{
+				InsecureSkipVerify: true,
+			})))
+		} else {
+			return nil, errors.New("parse endpoint failed: invalid scheme '" + u.Scheme + "'")
+		}
 		if err != nil {
 			return nil, err
 		}