Retry on deadlock in SqlServerClient.getTableHandle

ebyhr · ebyhr · commit e3a8127201fc · 2025-11-19T08:07:14.000+09:00
diff --git a/plugin/trino-sqlserver/src/main/java/io/trino/plugin/sqlserver/SqlServerClient.java b/plugin/trino-sqlserver/src/main/java/io/trino/plugin/sqlserver/SqlServerClient.java
@@ -443,6 +443,12 @@ public Iterator<RelationColumnsMetadata> getAllTableColumns(ConnectorSession ses
         return retryOnDeadlock(() -> super.getAllTableColumns(session, schema), "error when getting all table columns for '%s'".formatted(schema));
     }
 
+    @Override
+    public Optional<JdbcTableHandle> getTableHandle(ConnectorSession session, SchemaTableName schemaTableName)
+    {
+        return retryOnDeadlock(() -> super.getTableHandle(session, schemaTableName), "error when getting table handle for '%s'".formatted(schemaTableName));
+    }
+
     @Override
     public List<JdbcColumnHandle> getColumns(ConnectorSession session, SchemaTableName schemaTableName, RemoteTableName remoteTableName)
     {

Original file line number	Diff line number	Diff line change
`@@ -443,6 +443,12 @@ public Iterator<RelationColumnsMetadata> getAllTableColumns(ConnectorSession ses`
`443`	`443`	`return retryOnDeadlock(() -> super.getAllTableColumns(session, schema), "error when getting all table columns for '%s'".formatted(schema));`
`444`	`444`	`}`
`445`	`445`
	`446`	`+ @Override`
	`447`	`+ public Optional<JdbcTableHandle> getTableHandle(ConnectorSession session, SchemaTableName schemaTableName)`
	`448`	`+ {`
	`449`	`+ return retryOnDeadlock(() -> super.getTableHandle(session, schemaTableName), "error when getting table handle for '%s'".formatted(schemaTableName));`
	`450`	`+ }`
	`451`	`+`
`446`	`452`	`@Override`
`447`	`453`	`public List<JdbcColumnHandle> getColumns(ConnectorSession session, SchemaTableName schemaTableName, RemoteTableName remoteTableName)`
`448`	`454`	`{`