Ivan7702 12 lat temu
rodzic
commit
90d7ebec47
3 zmienionych plików z 10 dodań i 10 usunięć
  1. 2 2
      raft_server.go
  2. 7 7
      transporter.go
  3. 1 1
      transporter_test.go

+ 2 - 2
raft_server.go

@@ -31,7 +31,7 @@ func newRaftServer(name string, url string, listenHost string, tlsConf *TLSConfi
 
 
 	// Create transporter for raft
 	// Create transporter for raft
 	raftTransporter := newTransporter(tlsConf.Scheme, tlsConf.Client, ElectionTimeout)
 	raftTransporter := newTransporter(tlsConf.Scheme, tlsConf.Client, ElectionTimeout)
-	
+
 	// Create raft server
 	// Create raft server
 	server, err := raft.NewServer(name, dirPath, raftTransporter, etcdStore, nil)
 	server, err := raft.NewServer(name, dirPath, raftTransporter, etcdStore, nil)
 
 
@@ -198,7 +198,7 @@ func joinCluster(cluster []string) bool {
 			if _, ok := err.(etcdErr.Error); ok {
 			if _, ok := err.(etcdErr.Error); ok {
 				fatal(err)
 				fatal(err)
 			}
 			}
-			
+
 			debugf("cannot join to cluster via machine %s %s", machine, err)
 			debugf("cannot join to cluster via machine %s %s", machine, err)
 		}
 		}
 	}
 	}

+ 7 - 7
transporter.go

@@ -21,7 +21,7 @@ type transporter struct {
 // response struct
 // response struct
 type transporterResponse struct {
 type transporterResponse struct {
 	resp *http.Response
 	resp *http.Response
-	err error
+	err  error
 }
 }
 
 
 // Create transporter using by raft server
 // Create transporter using by raft server
@@ -162,11 +162,11 @@ func (t *transporter) SendSnapshotRecoveryRequest(server *raft.Server, peer *raf
 func (t *transporter) Post(path string, body io.Reader) (*http.Response, error) {
 func (t *transporter) Post(path string, body io.Reader) (*http.Response, error) {
 
 
 	c := make(chan *transporterResponse, 1)
 	c := make(chan *transporterResponse, 1)
-	
+
 	go func() {
 	go func() {
 		tr := new(transporterResponse)
 		tr := new(transporterResponse)
 		tr.resp, tr.err = t.client.Post(path, "application/json", body)
 		tr.resp, tr.err = t.client.Post(path, "application/json", body)
-		c <-tr
+		c <- tr
 	}()
 	}()
 
 
 	return t.waitResponse(c)
 	return t.waitResponse(c)
@@ -177,11 +177,11 @@ func (t *transporter) Post(path string, body io.Reader) (*http.Response, error)
 func (t *transporter) Get(path string) (*http.Response, error) {
 func (t *transporter) Get(path string) (*http.Response, error) {
 
 
 	c := make(chan *transporterResponse, 1)
 	c := make(chan *transporterResponse, 1)
-	
+
 	go func() {
 	go func() {
 		tr := new(transporterResponse)
 		tr := new(transporterResponse)
 		tr.resp, tr.err = t.client.Get(path)
 		tr.resp, tr.err = t.client.Get(path)
-		c <-tr
+		c <- tr
 	}()
 	}()
 
 
 	return t.waitResponse(c)
 	return t.waitResponse(c)
@@ -195,10 +195,10 @@ func (t *transporter) waitResponse(responseChan chan *transporterResponse) (*htt
 	case <-timeoutChan:
 	case <-timeoutChan:
 		return nil, fmt.Errorf("Wait Response Timeout: %v", t.timeout)
 		return nil, fmt.Errorf("Wait Response Timeout: %v", t.timeout)
 
 
-	case r:= <-responseChan:
+	case r := <-responseChan:
 		return r.resp, r.err
 		return r.resp, r.err
 	}
 	}
 
 
 	// for complier
 	// for complier
 	return nil, nil
 	return nil, nil
-}
+}

+ 1 - 1
transporter_test.go

@@ -15,7 +15,7 @@ func TestTransporterTimeout(t *testing.T) {
 	ts.Get("http://google.com")
 	ts.Get("http://google.com")
 	_, err := ts.Get("http://google.com:9999") // it doesn't exisit
 	_, err := ts.Get("http://google.com:9999") // it doesn't exisit
 	if err == nil || err.Error() != "Wait Response Timeout: 1s" {
 	if err == nil || err.Error() != "Wait Response Timeout: 1s" {
-		t.Fatal("timeout error: ", err.Error( ))
+		t.Fatal("timeout error: ", err.Error())
 	}
 	}
 
 
 	_, err = ts.Post("http://google.com:9999", nil) // it doesn't exisit
 	_, err = ts.Post("http://google.com:9999", nil) // it doesn't exisit