From 6d55850fdff1faff388fb2a0649739ea857bf52f Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Tue, 29 Mar 2016 13:07:43 +0200 Subject: Cleanup a bit --- server.go | 22 ++++++++-------------- 1 file changed, 8 insertions(+), 14 deletions(-) (limited to 'server.go') diff --git a/server.go b/server.go index c0449ee..6f0fe48 100644 --- a/server.go +++ b/server.go @@ -3,7 +3,6 @@ package goxy import ( "crypto/tls" "encoding/gob" - "log" "net/http" "net/http/httputil" "net/url" @@ -12,7 +11,7 @@ import ( ) type Server struct { - Data string + DataFile string Route http.Server } @@ -20,21 +19,16 @@ type Server struct { func NewServer(fname string) (*Server, error) { r := make(Route) s := http.Server{TLSConfig: &tls.Config{GetCertificate: r.GetCertificate}} - server := &Server{Route: r, Server: s, Data: fname} - if err := server.Load(); err != nil { - log.Println(err) - } - if err := server.Restore(); err != nil { - return nil, err - } + server := &Server{Route: r, Server: s, DataFile: fname} + server.Load() Register(server) - return server, nil + return server, server.Restore() } // Save routes to persistent file func (s Server) Save() error { - if s.Data != "" { - fd, err := os.Create(s.Data) + if s.DataFile != "" { + fd, err := os.Create(s.DataFile) if err != nil { return err } @@ -46,8 +40,8 @@ func (s Server) Save() error { // Load routes from persistent file func (s *Server) Load() error { - if s.Data != "" { - fd, err := os.Open(s.Data) + if s.DataFile != "" { + fd, err := os.Open(s.DataFile) if err != nil { return err } -- cgit v1.2.3