// set any missing global configuration on a new ServiceConfig. // ServiceRegistry *must* be locked. func (s *ServiceRegistry) setServiceDefaults(svc *client.ServiceConfig) { if svc.Balance == "" && s.cfg.Balance != "" { svc.Balance = s.cfg.Balance } if svc.CheckInterval == 0 && s.cfg.CheckInterval != 0 { svc.CheckInterval = s.cfg.CheckInterval } if svc.Fall == 0 && s.cfg.Fall != 0 { svc.Fall = s.cfg.Fall } if svc.Rise == 0 && s.cfg.Rise != 0 { svc.Rise = s.cfg.Rise } if svc.ClientTimeout == 0 && s.cfg.ClientTimeout != 0 { svc.ClientTimeout = s.cfg.ClientTimeout } if svc.ServerTimeout == 0 && s.cfg.ServerTimeout != 0 { svc.ServerTimeout = s.cfg.ServerTimeout } if svc.DialTimeout == 0 && s.cfg.DialTimeout != 0 { svc.DialTimeout = s.cfg.DialTimeout } if s.cfg.HTTPSRedirect { svc.HTTPSRedirect = true } }
func (s *HTTPSuite) TestUpdateServiceDefaults(c *C) { svcCfg := client.ServiceConfig{ Name: "TestService", Addr: "127.0.0.1:9000", Backends: []client.BackendConfig{ client.BackendConfig{ Name: "Backend1", Addr: "127.0.0.1:9001", }, }, } svcDef := bytes.NewBuffer(svcCfg.Marshal()) req, _ := http.NewRequest("PUT", s.httpSvr.URL+"/TestService", svcDef) resp, err := http.DefaultClient.Do(req) if err != nil { c.Fatal(err) } resp.Body.Close() // Now update the Service in-place svcCfg.ServerTimeout = 1234 svcDef.Reset() svcDef.Write(svcCfg.Marshal()) req, _ = http.NewRequest("PUT", s.httpSvr.URL+"/TestService", svcDef) resp, err = http.DefaultClient.Do(req) if err != nil { c.Fatal(err) } body, _ := ioutil.ReadAll(resp.Body) resp.Body.Close() config := client.Config{} err = json.Unmarshal(body, &config) if err != nil { c.Fatal(err) } // make sure we don't see a second value found := false for _, svc := range config.Services { if svc.Name == "TestService" { if svc.ServerTimeout != svcCfg.ServerTimeout { c.Fatal("Service not updated") } else if found { c.Fatal("Multiple Service Definitions") } found = true } } }