@@ -2925,22 +2925,25 @@ public void testPartitionedDmlDoesNotTimeout() {
2925
2925
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
2926
2926
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2927
2927
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2928
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
2928
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2929
2929
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2930
2930
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2931
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
2931
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2932
2932
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2933
2933
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2934
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
2934
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2935
2935
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2936
2936
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2937
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
2937
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2938
2938
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2939
2939
"google.spanner.v1.Spanner/BeginTransaction" ,
2940
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2940
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
2941
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2942
+ "google.spanner.v1.Spanner/CreateSession" ,
2943
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , 1 , 1 )),
2941
2944
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2942
2945
"google.spanner.v1.Spanner/ExecuteSql" ,
2943
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
2946
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 8 , 1 )),
2944
2947
};
2945
2948
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
2946
2949
}
@@ -2979,13 +2982,17 @@ public void testPartitionedDmlWithLowerTimeout() {
2979
2982
assertThat (updateCount ).isEqualTo (UPDATE_COUNT );
2980
2983
2981
2984
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
2982
- int channelId = dbImpl .getSession ().getChannel ();
2985
+
2986
+ int channelId = 0 ;
2987
+ try (Session session = dbImpl .getSession ()) {
2988
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
2989
+ }
2983
2990
int dbId = dbImpl .dbId ;
2984
2991
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
2985
2992
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
2986
2993
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2987
2994
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
2988
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
2995
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2989
2996
};
2990
2997
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
2991
2998
@@ -3072,7 +3079,7 @@ public void testPartitionedDmlWithHigherTimeout() {
3072
3079
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3073
3080
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3074
3081
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3075
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
3082
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3076
3083
};
3077
3084
3078
3085
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
@@ -3117,33 +3124,39 @@ public void testPartitionedDmlRetriesOnUnavailable() {
3117
3124
assertThat (updateCount ).isEqualTo (UPDATE_COUNT );
3118
3125
3119
3126
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
3120
- int channelId = dbImpl .getSession ().getChannel ();
3127
+ int channelId = 0 ;
3128
+ try (Session session = dbImpl .getSession ()) {
3129
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
3130
+ }
3121
3131
int dbId = dbImpl .dbId ;
3122
3132
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3123
3133
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3124
3134
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3125
3135
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3126
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
3136
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3127
3137
};
3128
3138
3129
3139
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3130
3140
3131
3141
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3132
3142
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3133
3143
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3134
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
3144
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3135
3145
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3136
3146
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3137
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
3147
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3138
3148
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3139
3149
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3140
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
3150
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3141
3151
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3142
3152
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3143
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
3153
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3144
3154
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3145
3155
"google.spanner.v1.Spanner/BeginTransaction" ,
3146
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3156
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
3157
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3158
+ "google.spanner.v1.Spanner/CreateSession" ,
3159
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , 1 , 1 )),
3147
3160
};
3148
3161
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3149
3162
}
@@ -3554,31 +3567,38 @@ public void testNestedTransactionsUsingTwoDatabases() throws InterruptedExceptio
3554
3567
assertThat (client1 .pool .getNumberOfSessionsInPool ()).isEqualTo (minSessions );
3555
3568
assertThat (client2 .pool .getNumberOfSessionsInPool ()).isEqualTo (minSessions );
3556
3569
3557
- int channelId = client1 .getSession ().getChannel ();
3570
+ int channelId = 0 ;
3571
+ try (Session session = client1 .getSession ()) {
3572
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
3573
+ }
3558
3574
int dbId = client1 .dbId ;
3575
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3559
3576
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3560
3577
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3561
3578
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3562
- new XGoogSpannerRequestId (dbId , 1 , 5 , 1 )),
3579
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 6 , 1 )),
3580
+ XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3581
+ "google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3582
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , 6 , 1 )),
3563
3583
};
3564
3584
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3565
3585
3566
3586
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3567
3587
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3568
3588
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3569
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3589
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3570
3590
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3571
3591
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3572
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3592
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3573
3593
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3574
3594
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3575
- new XGoogSpannerRequestId (dbId , 2 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3595
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3576
3596
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3577
3597
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3578
- new XGoogSpannerRequestId (dbId , 3 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3598
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3579
3599
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3580
3600
"google.spanner.v1.Spanner/BeginTransaction" ,
3581
- new XGoogSpannerRequestId (dbId , channelId , 6 , 1 )),
3601
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3582
3602
};
3583
3603
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3584
3604
}
@@ -5392,7 +5412,10 @@ public void testRetryOnResourceExhausted() {
5392
5412
}
5393
5413
5394
5414
DatabaseClientImpl dbClient = (DatabaseClientImpl ) client ;
5395
- int channelId = dbClient .getSession ().getChannel ();
5415
+ int channelId = 0 ;
5416
+ try (Session session = dbClient .getSession ()) {
5417
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
5418
+ }
5396
5419
int dbId = dbClient .dbId ;
5397
5420
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
5398
5421
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
0 commit comments