feature: make couchbase initial connection configurable (#32)

Co-authored-by: oguzhanyildirim <oguzhan.yildirim@trendyol.com>

README.md

@@ -75,6 +75,7 @@ $ go get github.com/Trendyol/go-dcp-client
 | `collectionNames`                     | []string          | no       | _default  |
 | `dcp.bufferSizeKb`                    | int               | no       | 16384     |
 | `dcp.connectionBufferSizeKb`          | uint              | no       | 20480     |
+| `dcp.connectionTimeout`               | time.Duration     | no       | 5s        |
 | `dcp.listener.bufferSize`             | int               | no       | 1         |
 | `dcp.group.membership.memberNumber`   | int               | no       | 1         |
 | `dcp.group.membership.totalMembers`   | int               | no       | 1         |

dcp/client.go

@@ -141,7 +141,7 @@ func (s *client) connect(bucketName string) (*gocbcore.Agent, error) {
 	ch := make(chan error)
 
 	_, err = client.WaitUntilReady(
-		time.Now().Add(time.Second*5),
+		time.Now().Add(s.config.Dcp.ConnectionTimeout),
 		gocbcore.WaitUntilReadyOptions{
 			RetryStrategy: gocbcore.NewBestEffortRetryStrategy(nil),
 		},

helpers/config.go

@@ -29,6 +29,7 @@ type ConfigDCP struct {
 	Group                  ConfigDCPGroup    `yaml:"group"`
 	BufferSizeKb           int               `yaml:"bufferSizeKb" default:"16384"`
 	ConnectionBufferSizeKb uint              `yaml:"connectionBufferSizeKb" default:"20480"`
+	ConnectionTimeout      time.Duration     `yaml:"connectionTimeout"`
 	Listener               ConfigDCPListener `yaml:"listener"`
 }
 
@@ -184,6 +185,10 @@ func applyUnhandledDefaults(_config *Config) {
 		_config.Dcp.Group.Membership.RebalanceDelay = 20 * time.Second
 	}
 
+	if _config.Dcp.ConnectionTimeout == 0 {
+		_config.Dcp.ConnectionTimeout = 5 * time.Second
+	}
+
 	if _config.CollectionNames == nil {
 		_config.CollectionNames = []string{DefaultCollectionName}
 	}