This is an automated email from the ASF dual-hosted git repository.

snuyanzin pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/flink-connector-hive.git

commit 74c0ad0d27d80847ced3c8687a658d85b205c30d
Author: Wencong Liu <liuwencle...@163.com>
AuthorDate: Fri Dec 15 10:25:32 2023 +0800

    [FLINK-33905][core] Unify the provision of metadata in RuntimeContext
---
 .../connectors/hive/HiveTableSpeculativeSinkITCase.java      | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git 
a/flink-connector-hive/src/test/java/org/apache/flink/connectors/hive/HiveTableSpeculativeSinkITCase.java
 
b/flink-connector-hive/src/test/java/org/apache/flink/connectors/hive/HiveTableSpeculativeSinkITCase.java
index 36ee5731..a18857de 100644
--- 
a/flink-connector-hive/src/test/java/org/apache/flink/connectors/hive/HiveTableSpeculativeSinkITCase.java
+++ 
b/flink-connector-hive/src/test/java/org/apache/flink/connectors/hive/HiveTableSpeculativeSinkITCase.java
@@ -119,15 +119,21 @@ class HiveTableSpeculativeSinkITCase {
                                             new RichMapFunction<Row, Row>() {
                                                 @Override
                                                 public Row map(Row value) 
throws Exception {
-                                                    if 
(getRuntimeContext().getAttemptNumber()
+                                                    if (getRuntimeContext()
+                                                                    
.getTaskInfo()
+                                                                    
.getAttemptNumber()
                                                             <= 0) {
                                                         
Thread.sleep(Integer.MAX_VALUE);
                                                     }
-                                                    assert 
getRuntimeContext().getAttemptNumber()
+                                                    assert getRuntimeContext()
+                                                                    
.getTaskInfo()
+                                                                    
.getAttemptNumber()
                                                             > 0;
                                                     value.setField(
                                                             1,
-                                                            
getRuntimeContext().getAttemptNumber());
+                                                            getRuntimeContext()
+                                                                    
.getTaskInfo()
+                                                                    
.getAttemptNumber());
                                                     return value;
                                                 }
                                             })

Reply via email to