Browse Source

move trans related func to trans.go

Xiang Li 12 years ago
parent
commit
3940196de0
2 changed files with 27 additions and 27 deletions
  1. 0 27
      etcd.go
  2. 27 0
      transporter.go

+ 0 - 27
etcd.go

@@ -7,9 +7,7 @@ import (
 	"flag"
 	"github.com/coreos/etcd/store"
 	"github.com/coreos/etcd/web"
-
 	"io/ioutil"
-	"net"
 	"net/http"
 	"net/url"
 	"os"
@@ -225,31 +223,6 @@ func main() {
 
 }
 
-// Create transporter using by raft server
-// Create http or https transporter based on
-// whether the user give the server cert and key
-func newTransporter(scheme string, tlsConf tls.Config) transporter {
-	t := transporter{}
-
-	tr := &http.Transport{
-		Dial: dialTimeout,
-	}
-
-	if scheme == "https" {
-		tr.TLSClientConfig = &tlsConf
-		tr.DisableCompression = true
-	}
-
-	t.client = &http.Client{Transport: tr}
-
-	return t
-}
-
-// Dial with timeout
-func dialTimeout(network, addr string) (net.Conn, error) {
-	return net.DialTimeout(network, addr, HTTPTimeout)
-}
-
 // Start to listen and response client command
 func startEtcdTransport(info Info, scheme string, tlsConf tls.Config) {
 	u, err := url.Parse(info.EtcdURL)

+ 27 - 0
transporter.go

@@ -2,10 +2,12 @@ package main
 
 import (
 	"bytes"
+	"crypto/tls"
 	"encoding/json"
 	"fmt"
 	"github.com/coreos/go-raft"
 	"io"
+	"net"
 	"net/http"
 )
 
@@ -14,6 +16,31 @@ type transporter struct {
 	client *http.Client
 }
 
+// Create transporter using by raft server
+// Create http or https transporter based on
+// whether the user give the server cert and key
+func newTransporter(scheme string, tlsConf tls.Config) transporter {
+	t := transporter{}
+
+	tr := &http.Transport{
+		Dial: dialTimeout,
+	}
+
+	if scheme == "https" {
+		tr.TLSClientConfig = &tlsConf
+		tr.DisableCompression = true
+	}
+
+	t.client = &http.Client{Transport: tr}
+
+	return t
+}
+
+// Dial with timeout
+func dialTimeout(network, addr string) (net.Conn, error) {
+	return net.DialTimeout(network, addr, HTTPTimeout)
+}
+
 // Sends AppendEntries RPCs to a peer when the server is the leader.
 func (t transporter) SendAppendEntriesRequest(server *raft.Server, peer *raft.Peer, req *raft.AppendEntriesRequest) *raft.AppendEntriesResponse {
 	var aersp *raft.AppendEntriesResponse