Expose dbsql.connOption type

connector.go

@@ -50,7 +50,6 @@
 		},
 		CanUseMultipleCatalogs: &c.cfg.CanUseMultipleCatalogs,
 	})
-
 	if err != nil {
 		return nil, dbsqlerrint.NewRequestError(ctx, fmt.Sprintf("error connecting: host=%s port=%d, httpPath=%s", c.cfg.Host, c.cfg.Port, c.cfg.HTTPPath), err)
 	}
@@ -83,11 +82,11 @@
 
 var _ driver.Connector = (*connector)(nil)
 
-type connOption func(*config.Config)
+type ConnOption func(*config.Config)
 
 // NewConnector creates a connection that can be used with `sql.OpenDB()`.
 // This is an easier way to set up the DB instead of having to construct a DSN string.
-func NewConnector(options ...connOption) (driver.Connector, error) {
+func NewConnector(options ...ConnOption) (driver.Connector, error) {
 	// config with default options
 	cfg := config.WithDefaults()
 	cfg.DriverVersion = DriverVersion
@@ -101,14 +100,14 @@
 	return &connector{cfg: cfg, client: client}, nil
 }
 
-func withUserConfig(ucfg config.UserConfig) connOption {
+func withUserConfig(ucfg config.UserConfig) ConnOption {
 	return func(c *config.Config) {
 		c.UserConfig = ucfg
 	}
 }
 
 // WithServerHostname sets up the server hostname. Mandatory.
-func WithServerHostname(host string) connOption {
+func WithServerHostname(host string) ConnOption {
 	return func(c *config.Config) {
 		protocol, hostname := parseHostName(host)
 		if protocol != "" {
@@ -142,7 +141,7 @@
 }
 
 // WithPort sets up the server port. Mandatory.
-func WithPort(port int) connOption {
+func WithPort(port int) ConnOption {
 	return func(c *config.Config) {
 		c.Port = port
 	}
@@ -152,7 +151,7 @@
 // By default retryWaitMin = 1 * time.Second
 // By default retryWaitMax = 30 * time.Second
 // By default retryMax = 4
-func WithRetries(retryMax int, retryWaitMin time.Duration, retryWaitMax time.Duration) connOption {
+func WithRetries(retryMax int, retryWaitMin time.Duration, retryWaitMax time.Duration) ConnOption {
 	return func(c *config.Config) {
 		c.RetryWaitMax = retryWaitMax
 		c.RetryWaitMin = retryWaitMin
@@ -161,7 +160,7 @@
 }
 
 // WithAccessToken sets up the Personal Access Token. Mandatory for now.
-func WithAccessToken(token string) connOption {
+func WithAccessToken(token string) ConnOption {
 	return func(c *config.Config) {
 		if token != "" {
 			c.AccessToken = token
@@ -174,7 +173,7 @@
 }
 
 // WithHTTPPath sets up the endpoint to the warehouse. Mandatory.
-func WithHTTPPath(path string) connOption {
+func WithHTTPPath(path string) ConnOption {
 	return func(c *config.Config) {
 		if !strings.HasPrefix(path, "/") {
 			path = "/" + path
@@ -184,7 +183,7 @@
 }
 
 // WithMaxRows sets up the max rows fetched per request. Default is 10000
-func WithMaxRows(n int) connOption {
+func WithMaxRows(n int) ConnOption {
 	return func(c *config.Config) {
 		if n != 0 {
 			c.MaxRows = n
@@ -193,31 +192,31 @@
 }
 
 // WithTimeout adds timeout for the server query execution. Default is no timeout.
-func WithTimeout(n time.Duration) connOption {
+func WithTimeout(n time.Duration) ConnOption {
 	return func(c *config.Config) {
 		c.QueryTimeout = n
 	}
 }
 
 // Sets the initial catalog name and schema name in the session.
 // Use <select * from foo> instead of <select * from catalog.schema.foo>
-func WithInitialNamespace(catalog, schema string) connOption {
+func WithInitialNamespace(catalog, schema string) ConnOption {
 	return func(c *config.Config) {
 		c.Catalog = catalog
 		c.Schema = schema
 	}
 }
 
 // Used to identify partners. Set as a string with format <isv-name+product-name>.
-func WithUserAgentEntry(entry string) connOption {
+func WithUserAgentEntry(entry string) ConnOption {
 	return func(c *config.Config) {
 		c.UserAgentEntry = entry
 	}
 }
 
 // Sessions params will be set upon opening the session by calling SET function.
 // If using connection pool, session params can avoid successive calls of "SET ..."
-func WithSessionParams(params map[string]string) connOption {
+func WithSessionParams(params map[string]string) ConnOption {
 	return func(c *config.Config) {
 		for k, v := range params {
 			if strings.ToLower(k) == "timezone" {
@@ -226,43 +225,42 @@
 				} else {
 					c.Location = loc
 				}
-
 			}
 		}
 		c.SessionParams = params
 	}
 }
 
 // WithAuthenticator sets up the Authentication. Mandatory if access token is not provided.
-func WithAuthenticator(authr auth.Authenticator) connOption {
+func WithAuthenticator(authr auth.Authenticator) ConnOption {
 	return func(c *config.Config) {
 		c.Authenticator = authr
 	}
 }
 
 // WithTransport sets up the transport configuration to be used by the httpclient.
-func WithTransport(t http.RoundTripper) connOption {
+func WithTransport(t http.RoundTripper) ConnOption {
 	return func(c *config.Config) {
 		c.Transport = t
 	}
 }
 
 // WithCloudFetch sets up the use of cloud fetch for query execution. Default is false.
-func WithCloudFetch(useCloudFetch bool) connOption {
+func WithCloudFetch(useCloudFetch bool) ConnOption {
 	return func(c *config.Config) {
 		c.UseCloudFetch = useCloudFetch
 	}
 }
 
 // WithMaxDownloadThreads sets up maximum download threads for cloud fetch. Default is 10.
-func WithMaxDownloadThreads(numThreads int) connOption {
+func WithMaxDownloadThreads(numThreads int) ConnOption {
 	return func(c *config.Config) {
 		c.MaxDownloadThreads = numThreads
 	}
 }
 
 // Setup of Oauth M2m authentication
-func WithClientCredentials(clientID, clientSecret string) connOption {
+func WithClientCredentials(clientID, clientSecret string) ConnOption {
 	return func(c *config.Config) {
 		if clientID != "" && clientSecret != "" {
 			authr := m2m.NewAuthenticator(clientID, clientSecret, c.Host)