Merge afe2c90 into 6ba2546

.gitignore

@@ -37,4 +37,5 @@ jspm_packages
 .node_repl_history
 
 .DS_Store
-index.js
+package-lock.json
+yarn.lock

.travis.yml

@@ -1,5 +1,6 @@
 language: node_js
 node_js:
+- '9'
 - '8'
 - '7'
 - '6'

build/index.d.ts

@@ -0,0 +1,3 @@
+import * as aws from 'aws-sdk';
+import { AthenaClient, AthenaClientConfig } from './lib/client';
+export declare function createClient(clientConfig: AthenaClientConfig, awsConfig: aws.Config): AthenaClient;

build/lib/client.d.ts

@@ -0,0 +1,29 @@
+import { Athena } from 'aws-sdk';
+import { AthenaRequest, AthenaRequestConfig } from './request';
+import { AthenaStream, AthenaStreamConfig } from './stream';
+export interface AthenaExecutionResult<T> {
+    records: T[];
+    queryExecution: Athena.QueryExecution;
+}
+export interface AthenaExecutionSelect<T> {
+    toPromise: () => Promise<AthenaExecutionResult<T>>;
+    toStream: () => AthenaStream<T>;
+}
+export interface AthenaClientConfig extends AthenaRequestConfig, AthenaStreamConfig {
+    pollingInterval?: number;
+    queryTimeout?: number;
+    concurrentExecMax?: number;
+    execRightCheckInterval?: number;
+}
+export declare class AthenaClient {
+    private config;
+    private concurrentExecNum;
+    private request;
+    constructor(request: AthenaRequest, config: AthenaClientConfig);
+    execute<T>(query: string): AthenaExecutionSelect<T>;
+    execute<T>(query: string, callback: (err?: Error, result?: AthenaExecutionResult<T>) => void): void;
+    private _execute<T>(query, athenaStream, config);
+    private canStartQuery();
+    private startQuery();
+    private endQuery();
+}

build/lib/request.d.ts

@@ -0,0 +1,18 @@
+/// <reference types="node" />
+import { Readable } from 'stream';
+export interface AthenaRequestConfig {
+    bucketUri: string;
+    baseRetryWait?: number;
+    retryWaitMax?: number;
+    retryCountMax?: number;
+}
+export declare class AthenaRequest {
+    private athena;
+    private s3;
+    constructor(athena: any, s3: any);
+    startQuery(query: string, config: AthenaRequestConfig): Promise<string>;
+    checkQuery(queryId: string, config: AthenaRequestConfig): Promise<boolean>;
+    stopQuery(queryId: string, config: AthenaRequestConfig): Promise<void>;
+    getQueryExecution(queryId: string, config: AthenaRequestConfig): Promise<any>;
+    getResultsStream(s3Uri: string): Readable;
+}

build/lib/stream.d.ts

@@ -0,0 +1,15 @@
+/// <reference types="node" />
+import { Duplex } from 'stream';
+export interface AthenaStreamConfig {
+    maxBufferSize?: string;
+}
+export declare class AthenaStream<T> extends Duplex {
+    private buffer;
+    private bufferSize;
+    private maxBufferSize;
+    private columns;
+    private isEnd;
+    constructor(config: AthenaStreamConfig);
+    _write(buffer: Buffer, _: string, callback: (err?: Error) => void): Promise<void>;
+    _read(): Promise<void>;
+}

build/lib/util.d.ts

@@ -0,0 +1,2 @@
+export declare function sleep(ms: number): Promise<void>;
+export declare function getBytes(text: string): number;