set session in header (#12)

* set session in header * set session in header * increase timeout
tidbcloud · Sep 12, 2023 · c390f24 · c390f24
1 parent f8b39eb
commit c390f24
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/integration-test/type.test.ts b/integration-test/type.test.ts
@@ -143,7 +143,7 @@ beforeAll(async () => {
   await con.execute(`DROP DATABASE IF EXISTS ${database}`)
   await con.execute(`CREATE DATABASE ${database}`)
   await con.execute(multiDataTable)
-},10000);
+},20000);
 
 describe('types', () => {
 

diff --git a/src/config.ts b/src/config.ts
@@ -9,6 +9,7 @@ type Res = {
   ok: boolean
   status: number
   statusText: string
+  headers: any
   json(): Promise<any>
   text(): Promise<string>
 }

diff --git a/src/index.ts b/src/index.ts
@@ -100,7 +100,7 @@ export class Connection {
     const resp = await postQuery<QueryExecuteResponse>(this.config, body, this.session ?? '')
 
     this.session = resp?.session ?? null
-    if (this.session === null) {
+    if (this.session === null || this.session === '') {
       throw new DatabaseError('empty session, please try again', 500, null)
     }
 

diff --git a/src/serverless.ts b/src/serverless.ts
@@ -28,7 +28,10 @@ export async function postQuery<T>(config: Config, body, session = ''): Promise<
   })
 
   if (response.ok) {
-    return await response.json()
+    const resp = await response.json()
+    const session = response.headers.get('TiDB-Session')
+    resp.session = session ?? ''
+    return resp
   } else {
     let error
     try {