From a159598f36fe70b4935655ca5e9992de5eec6ca6 Mon Sep 17 00:00:00 2001
From: simon <simon@crcms.cn>
Date: Fri, 3 Sep 2021 01:10:24 +0800
Subject: [PATCH] Add grpc,memory,quic transport automatically discover (#2237)

* Add grpc,memory,quic transport automatically discover

* Add grpc,memory,quic transport automatically discover
---
 plugins/transport/grpc/grpc.go     | 6 +++++-
 plugins/transport/memory/memory.go | 5 +++++
 plugins/transport/quic/quic.go     | 5 +++++
 3 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/plugins/transport/grpc/grpc.go b/plugins/transport/grpc/grpc.go
index d1d996bc..47d6c914 100644
--- a/plugins/transport/grpc/grpc.go
+++ b/plugins/transport/grpc/grpc.go
@@ -10,7 +10,7 @@ import (
 	maddr "github.com/asim/go-micro/v3/util/addr"
 	mnet "github.com/asim/go-micro/v3/util/net"
 	mls "github.com/asim/go-micro/v3/util/tls"
-
+	"github.com/asim/go-micro/v3/cmd"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
 
@@ -27,6 +27,10 @@ type grpcTransportListener struct {
 	tls      *tls.Config
 }
 
+func init() {
+	cmd.DefaultTransports["grpc"] = NewTransport
+}
+
 func getTLSConfig(addr string) (*tls.Config, error) {
 	hosts := []string{addr}
 
diff --git a/plugins/transport/memory/memory.go b/plugins/transport/memory/memory.go
index 7abe4005..d0ba4f96 100644
--- a/plugins/transport/memory/memory.go
+++ b/plugins/transport/memory/memory.go
@@ -13,6 +13,7 @@ import (
 	"github.com/asim/go-micro/v3/transport"
 	maddr "github.com/asim/go-micro/v3/util/addr"
 	mnet "github.com/asim/go-micro/v3/util/net"
+	"github.com/asim/go-micro/v3/cmd"
 )
 
 type memorySocket struct {
@@ -53,6 +54,10 @@ type memoryTransport struct {
 	listeners map[string]*memoryListener
 }
 
+func init() {
+	cmd.DefaultTransports["memory"] = NewTransport
+}
+
 func (ms *memorySocket) Recv(m *transport.Message) error {
 	ms.RLock()
 	defer ms.RUnlock()
diff --git a/plugins/transport/quic/quic.go b/plugins/transport/quic/quic.go
index 8d5eb999..1230b1fe 100644
--- a/plugins/transport/quic/quic.go
+++ b/plugins/transport/quic/quic.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/tls"
 	"encoding/gob"
+	"github.com/asim/go-micro/v3/cmd"
 	"time"
 
 	"github.com/asim/go-micro/v3/transport"
@@ -34,6 +35,10 @@ type quicListener struct {
 	opts transport.ListenOptions
 }
 
+func init() {
+	cmd.DefaultTransports["quic"] = NewTransport
+}
+
 func (q *quicClient) Close() error {
 	return q.quicSocket.st.Close()
 }