@@ -2909,35 +2909,41 @@ public void testPartitionedDmlDoesNotTimeout() {
2909
2909
assertEquals (ErrorCode .DEADLINE_EXCEEDED , e .getErrorCode ());
2910
2910
2911
2911
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
2912
- int channelId = dbImpl .getSession ().getChannel ();
2912
+ int channelId = 0 ;
2913
+ try (Session session = dbImpl .getSession ()) {
2914
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
2915
+ }
2913
2916
int dbId = dbImpl .dbId ;
2914
2917
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
2915
2918
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
2916
2919
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2917
2920
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
2918
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
2921
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2919
2922
};
2920
2923
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
2921
2924
2922
2925
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
2923
2926
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2924
2927
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2925
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
2928
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2926
2929
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2927
2930
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2928
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
2931
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2929
2932
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2930
2933
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2931
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
2934
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2932
2935
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2933
2936
"google.spanner.v1.Spanner/BatchCreateSessions" ,
2934
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
2937
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
2935
2938
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2936
2939
"google.spanner.v1.Spanner/BeginTransaction" ,
2937
- 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 )),
2938
2944
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2939
2945
"google.spanner.v1.Spanner/ExecuteSql" ,
2940
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 7 , 1 )),
2946
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 8 , 1 )),
2941
2947
};
2942
2948
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
2943
2949
}
@@ -2976,13 +2982,17 @@ public void testPartitionedDmlWithLowerTimeout() {
2976
2982
assertThat (updateCount ).isEqualTo (UPDATE_COUNT );
2977
2983
2978
2984
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
2979
- int channelId = dbImpl .getSession ().getChannel ();
2985
+
2986
+ int channelId = 0 ;
2987
+ try (Session session = dbImpl .getSession ()) {
2988
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
2989
+ }
2980
2990
int dbId = dbImpl .dbId ;
2981
2991
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
2982
2992
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
2983
2993
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
2984
2994
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
2985
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
2995
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
2986
2996
};
2987
2997
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
2988
2998
@@ -3060,13 +3070,16 @@ public void testPartitionedDmlWithHigherTimeout() {
3060
3070
assertThat (updateCount ).isEqualTo (UPDATE_COUNT );
3061
3071
3062
3072
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
3063
- int channelId = dbImpl .getSession ().getChannel ();
3073
+ int channelId = 0 ;
3074
+ try (Session session = dbImpl .getSession ()) {
3075
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
3076
+ }
3064
3077
int dbId = dbImpl .dbId ;
3065
3078
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3066
3079
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3067
3080
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3068
3081
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3069
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
3082
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3070
3083
};
3071
3084
3072
3085
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
@@ -3111,33 +3124,39 @@ public void testPartitionedDmlRetriesOnUnavailable() {
3111
3124
assertThat (updateCount ).isEqualTo (UPDATE_COUNT );
3112
3125
3113
3126
DatabaseClientImpl dbImpl = ((DatabaseClientImpl ) client );
3114
- int channelId = dbImpl .getSession ().getChannel ();
3127
+ int channelId = 0 ;
3128
+ try (Session session = dbImpl .getSession ()) {
3129
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
3130
+ }
3115
3131
int dbId = dbImpl .dbId ;
3116
3132
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3117
3133
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3118
3134
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3119
3135
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3120
- new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 5 , 1 )),
3136
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , channelId , 6 , 1 )),
3121
3137
};
3122
3138
3123
3139
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3124
3140
3125
3141
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3126
3142
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3127
3143
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3128
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 0 , NON_DETERMINISTIC , 1 )),
3144
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3129
3145
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3130
3146
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3131
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 1 , NON_DETERMINISTIC , 1 )),
3147
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3132
3148
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3133
3149
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3134
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 2 , NON_DETERMINISTIC , 1 )),
3150
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3135
3151
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3136
3152
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3137
- new XGoogSpannerRequestId (NON_DETERMINISTIC , 3 , NON_DETERMINISTIC , 1 )),
3153
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3138
3154
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3139
3155
"google.spanner.v1.Spanner/BeginTransaction" ,
3140
- 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 )),
3141
3160
};
3142
3161
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3143
3162
}
@@ -3548,31 +3567,38 @@ public void testNestedTransactionsUsingTwoDatabases() throws InterruptedExceptio
3548
3567
assertThat (client1 .pool .getNumberOfSessionsInPool ()).isEqualTo (minSessions );
3549
3568
assertThat (client2 .pool .getNumberOfSessionsInPool ()).isEqualTo (minSessions );
3550
3569
3551
- int channelId = client1 .getSession ().getChannel ();
3570
+ int channelId = 0 ;
3571
+ try (Session session = client1 .getSession ()) {
3572
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
3573
+ }
3552
3574
int dbId = client1 .dbId ;
3575
+ long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
3553
3576
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
3554
3577
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3555
3578
"google.spanner.v1.Spanner/ExecuteStreamingSql" ,
3556
- 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 )),
3557
3583
};
3558
3584
xGoogReqIdInterceptor .checkExpectedStreamingXGoogRequestIds (wantStreamingValues );
3559
3585
3560
3586
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantUnaryValues = {
3561
3587
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3562
3588
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3563
- new XGoogSpannerRequestId (dbId , 0 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3589
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3564
3590
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3565
3591
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3566
- new XGoogSpannerRequestId (dbId , 1 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3592
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3567
3593
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3568
3594
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3569
- new XGoogSpannerRequestId (dbId , 2 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3595
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3570
3596
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3571
3597
"google.spanner.v1.Spanner/BatchCreateSessions" ,
3572
- new XGoogSpannerRequestId (dbId , 3 , XGoogSpannerRequestIdTest . NON_DETERMINISTIC , 1 )),
3598
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3573
3599
XGoogSpannerRequestIdTest .ofMethodAndRequestId (
3574
3600
"google.spanner.v1.Spanner/BeginTransaction" ,
3575
- new XGoogSpannerRequestId (dbId , channelId , 6 , 1 )),
3601
+ new XGoogSpannerRequestId (NON_DETERMINISTIC , NON_DETERMINISTIC , NON_DETERMINISTIC , 1 )),
3576
3602
};
3577
3603
xGoogReqIdInterceptor .checkExpectedUnaryXGoogRequestIds (wantUnaryValues );
3578
3604
}
@@ -5386,7 +5412,10 @@ public void testRetryOnResourceExhausted() {
5386
5412
}
5387
5413
5388
5414
DatabaseClientImpl dbClient = (DatabaseClientImpl ) client ;
5389
- int channelId = dbClient .getSession ().getChannel ();
5415
+ int channelId = 0 ;
5416
+ try (Session session = dbClient .getSession ()) {
5417
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
5418
+ }
5390
5419
int dbId = dbClient .dbId ;
5391
5420
long NON_DETERMINISTIC = XGoogSpannerRequestIdTest .NON_DETERMINISTIC ;
5392
5421
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {
@@ -5549,7 +5578,10 @@ public void testSessionPoolExhaustedError_containsStackTraces() {
5549
5578
// Closing the transactions should return the sessions to the pool.
5550
5579
assertEquals (4 , pool .getNumberOfSessionsInPool ());
5551
5580
5552
- int channelId = dbClient .getSession ().getChannel ();
5581
+ int channelId = 0 ;
5582
+ try (Session session = dbClient .getSession ()) {
5583
+ channelId = ((PooledSessionFuture ) session ).getChannel ();
5584
+ }
5553
5585
int dbId = dbClient .dbId ;
5554
5586
XGoogSpannerRequestIdTest .MethodAndRequestId [] wantStreamingValues = {};
5555
5587
0 commit comments