4242
4343import io .airlift .log .Logger ;
4444import io .airlift .slice .Slice ;
45+ import io .trino .spi .StandardErrorCode ;
46+ import io .trino .spi .TrinoException ;
4547import io .trino .spi .connector .AggregateFunction ;
4648import io .trino .spi .connector .AggregationApplicationResult ;
4749import io .trino .spi .connector .Assignment ;
6062import io .trino .spi .connector .ConstraintApplicationResult ;
6163import io .trino .spi .connector .LimitApplicationResult ;
6264import io .trino .spi .connector .NotFoundException ;
65+ import io .trino .spi .connector .RetryMode ;
6366import io .trino .spi .connector .SchemaTableName ;
6467import io .trino .spi .connector .SchemaTablePrefix ;
6568import io .trino .spi .connector .TableNotFoundException ;
@@ -182,7 +185,10 @@ public void dropColumn(ConnectorSession session, ConnectorTableHandle tableHandl
182185
183186 @ Override
184187 public ConnectorOutputTableHandle beginCreateTable (ConnectorSession session , ConnectorTableMetadata tableMetadata ,
185- Optional <ConnectorTableLayout > layout ) {
188+ Optional <ConnectorTableLayout > layout , RetryMode retryMode ) {
189+ if (retryMode != RetryMode .NO_RETRIES ) {
190+ throw new TrinoException (StandardErrorCode .NOT_SUPPORTED , "This connector does not support query retries" );
191+ }
186192 List <RediSearchColumnHandle > columns = buildColumnHandles (tableMetadata );
187193
188194 rediSearchSession .createTable (tableMetadata .getTable (), columns );
@@ -202,7 +208,11 @@ public Optional<ConnectorOutputMetadata> finishCreateTable(ConnectorSession sess
202208 }
203209
204210 @ Override
205- public ConnectorInsertTableHandle beginInsert (ConnectorSession session , ConnectorTableHandle tableHandle ) {
211+ public ConnectorInsertTableHandle beginInsert (ConnectorSession session , ConnectorTableHandle tableHandle ,
212+ List <ColumnHandle > insertedColumns , RetryMode retryMode ) {
213+ if (retryMode != RetryMode .NO_RETRIES ) {
214+ throw new TrinoException (StandardErrorCode .NOT_SUPPORTED , "This connector does not support query retries" );
215+ }
206216 RediSearchTableHandle table = (RediSearchTableHandle ) tableHandle ;
207217 List <RediSearchColumnHandle > columns = rediSearchSession .getTable (table .getSchemaTableName ()).getColumns ();
208218
0 commit comments