@@ -37,7 +37,6 @@ public class DatabricksSdkClient implements DatabricksClient {
3737 private static final Logger LOGGER = LoggerFactory .getLogger (DatabricksSdkClient .class );
3838 private static final String SYNC_TIMEOUT_VALUE = "10s" ;
3939 private static final String ASYNC_TIMEOUT_VALUE = "0s" ;
40- private int statementResultPollIntervalMillis = 200 ;
4140
4241 private final IDatabricksConnectionContext connectionContext ;
4342 private final DatabricksConfig databricksConfig ;
@@ -61,7 +60,6 @@ public DatabricksSdkClient(IDatabricksConnectionContext connectionContext)
6160
6261 OAuthAuthenticator authenticator = new OAuthAuthenticator (connectionContext );
6362 this .workspaceClient = authenticator .getWorkspaceClient ();
64- this .statementResultPollIntervalMillis = connectionContext .getAsyncExecPollInterval ();
6563 }
6664
6765 public DatabricksSdkClient (
@@ -78,8 +76,6 @@ public DatabricksSdkClient(
7876 this .workspaceClient =
7977 new WorkspaceClient (true /* mock */ , apiClient )
8078 .withStatementExecutionImpl (statementExecutionService );
81-
82- this .statementResultPollIntervalMillis = connectionContext .getAsyncExecPollInterval ();
8379 }
8480
8581 @ Override
@@ -163,7 +159,7 @@ public DatabricksResultSet executeStatement(
163159 while (responseState == StatementState .PENDING || responseState == StatementState .RUNNING ) {
164160 if (pollCount > 0 ) { // First poll happens without a delay
165161 try {
166- Thread .sleep (statementResultPollIntervalMillis );
162+ Thread .sleep (this . connectionContext . getAsyncExecPollInterval () );
167163 } catch (InterruptedException e ) {
168164 throw new DatabricksTimeoutException ("Thread interrupted due to statement timeout" );
169165 }
0 commit comments