Use probability parameter

m-lab · Jul 10, 2024 · 5986342 · 5986342
1 parent 2d354a7
commit 5986342
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/handler/handler.go b/handler/handler.go
@@ -198,10 +198,10 @@ func (s *Server) Register(rw http.ResponseWriter, req *http.Request) {
 	}
 	param.Geo = record
 	param.Network = s.ASN.AnnotateIP(param.IPv4)
-	r := register.CreateRegisterResponse(param)
 	// Override site probability with user-provided parameter.
 	// TODO(soltesz): include M-Lab override option
-	r.Registration.Heartbeat.Probability = getProbability(req)
+	param.Probability = getProbability(req)
+	r := register.CreateRegisterResponse(param)
 
 	// Register the hostname under the organization zone.
 	m := dnsx.NewManager(s.DNS, s.Project, register.OrgZone(param.Org, s.Project))