session: expose the ProbeASN as an integer (#74)

ooni · Oct 2, 2019 · 20fcd34 · 20fcd34
1 parent ddc9bae
commit 20fcd34
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/session/session.go b/session/session.go
@@ -139,11 +139,16 @@ func (s *Session) AddAvailableHTTPSCollector(baseURL string) {
 
 // ProbeASNString returns the probe ASN as a string.
 func (s *Session) ProbeASNString() string {
+	return fmt.Sprintf("AS%d", s.ProbeASN())
+}
+
+// ProbeASN returns the probe ASN as an integer.
+func (s *Session) ProbeASN() uint {
 	asn := model.DefaultProbeASN
 	if s.location != nil {
 		asn = s.location.ASN
 	}
-	return fmt.Sprintf("AS%d", asn)
+	return asn
 }
 
 // ProbeCC returns the probe CC.