From 3071e59632c668696f8bf1c6d9e3bc07b61f6391 Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Wed, 30 Mar 2016 11:17:28 +0200 Subject: Cleanup --- rpc.go | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) (limited to 'rpc.go') diff --git a/rpc.go b/rpc.go index 49b9d21..e6a674d 100644 --- a/rpc.go +++ b/rpc.go @@ -10,29 +10,26 @@ func init() { rpc.HandleHTTP() } -func Register(s *Server) { - rpc.Register(&GoXY{s}) +func RegisterRPC(s *Server) error { + return rpc.Register(&GoXY{s}) } func DialRPC(server string) (*rpc.Client, error) { - if server == "" { - server = ":http-alt" - } return rpc.DialHTTP("tcp", server) } // Add adds a new route func (s *GoXY) Add(e Entry, _ *struct{}) error { - defer s.server.Save() + defer s.server.Save(s.server.DataFile) s.server.Route[e.Host] = e - return s.server.Restore() + return s.server.Update() } // Del removes a route func (s *GoXY) Del(host string, _ *struct{}) error { - defer s.server.Save() + defer s.server.Save(s.server.DataFile) delete(s.server.Route, host) - return s.server.Restore() + return s.server.Update() } // Get returns Entry -- cgit v1.2.3