add Probers to Probe pkgs.

pkg/probe/tcp/tcp.go

@@ -25,7 +25,13 @@ import (
 	"github.com/golang/glog"
 )
 
-func Probe(host string, port int) (probe.Status, error) {
+func New() TCPProber {
+	return TCPProber{}
+}
+
+type TCPProber struct{}
+
+func (pr TCPProber) Probe(host string, port int) (probe.Status, error) {
 	return DoTCPProbe(net.JoinHostPort(host, strconv.Itoa(port)))
 }
 

pkg/probe/tcp/tcp_test.go

@@ -28,6 +28,7 @@ import (
 )
 
 func TestTcpHealthChecker(t *testing.T) {
+	prober := New()
 	tests := []struct {
 		expectedStatus probe.Status
 		usePort        bool
@@ -58,7 +59,7 @@ func TestTcpHealthChecker(t *testing.T) {
 		if !test.usePort {
 			p = -1
 		}
-		status, err := Probe(host, p)
+		status, err := prober.Probe(host, p)
 		if status != test.expectedStatus {
 			t.Errorf("expected: %v, got: %v", test.expectedStatus, status)
 		}