[Snowflake] Fix idle auth error (#906)

clients/snowflake/writes.go

@@ -17,6 +17,7 @@ func (s *Store) Append(tableData *optimization.TableData, _ bool) error {
 	var err error
 	for i := 0; i < maxRetries; i++ {
 		if i > 0 {
+			// TODO: remove
 			if IsAuthExpiredError(err) {
 				slog.Warn("Authentication has expired, will reload the Snowflake store and retry appending", slog.Any("err", err))
 				if connErr := s.reestablishConnection(); connErr != nil {
@@ -49,6 +50,7 @@ func (s *Store) Merge(tableData *optimization.TableData) error {
 	var err error
 	for i := 0; i < maxRetries; i++ {
 		if i > 0 {
+			// TODO: Remove
 			if IsAuthExpiredError(err) {
 				slog.Warn("Authentication has expired, will reload the Snowflake store and retry merging", slog.Any("err", err))
 				if connErr := s.reestablishConnection(); connErr != nil {

lib/config/snowflake.go

@@ -30,8 +30,12 @@ func (s Snowflake) ToConfig() (*gosnowflake.Config, error) {
 		Region:      s.Region,
 		Application: s.Application,
 		Params: map[string]*string{
+			// This parameter will cancel in-progress queries if connectivity is lost.
 			// https://docs.snowflake.com/en/sql-reference/parameters#abort-detached-query
 			"ABORT_DETACHED_QUERY": ptr.ToString("true"),
+			// This parameter must be set to prevent the auth token from expiring after 4 hours.
+			// https://docs.snowflake.com/en/user-guide/session-policies#considerations
+			"CLIENT_SESSION_KEEP_ALIVE": ptr.ToString("true"),
 		},
 	}