@@ -60,12 +60,11 @@ type (
60
60
)
61
61
62
62
var (
63
- HealthzPort string = "" // expose var to be able to operator
63
+ HealthzPort int = 0 // expose var to be able to operator
64
64
HealthzPath string = "" // expose var to be able to operator
65
- metricsPort string = ""
65
+ metricsPort int = 0
66
66
metricsPath string = ""
67
67
68
- defaultAddr string = ":8080"
69
68
timeoutR = 5 * time .Second
70
69
DefaultFuncHealthz HealthzFunc = func () (bool , error ) {
71
70
_ , err := net .DialTimeout ("tcp" , models .HealthzDefaultAddr , timeoutR )
@@ -79,12 +78,12 @@ var (
79
78
func init () {
80
79
// * Healthz
81
80
flag .Bool (models .HealthzFlagName , false , "Enable the healthz server." )
82
- flag .StringVar (& HealthzPort , models .HealthzPortFlagName , models .HealthzDefaultAddr , "Healthz server port." )
81
+ flag .IntVar (& HealthzPort , models .HealthzPortFlagName , int ( models .HealthzDefaultPort ) , "Healthz server port." )
83
82
flag .StringVar (& HealthzPath , models .HealthzPathFlagName , models .HealthzDefaultPath , "Healthz server path." )
84
83
85
84
// * Metrics
86
85
flag .Bool (models .MetricsFlagName , false , "Enable the metrics server." )
87
- flag .StringVar (& metricsPort , models .MetricsPortFlagName , models .MetricsDefaultAddr , "Metrics server port." )
86
+ flag .IntVar (& metricsPort , models .MetricsPortFlagName , int ( models .MetricsDefaultPort ) , "Metrics server port." )
88
87
flag .StringVar (& metricsPath , models .MetricsPathFlagName , models .MetricsDefaultPath , "Metrics server path." )
89
88
}
90
89
@@ -135,13 +134,13 @@ func DisableMetrics() OptionServer {
135
134
136
135
// Function to create a new server for health
137
136
func (a * app ) createHealth () * server {
138
- s := a .new (WithAddr (HealthzPort ))
137
+ s := a .new (WithAddr (fmt . Sprintf ( ":%d" , HealthzPort ) ))
139
138
return s
140
139
}
141
140
142
141
// Function to create a new server for metrics
143
142
func (a * app ) createMetrics () * server {
144
- s := a .new (WithAddr (metricsPort ))
143
+ s := a .new (WithAddr (fmt . Sprintf ( ":%d" , metricsPort ) ))
145
144
s .Config .Get (metricsPath , metrics .Handler ().ServeHTTP )
146
145
return s
147
146
}
@@ -156,7 +155,6 @@ func (a *app) new(opts ...Option) *server {
156
155
// create a new server with default parameters
157
156
s := & server {
158
157
http : & http.Server {
159
- Addr : defaultAddr ,
160
158
ReadTimeout : timeoutR ,
161
159
},
162
160
Config : r ,
@@ -174,9 +172,7 @@ func (a *app) new(opts ...Option) *server {
174
172
func WithAddr (addr string ) Option {
175
173
_ , _ , err := net .SplitHostPort (addr )
176
174
if err != nil {
177
- return func (s * server ) {
178
- s .http .Addr = defaultAddr
179
- }
175
+ panic (fmt .Sprintf ("invalid address: %s" , addr ))
180
176
}
181
177
return func (s * server ) {
182
178
s .http .Addr = addr
0 commit comments