@databricks/sdk-uc-connections 0.0.0-dev → 0.1.0-dev.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,113 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+ import { execute } from '@databricks/sdk-core/api';
3
+ import { ApiError } from '@databricks/sdk-core/apierror';
4
+ import JSONBig from 'json-bigint';
5
+ // JSON codec that preserves int64 precision. On the way in, large integer
6
+ // literals come back as bigint instead of being rounded to JS Number. On the
7
+ // way out, bigint values are emitted as raw JSON number digits.
8
+ const jsonBigint = JSONBig({ useNativeBigInt: true });
9
+ /**
10
+ * Translates public CallOptions to the internal Options shape accepted by
11
+ * execute(). Even though the shapes match today, this isolates the public
12
+ * API from the executor's internal type so they can diverge.
13
+ */
14
+ export async function executeCall(call, options) {
15
+ const opts = {
16
+ ...(options?.retrier !== undefined && { retrier: options.retrier }),
17
+ ...(options?.rateLimiter !== undefined && {
18
+ rateLimiter: options.rateLimiter,
19
+ }),
20
+ ...(options?.timeout !== undefined && { timeout: options.timeout }),
21
+ };
22
+ return execute(options?.signal, call, opts);
23
+ }
24
+ async function readAll(body) {
25
+ if (body === null) {
26
+ return new Uint8Array(0);
27
+ }
28
+ const reader = body.getReader();
29
+ const chunks = [];
30
+ for (;;) {
31
+ const { done, value } = await reader.read();
32
+ if (done) {
33
+ break;
34
+ }
35
+ chunks.push(value);
36
+ }
37
+ const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
38
+ const result = new Uint8Array(totalLength);
39
+ let offset = 0;
40
+ for (const chunk of chunks) {
41
+ result.set(chunk, offset);
42
+ offset += chunk.length;
43
+ }
44
+ return result;
45
+ }
46
+ export async function executeHttpCall(opts) {
47
+ opts.logger.debug('HTTP request', {
48
+ method: opts.request.method,
49
+ url: opts.request.url,
50
+ });
51
+ let resp;
52
+ try {
53
+ resp = await opts.httpClient.send(opts.request);
54
+ }
55
+ catch (e) {
56
+ opts.logger.debug('HTTP request failed');
57
+ throw e;
58
+ }
59
+ const body = await readAll(resp.body);
60
+ opts.logger.debug('HTTP response', {
61
+ statusCode: resp.statusCode,
62
+ body: new TextDecoder().decode(body),
63
+ });
64
+ const apiErr = ApiError.fromHttpError(resp.statusCode, resp.headers, body);
65
+ if (apiErr !== undefined) {
66
+ throw apiErr;
67
+ }
68
+ return body;
69
+ }
70
+ export function buildHttpRequest(method, url, headers, signal, body) {
71
+ const req = { url, method, headers };
72
+ if (body !== undefined) {
73
+ req.body = body;
74
+ }
75
+ if (signal !== undefined) {
76
+ req.signal = signal;
77
+ }
78
+ return req;
79
+ }
80
+ export function parseResponse(body, schema) {
81
+ const text = new TextDecoder().decode(body);
82
+ const parsed = jsonBigint.parse(text);
83
+ return schema.parse(parsed);
84
+ }
85
+ export function marshalRequest(data, schema) {
86
+ return jsonBigint.stringify(schema.parse(data));
87
+ }
88
+ export function flattenQueryParams(prefix, value, params) {
89
+ if (value === null || value === undefined) {
90
+ return;
91
+ }
92
+ if (Array.isArray(value)) {
93
+ // arrays of objects are not yet supported
94
+ for (const item of value) {
95
+ params.append(prefix, String(item));
96
+ }
97
+ }
98
+ else if (typeof value === 'object') {
99
+ for (const [key, val] of Object.entries(value)) {
100
+ flattenQueryParams(`${prefix}.${key}`, val, params);
101
+ }
102
+ }
103
+ else if (typeof value === 'string' ||
104
+ typeof value === 'number' ||
105
+ typeof value === 'boolean' ||
106
+ typeof value === 'bigint') {
107
+ params.append(prefix, String(value));
108
+ }
109
+ else {
110
+ throw new Error(`Unsupported query parameter type: ${typeof value}`);
111
+ }
112
+ }
113
+ //# sourceMappingURL=utils.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,OAAO,EAAC,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAC,QAAQ,EAAC,MAAM,+BAA+B,CAAC;AAQvD,OAAO,OAAO,MAAM,aAAa,CAAC;AAGlC,0EAA0E;AAC1E,6EAA6E;AAC7E,gEAAgE;AAChE,MAAM,UAAU,GAAG,OAAO,CAAC,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;AAQpD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAU,EACV,OAAqB;IAErB,MAAM,IAAI,GAAY;QACpB,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;QACjE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QACF,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;KAClE,CAAC;IACF,OAAO,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAuC;IAEvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,SAAS,CAAC;QACR,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM;QACR,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;QAC3B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;KACtB,CAAC,CAAC;IAEH,IAAI,IAAkB,CAAC;IACvB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;IACV,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;QACjC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;KACrC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,GAAW,EACX,OAAgB,EAChB,MAAoB,EACpB,IAA0C;IAE1C,MAAM,GAAG,GAAgB,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC;IAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAI,IAAgB,EAAE,MAAoB;IACrE,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAY,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa,EAAE,MAAiB;IAC7D,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,KAAc,EACd,MAAuB;IAEvB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC1E,kBAAkB,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}
package/package.json CHANGED
@@ -1,7 +1,41 @@
1
1
  {
2
2
  "name": "@databricks/sdk-uc-connections",
3
- "version": "0.0.0-dev",
4
- "description": "Bootstrap placeholder; real contents in a later release.",
5
- "main": "index.js",
6
- "license": "Apache-2.0"
3
+ "version": "0.1.0-dev.1",
4
+ "description": "",
5
+ "type": "module",
6
+ "exports": {
7
+ "./v1": {
8
+ "types": "./dist/v1/index.d.ts",
9
+ "import": "./dist/v1/index.js"
10
+ }
11
+ },
12
+ "files": [
13
+ "dist",
14
+ "src",
15
+ "LICENSE"
16
+ ],
17
+ "scripts": {
18
+ "build": "tsc -b",
19
+ "lint": "eslint src --ext .ts",
20
+ "lint:fix": "eslint src --ext .ts --fix",
21
+ "format": "prettier --write \"src/**/*.ts\"",
22
+ "format:check": "prettier --check \"src/**/*.ts\"",
23
+ "typecheck": "tsc --noEmit",
24
+ "clean": "rm -rf dist tsconfig.tsbuildinfo",
25
+ "test": "echo 'no tests'",
26
+ "test:browser": "echo 'no tests'"
27
+ },
28
+ "author": "Databricks",
29
+ "license": "Apache-2.0",
30
+ "dependencies": {
31
+ "@databricks/sdk-auth": ">=0.1.0-dev.2 <1.0.0",
32
+ "@databricks/sdk-core": ">=0.1.0-dev.3 <1.0.0",
33
+ "@databricks/sdk-options": ">=0.1.0-dev.2 <1.0.0",
34
+ "@js-temporal/polyfill": "^0.5.0",
35
+ "json-bigint": "^1.0.0",
36
+ "zod": "^4.3.6"
37
+ },
38
+ "engines": {
39
+ "node": ">=22.0.0"
40
+ }
7
41
  }
@@ -0,0 +1,257 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+
3
+ import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
4
+ import type {Call} from '@databricks/sdk-core/api';
5
+ import {createDefault} from '@databricks/sdk-core/clientinfo';
6
+ import type {Logger} from '@databricks/sdk-core/logger';
7
+ import {NoOpLogger} from '@databricks/sdk-core/logger';
8
+ import type {CallOptions} from '@databricks/sdk-options/call';
9
+ import type {ClientOptions} from '@databricks/sdk-options/client';
10
+ import type {HttpClient} from '@databricks/sdk-core/http';
11
+ import {newHttpClient} from './transport';
12
+ import {
13
+ buildHttpRequest,
14
+ executeCall,
15
+ executeHttpCall,
16
+ marshalRequest,
17
+ parseResponse,
18
+ } from './utils';
19
+ import pkgJson from '../../package.json' with {type: 'json'};
20
+ import type {
21
+ ConnectionInfo,
22
+ CreateConnectionRequest,
23
+ DeleteConnectionRequest,
24
+ DeleteConnectionRequest_Response,
25
+ GetConnectionRequest,
26
+ ListConnectionsRequest,
27
+ ListConnectionsRequest_Response,
28
+ UpdateConnectionRequest,
29
+ } from './model';
30
+ import {
31
+ marshalCreateConnectionRequestSchema,
32
+ marshalUpdateConnectionRequestSchema,
33
+ unmarshalConnectionInfoSchema,
34
+ unmarshalDeleteConnectionRequest_ResponseSchema,
35
+ unmarshalListConnectionsRequest_ResponseSchema,
36
+ } from './model';
37
+
38
+ // Package identity segment for this client to be used in the User-Agent header.
39
+ const PACKAGE_SEGMENT = {
40
+ key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
41
+ value: pkgJson.version,
42
+ };
43
+
44
+ export class ConnectionsClient {
45
+ private readonly host: string;
46
+ // Workspace ID used to route workspace-level calls on unified hosts (SPOG).
47
+ // When set, workspace-level methods send X-Databricks-Org-Id on every
48
+ // request.
49
+ private readonly workspaceId: string | undefined;
50
+ private readonly httpClient: HttpClient;
51
+ private readonly logger: Logger;
52
+ // User-Agent header value. Composed once at construction from
53
+ // createDefault() merged with this package's identity and the active
54
+ // credential's name.
55
+ private readonly userAgent: string;
56
+
57
+ constructor(options: ClientOptions) {
58
+ if (options.host === undefined) {
59
+ throw new Error('Host is required.');
60
+ }
61
+ this.host = options.host.replace(/\/$/, '');
62
+ this.workspaceId = options.workspaceId;
63
+ this.logger = options.logger ?? new NoOpLogger();
64
+ const info = createDefault()
65
+ .with(PACKAGE_SEGMENT)
66
+ .with({key: 'sdk-js-auth', value: AUTH_VERSION})
67
+ .with({key: 'auth', value: options.credentials?.name() ?? 'default'});
68
+ this.userAgent = info.toString();
69
+ this.httpClient = newHttpClient(options);
70
+ }
71
+
72
+ /**
73
+ * Creates a new connection
74
+ *
75
+ * Creates a new connection to an external data source. It allows users to specify connection details and
76
+ * configurations for interaction with the external server.
77
+ */
78
+ async createConnection(
79
+ req: CreateConnectionRequest,
80
+ options?: CallOptions
81
+ ): Promise<ConnectionInfo> {
82
+ const url = `${this.host}/api/2.1/unity-catalog/connections`;
83
+ const body = marshalRequest(req, marshalCreateConnectionRequestSchema);
84
+ let resp: ConnectionInfo | undefined;
85
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
86
+ const headers = new Headers({'Content-Type': 'application/json'});
87
+ if (this.workspaceId !== undefined) {
88
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
89
+ }
90
+ headers.set('User-Agent', this.userAgent);
91
+ const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
92
+ const respBody = await executeHttpCall({
93
+ request: httpReq,
94
+ httpClient: this.httpClient,
95
+ logger: this.logger,
96
+ });
97
+ resp = parseResponse(respBody, unmarshalConnectionInfoSchema);
98
+ };
99
+ await executeCall(call, options);
100
+ if (resp === undefined) {
101
+ throw new Error('API call completed without a result.');
102
+ }
103
+ return resp;
104
+ }
105
+
106
+ /** Deletes the connection that matches the supplied name. */
107
+ async deleteConnection(
108
+ req: DeleteConnectionRequest,
109
+ options?: CallOptions
110
+ ): Promise<DeleteConnectionRequest_Response> {
111
+ const url = `${this.host}/api/2.1/unity-catalog/connections/${req.nameArg ?? ''}`;
112
+ let resp: DeleteConnectionRequest_Response | undefined;
113
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
114
+ const headers = new Headers();
115
+ if (this.workspaceId !== undefined) {
116
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
117
+ }
118
+ headers.set('User-Agent', this.userAgent);
119
+ const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
120
+ const respBody = await executeHttpCall({
121
+ request: httpReq,
122
+ httpClient: this.httpClient,
123
+ logger: this.logger,
124
+ });
125
+ resp = parseResponse(
126
+ respBody,
127
+ unmarshalDeleteConnectionRequest_ResponseSchema
128
+ );
129
+ };
130
+ await executeCall(call, options);
131
+ if (resp === undefined) {
132
+ throw new Error('API call completed without a result.');
133
+ }
134
+ return resp;
135
+ }
136
+
137
+ /** Gets a connection from it's name. */
138
+ async getConnection(
139
+ req: GetConnectionRequest,
140
+ options?: CallOptions
141
+ ): Promise<ConnectionInfo> {
142
+ const url = `${this.host}/api/2.1/unity-catalog/connections/${req.nameArg ?? ''}`;
143
+ let resp: ConnectionInfo | undefined;
144
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
145
+ const headers = new Headers();
146
+ if (this.workspaceId !== undefined) {
147
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
148
+ }
149
+ headers.set('User-Agent', this.userAgent);
150
+ const httpReq = buildHttpRequest('GET', url, headers, callSignal);
151
+ const respBody = await executeHttpCall({
152
+ request: httpReq,
153
+ httpClient: this.httpClient,
154
+ logger: this.logger,
155
+ });
156
+ resp = parseResponse(respBody, unmarshalConnectionInfoSchema);
157
+ };
158
+ await executeCall(call, options);
159
+ if (resp === undefined) {
160
+ throw new Error('API call completed without a result.');
161
+ }
162
+ return resp;
163
+ }
164
+
165
+ /**
166
+ * List all connections.
167
+ *
168
+ * NOTE: we recommend using max_results=0 to use the paginated version of this API. Unpaginated calls will be deprecated soon.
169
+ *
170
+ * PAGINATION BEHAVIOR: When using pagination (max_results >= 0), a page may contain zero results while still providing a next_page_token.
171
+ * Clients must continue reading pages until next_page_token is absent, which is the only indication that the end of results has been reached.
172
+ */
173
+ async listConnections(
174
+ req: ListConnectionsRequest,
175
+ options?: CallOptions
176
+ ): Promise<ListConnectionsRequest_Response> {
177
+ const url = `${this.host}/api/2.1/unity-catalog/connections`;
178
+ const params = new URLSearchParams();
179
+ if (req.maxResults !== undefined) {
180
+ params.append('max_results', String(req.maxResults));
181
+ }
182
+ if (req.pageToken !== undefined) {
183
+ params.append('page_token', req.pageToken);
184
+ }
185
+ const query = params.toString();
186
+ const fullUrl = query !== '' ? `${url}?${query}` : url;
187
+ let resp: ListConnectionsRequest_Response | undefined;
188
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
189
+ const headers = new Headers();
190
+ if (this.workspaceId !== undefined) {
191
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
192
+ }
193
+ headers.set('User-Agent', this.userAgent);
194
+ const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
195
+ const respBody = await executeHttpCall({
196
+ request: httpReq,
197
+ httpClient: this.httpClient,
198
+ logger: this.logger,
199
+ });
200
+ resp = parseResponse(
201
+ respBody,
202
+ unmarshalListConnectionsRequest_ResponseSchema
203
+ );
204
+ };
205
+ await executeCall(call, options);
206
+ if (resp === undefined) {
207
+ throw new Error('API call completed without a result.');
208
+ }
209
+ return resp;
210
+ }
211
+
212
+ async *listConnectionsIter(
213
+ req: ListConnectionsRequest,
214
+ options?: CallOptions
215
+ ): AsyncGenerator<ConnectionInfo> {
216
+ const pageReq: ListConnectionsRequest = {...req};
217
+ for (;;) {
218
+ const resp = await this.listConnections(pageReq, options);
219
+ for (const item of resp.connections ?? []) {
220
+ yield item;
221
+ }
222
+ if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
223
+ return;
224
+ }
225
+ pageReq.pageToken = resp.nextPageToken;
226
+ }
227
+ }
228
+
229
+ /** Updates the connection that matches the supplied name. */
230
+ async updateConnection(
231
+ req: UpdateConnectionRequest,
232
+ options?: CallOptions
233
+ ): Promise<ConnectionInfo> {
234
+ const url = `${this.host}/api/2.1/unity-catalog/connections/${req.nameArg ?? ''}`;
235
+ const body = marshalRequest(req, marshalUpdateConnectionRequestSchema);
236
+ let resp: ConnectionInfo | undefined;
237
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
238
+ const headers = new Headers({'Content-Type': 'application/json'});
239
+ if (this.workspaceId !== undefined) {
240
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
241
+ }
242
+ headers.set('User-Agent', this.userAgent);
243
+ const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
244
+ const respBody = await executeHttpCall({
245
+ request: httpReq,
246
+ httpClient: this.httpClient,
247
+ logger: this.logger,
248
+ });
249
+ resp = parseResponse(respBody, unmarshalConnectionInfoSchema);
250
+ };
251
+ await executeCall(call, options);
252
+ if (resp === undefined) {
253
+ throw new Error('API call completed without a result.');
254
+ }
255
+ return resp;
256
+ }
257
+ }
@@ -0,0 +1,28 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+
3
+ export {ConnectionsClient} from './client';
4
+
5
+ export {
6
+ ConnectionType,
7
+ CredentialType,
8
+ SecurableType,
9
+ ProvisioningInfo_State,
10
+ } from './model';
11
+
12
+ export type {
13
+ ConnectionInfo,
14
+ ConnectionInfo_OptionsEntry,
15
+ ConnectionInfo_PropertiesEntry,
16
+ CreateConnectionRequest,
17
+ CreateConnectionRequest_OptionsEntry,
18
+ CreateConnectionRequest_PropertiesEntry,
19
+ DeleteConnectionRequest,
20
+ DeleteConnectionRequest_Response,
21
+ GetConnectionRequest,
22
+ ListConnectionsRequest,
23
+ ListConnectionsRequest_Response,
24
+ ProvisioningInfo,
25
+ UpdateConnectionRequest,
26
+ UpdateConnectionRequest_OptionsEntry,
27
+ UpdateConnectionRequest_PropertiesEntry,
28
+ } from './model';