@@ -106,15 +106,16 @@ export async function invokeCommitRpc(
106
106
datastore instanceof DatastoreImpl ,
107
107
'invokeCommitRpc() requires DatastoreImpl'
108
108
) ;
109
+ const datastoreImpl : DatastoreImpl = datastore ;
109
110
const params = {
110
- database : datastore . serializer . encodedDatabaseId ,
111
- writes : mutations . map ( m => datastore . serializer . toMutation ( m ) )
111
+ database : datastoreImpl . serializer . encodedDatabaseId ,
112
+ writes : mutations . map ( m => datastoreImpl . serializer . toMutation ( m ) )
112
113
} ;
113
- const response = await datastore . invokeRPC <
114
+ const response = await datastoreImpl . invokeRPC <
114
115
api . CommitRequest ,
115
116
api . CommitResponse
116
117
> ( 'Commit' , params ) ;
117
- return datastore . serializer . fromWriteResults (
118
+ return datastoreImpl . serializer . fromWriteResults (
118
119
response . writeResults ,
119
120
response . commitTime
120
121
) ;
@@ -128,18 +129,19 @@ export async function invokeBatchGetDocumentsRpc(
128
129
datastore instanceof DatastoreImpl ,
129
130
'invokeBatchGetDocumentsRpc() requires DatastoreImpl'
130
131
) ;
132
+ const datastoreImpl : DatastoreImpl = datastore ;
131
133
const params = {
132
- database : datastore . serializer . encodedDatabaseId ,
133
- documents : keys . map ( k => datastore . serializer . toName ( k ) )
134
+ database : datastoreImpl . serializer . encodedDatabaseId ,
135
+ documents : keys . map ( k => datastoreImpl . serializer . toName ( k ) )
134
136
} ;
135
- const response = await datastore . invokeStreamingRPC <
137
+ const response = await datastoreImpl . invokeStreamingRPC <
136
138
api . BatchGetDocumentsRequest ,
137
139
api . BatchGetDocumentsResponse
138
140
> ( 'BatchGetDocuments' , params ) ;
139
141
140
142
const docs = new Map < string , MaybeDocument > ( ) ;
141
143
response . forEach ( proto => {
142
- const doc = datastore . serializer . fromMaybeDocument ( proto ) ;
144
+ const doc = datastoreImpl . serializer . fromMaybeDocument ( proto ) ;
143
145
docs . set ( doc . key . toString ( ) , doc ) ;
144
146
} ) ;
145
147
const result : MaybeDocument [ ] = [ ] ;
@@ -160,11 +162,12 @@ export function newPersistentWriteStream(
160
162
datastore instanceof DatastoreImpl ,
161
163
'newPersistentWriteStream() requires DatastoreImpl'
162
164
) ;
165
+ const datastoreImpl : DatastoreImpl = datastore ;
163
166
return new PersistentWriteStream (
164
167
queue ,
165
- datastore . connection ,
166
- datastore . credentials ,
167
- datastore . serializer ,
168
+ datastoreImpl . connection ,
169
+ datastoreImpl . credentials ,
170
+ datastoreImpl . serializer ,
168
171
listener
169
172
) ;
170
173
}
@@ -178,11 +181,12 @@ export function newPersistentWatchStream(
178
181
datastore instanceof DatastoreImpl ,
179
182
'newPersistentWatchStream() requires DatastoreImpl'
180
183
) ;
184
+ const datastoreImpl : DatastoreImpl = datastore ;
181
185
return new PersistentListenStream (
182
186
queue ,
183
- datastore . connection ,
184
- datastore . credentials ,
185
- datastore . serializer ,
187
+ datastoreImpl . connection ,
188
+ datastoreImpl . credentials ,
189
+ datastoreImpl . serializer ,
186
190
listener
187
191
) ;
188
192
}
0 commit comments