luxolus
/
open-nomad
2
0
Fork 0
Code Issues 1 Pull Requests Packages Releases Wiki Activity

Remove unnecessary parameter from NewHTTPServer

This commit is contained in:
Adam Stankiewicz 2017-04-10 14:15:40 +02:00
parent d2a64d3fd3
commit 4daf4cb8c9
No known key found for this signature in database
GPG Key ID: A62480DCEAC884DF
3 changed files with 6 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
command/agent/command.go
View File

@ -350,7 +350,7 @@ func (c *Command) setupAgent(config *Config, logOutput io.Writer) error {
}
// Setup the HTTP server
http, err := NewHTTPServer(agent, config, logOutput)
http, err := NewHTTPServer(agent, config)
if err != nil {
agent.Shutdown()
c.Ui.Error(fmt.Sprintf("Error starting http server: %s", err))

3
command/agent/http.go
View File

@ -5,7 +5,6 @@ import (
"crypto/tls"
"encoding/json"
"fmt"
"io"
"log"
"net"
"net/http"
@ -52,7 +51,7 @@ type HTTPServer struct {
}
// NewHTTPServer starts new HTTP server over the agent
func NewHTTPServer(agent *Agent, config *Config, logOutput io.Writer) (*HTTPServer, error) {
func NewHTTPServer(agent *Agent, config *Config) (*HTTPServer, error) {
// Start the listener
lnAddr, err := net.ResolveTCPAddr("tcp", config.normalizedAddrs.HTTP)
if err != nil {

22
command/agent/http_test.go
View File

@ -31,25 +31,11 @@ func (s *TestServer) Cleanup() {
os.RemoveAll(s.Dir)
}
// makeHTTPServerNoLogs returns a test server with full logging.
func makeHTTPServer(t testing.TB, cb func(c *Config)) *TestServer {
return makeHTTPServerWithWriter(t, nil, cb)
}
// makeHTTPServerNoLogs returns a test server which only prints agent logs and
// no http server logs
func makeHTTPServerNoLogs(t testing.TB, cb func(c *Config)) *TestServer {
return makeHTTPServerWithWriter(t, ioutil.Discard, cb)
}
// makeHTTPServerWithWriter returns a test server whose logs will be written to
// makeHTTPServer returns a test server whose logs will be written to
// the passed writer. If the writer is nil, the logs are written to stderr.
func makeHTTPServerWithWriter(t testing.TB, w io.Writer, cb func(c *Config)) *TestServer {
func makeHTTPServer(t testing.TB, cb func(c *Config)) *TestServer {
dir, agent := makeAgent(t, cb)
if w == nil {
w = agent.logOutput
}
srv, err := NewHTTPServer(agent, agent.config, w)
srv, err := NewHTTPServer(agent, agent.config)
if err != nil {
t.Fatalf("err: %v", err)
}
@ -63,7 +49,7 @@ func makeHTTPServerWithWriter(t testing.TB, w io.Writer, cb func(c *Config)) *Te
}
func BenchmarkHTTPRequests(b *testing.B) {
s := makeHTTPServerNoLogs(b, func(c *Config) {
s := makeHTTPServer(b, func(c *Config) {
c.Client.Enabled = false
})
defer s.Cleanup()