Skip to content

Commit cb1716e

Browse files
hessjcghessjcg
authored
feat: Refresh the connector if the domain name is changed to point to a new instance. (#427)
If the connector is configured with a domain name, when that domain name record changes to resolve to a new instance, the connector should detect that change, close all connections to the old instance, and create connections to the new instance. Fixes #421
1 parent b94cc98 commit cb1716e

File tree

9 files changed

+593
-69
lines changed

9 files changed

+593
-69
lines changed

‎.github/workflows/tests.yml

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,8 @@ jobs:
161161
POSTGRES_CAS_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CAS_PASS
162162
POSTGRES_CUSTOMER_CAS_CONNECTION_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_CONNECTION_NAME
163163
POSTGRES_CUSTOMER_CAS_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_PASS
164+
POSTGRES_CUSTOMER_CAS_DOMAIN_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_DOMAIN_NAME
165+
POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME
164166
SQLSERVER_CONNECTION_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/SQLSERVER_CONNECTION_NAME
165167
SQLSERVER_USER:${{ vars.GOOGLE_CLOUD_PROJECT }}/SQLSERVER_USER
166168
SQLSERVER_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/SQLSERVER_PASS
@@ -188,6 +190,8 @@ jobs:
188190
POSTGRES_CAS_PASS: "${{ steps.secrets.outputs.POSTGRES_CAS_PASS }}"
189191
POSTGRES_CUSTOMER_CAS_CONNECTION_NAME: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_CONNECTION_NAME }}"
190192
POSTGRES_CUSTOMER_CAS_PASS: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_PASS }}"
193+
POSTGRES_CUSTOMER_CAS_DOMAIN_NAME: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_DOMAIN_NAME }}"
194+
POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME }}"
191195
SQLSERVER_CONNECTION_NAME: "${{ steps.secrets.outputs.SQLSERVER_CONNECTION_NAME }}"
192196
SQLSERVER_USER: "${{ steps.secrets.outputs.SQLSERVER_USER }}"
193197
SQLSERVER_PASS: "${{ steps.secrets.outputs.SQLSERVER_PASS }}"
@@ -278,6 +282,12 @@ jobs:
278282
POSTGRES_IAM_USER:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_USER_IAM_NODE
279283
POSTGRES_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_PASS
280284
POSTGRES_DB:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_DB
285+
POSTGRES_CAS_CONNECTION_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CAS_CONNECTION_NAME
286+
POSTGRES_CAS_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CAS_PASS
287+
POSTGRES_CUSTOMER_CAS_CONNECTION_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_CONNECTION_NAME
288+
POSTGRES_CUSTOMER_CAS_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_PASS
289+
POSTGRES_CUSTOMER_CAS_DOMAIN_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_DOMAIN_NAME
290+
POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME
281291
SQLSERVER_CONNECTION_NAME:${{ vars.GOOGLE_CLOUD_PROJECT }}/SQLSERVER_CONNECTION_NAME
282292
SQLSERVER_USER:${{ vars.GOOGLE_CLOUD_PROJECT }}/SQLSERVER_USER
283293
SQLSERVER_PASS:${{ vars.GOOGLE_CLOUD_PROJECT }}/SQLSERVER_PASS
@@ -295,6 +305,12 @@ jobs:
295305
POSTGRES_IAM_USER: "${{ steps.secrets.outputs.POSTGRES_IAM_USER }}"
296306
POSTGRES_PASS: "${{ steps.secrets.outputs.POSTGRES_PASS }}"
297307
POSTGRES_DB: "${{ steps.secrets.outputs.POSTGRES_DB }}"
308+
POSTGRES_CAS_CONNECTION_NAME: "${{ steps.secrets.outputs.POSTGRES_CAS_CONNECTION_NAME }}"
309+
POSTGRES_CAS_PASS: "${{ steps.secrets.outputs.POSTGRES_CAS_PASS }}"
310+
POSTGRES_CUSTOMER_CAS_CONNECTION_NAME: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_CONNECTION_NAME }}"
311+
POSTGRES_CUSTOMER_CAS_PASS: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_PASS }}"
312+
POSTGRES_CUSTOMER_CAS_DOMAIN_NAME: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_DOMAIN_NAME }}"
313+
POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME: "${{ steps.secrets.outputs.POSTGRES_CUSTOMER_CAS_INVALID_DOMAIN_NAME }}"
298314
SQLSERVER_CONNECTION_NAME: "${{ steps.secrets.outputs.SQLSERVER_CONNECTION_NAME }}"
299315
SQLSERVER_USER: "${{ steps.secrets.outputs.SQLSERVER_USER }}"
300316
SQLSERVER_PASS: "${{ steps.secrets.outputs.SQLSERVER_PASS }}"

‎README.md

Lines changed: 82 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -444,6 +444,88 @@ variables. Here is a quick reference to supported values and their effect:
444444
- `GOOGLE_CLOUD_QUOTA_PROJECT`: Used to set a custom quota project to Cloud SQL
445445
APIs when defined.
446446

447+
## Using DNS domain names to identify instances
448+
449+
The connector can be configured to use DNS to look up an instance. This would
450+
allow you to configure your application to connect to a database instance, and
451+
centrally configure which instance in your DNS zone.
452+
453+
### Configure your DNS Records
454+
455+
Add a DNS TXT record for the Cloud SQL instance to a **private** DNS server
456+
or a private Google Cloud DNS Zone used by your application.
457+
458+
**Note:** You are strongly discouraged from adding DNS records for your
459+
Cloud SQL instances to a public DNS server. This would allow anyone on the
460+
internet to discover the Cloud SQL instance name.
461+
462+
For example: suppose you wanted to use the domain name
463+
`prod-db.mycompany.example.com` to connect to your database instance
464+
`my-project:region:my-instance`. You would create the following DNS record:
465+
466+
- Record type: `TXT`
467+
- Name: `prod-db.mycompany.example.com` – This is the domain name used by the application
468+
- Value: `my-project:region:my-instance` – This is the instance name
469+
470+
### Configure the connector
471+
472+
Configure the connector as described above, replacing the connector ID with
473+
the DNS name.
474+
475+
Adapting the MySQL + database/sql example above:
476+
477+
```js
478+
import mysql from 'mysql2/promise';
479+
import {Connector} from '@google-cloud/cloud-sql-connector';
480+
481+
const connector = new Connector();
482+
const clientOpts = await connector.getOptions({
483+
domainName: 'prod-db.mycompany.example.com',
484+
ipType: 'PUBLIC',
485+
});
486+
487+
const pool = await mysql.createPool({
488+
...clientOpts,
489+
user: 'my-user',
490+
password: 'my-password',
491+
database: 'db-name',
492+
});
493+
const conn = await pool.getConnection();
494+
const [result] = await conn.query(`SELECT NOW();`);
495+
console.table(result); // prints returned time value from server
496+
497+
await pool.end();
498+
connector.close();
499+
```
500+
501+
## Automatic failover using DNS domain names
502+
503+
For example: suppose application is configured to connect using the
504+
domain name `prod-db.mycompany.example.com`. Initially the private DNS
505+
zone has a TXT record with the value `my-project:region:my-instance`. The
506+
application establishes connections to the `my-project:region:my-instance`
507+
Cloud SQL instance. Configure the connector using the `domainName` option:
508+
509+
Then, to reconfigure the application to use a different database
510+
instance, change the value of the `prod-db.mycompany.example.com` DNS record
511+
from `my-project:region:my-instance` to `my-project:other-region:my-instance-2`
512+
513+
The connector inside the application detects the change to this
514+
DNS record. Now, when the application connects to its database using the
515+
domain name `prod-db.mycompany.example.com`, it will connect to the
516+
`my-project:other-region:my-instance-2` Cloud SQL instance.
517+
518+
The connector will automatically close all existing connections to
519+
`my-project:region:my-instance`. This will force the connection pools to
520+
establish new connections. Also, it may cause database queries in progress
521+
to fail.
522+
523+
The connector will poll for changes to the DNS name every 30 seconds by default.
524+
You may configure the frequency of the connections using the Connector's
525+
`failoverPeriod` option. When this is set to 0, the connector will disable
526+
polling and only check if the DNS record changed when it is creating a new
527+
connection.
528+
447529
## Support policy
448530

449531
### Major version lifecycle

‎src/cloud-sql-instance.ts

Lines changed: 72 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,26 @@
1414

1515
import {IpAddressTypes, selectIpAddress} from './ip-addresses';
1616
import {InstanceConnectionInfo} from './instance-connection-info';
17-
import {resolveInstanceName} from './parse-instance-connection-name';
17+
import {
18+
isSameInstance,
19+
resolveInstanceName,
20+
} from './parse-instance-connection-name';
1821
import {InstanceMetadata} from './sqladmin-fetcher';
1922
import {generateKeys} from './crypto';
2023
import {RSAKeys} from './rsa-keys';
2124
import {SslCert} from './ssl-cert';
2225
import {getRefreshInterval, isExpirationTimeValid} from './time';
2326
import {AuthTypes} from './auth-types';
27+
import {CloudSQLConnectorError} from './errors';
28+
29+
// Private types that describe exactly the methods
30+
// needed from tls.Socket to be able to close
31+
// sockets when the DNS Name changes.
32+
type EventFn = () => void;
33+
type DestroyableSocket = {
34+
destroy: (error?: Error) => void;
35+
once: (name: string, handler: EventFn) => void;
36+
};
2437

2538
interface Fetcher {
2639
getInstanceMetadata({
@@ -42,6 +55,7 @@ interface CloudSQLInstanceOptions {
4255
ipType: IpAddressTypes;
4356
limitRateInterval?: number;
4457
sqlAdminFetcher: Fetcher;
58+
failoverPeriod?: number;
4559
}
4660

4761
interface RefreshResult {
@@ -74,9 +88,13 @@ export class CloudSQLInstance {
7488
// The ongoing refresh promise is referenced by the `next` property
7589
private next?: Promise<RefreshResult>;
7690
private scheduledRefreshID?: ReturnType<typeof setTimeout> | null = undefined;
91+
private checkDomainID?: ReturnType<typeof setInterval> | null = undefined;
7792
/* eslint-disable-next-line @typescript-eslint/no-explicit-any */
7893
private throttle?: any;
7994
private closed = false;
95+
private failoverPeriod: number;
96+
private sockets = new Set<DestroyableSocket>();
97+
8098
public readonly instanceInfo: InstanceConnectionInfo;
8199
public ephemeralCert?: SslCert;
82100
public host?: string;
@@ -98,6 +116,7 @@ export class CloudSQLInstance {
98116
this.ipType = options.ipType || IpAddressTypes.PUBLIC;
99117
this.limitRateInterval = options.limitRateInterval || 30 * 1000; // 30 seconds
100118
this.sqlAdminFetcher = options.sqlAdminFetcher;
119+
this.failoverPeriod = options.failoverPeriod || 30 * 1000; // 30 seconds
101120
}
102121

103122
// p-throttle library has to be initialized in an async scope in order to
@@ -153,6 +172,19 @@ export class CloudSQLInstance {
153172
return Promise.reject('closed');
154173
}
155174

175+
// Lazy instantiation of the checkDomain interval on the first refresh
176+
// This avoids issues with test cases that instantiate a CloudSqlInstance.
177+
// If failoverPeriod is 0 (or negative) don't check for DNS updates.
178+
if (
179+
this?.instanceInfo?.domainName &&
180+
!this.checkDomainID &&
181+
this.failoverPeriod > 0
182+
) {
183+
this.checkDomainID = setInterval(() => {
184+
this.checkDomainChanged();
185+
}, this.failoverPeriod);
186+
}
187+
156188
const currentRefreshId = this.scheduledRefreshID;
157189

158190
// Since forceRefresh might be invoked during an ongoing refresh
@@ -312,9 +344,48 @@ export class CloudSQLInstance {
312344
close(): void {
313345
this.closed = true;
314346
this.cancelRefresh();
347+
if (this.checkDomainID) {
348+
clearInterval(this.checkDomainID);
349+
this.checkDomainID = null;
350+
}
351+
for (const socket of this.sockets) {
352+
socket.destroy(
353+
new CloudSQLConnectorError({
354+
code: 'ERRCLOSED',
355+
message: 'The connector was closed.',
356+
})
357+
);
358+
}
315359
}
316360

317361
isClosed(): boolean {
318362
return this.closed;
319363
}
364+
async checkDomainChanged() {
365+
if (!this.instanceInfo.domainName) {
366+
return;
367+
}
368+
369+
const newInfo = await resolveInstanceName(
370+
undefined,
371+
this.instanceInfo.domainName
372+
);
373+
if (!isSameInstance(this.instanceInfo, newInfo)) {
374+
// Domain name changed. Close and remove, then create a new map entry.
375+
this.close();
376+
}
377+
}
378+
addSocket(socket: DestroyableSocket) {
379+
if (!this.instanceInfo.domainName) {
380+
// This was not connected by domain name. Ignore all sockets.
381+
return;
382+
}
383+
384+
// Add the socket to the list
385+
this.sockets.add(socket);
386+
// When the socket is closed, remove it.
387+
socket.once('closed', () => {
388+
this.sockets.delete(socket);
389+
});
390+
}
320391
}

‎src/connector.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ export declare interface ConnectionOptions {
4444
ipType?: IpAddressTypes;
4545
instanceConnectionName: string;
4646
domainName?: string;
47+
failoverPeriod?: number;
4748
limitRateInterval?: number;
4849
}
4950

@@ -129,6 +130,7 @@ class CloudSQLInstanceMap extends Map<string, CacheEntry> {
129130
const entry = this.get(key);
130131
if (entry) {
131132
if (entry.isResolved()) {
133+
await entry.instance?.checkDomainChanged();
132134
if (!entry.instance?.isClosed()) {
133135
// The instance is open and the domain has not changed.
134136
// use the cached instance.
@@ -154,6 +156,7 @@ class CloudSQLInstanceMap extends Map<string, CacheEntry> {
154156
ipType: opts.ipType || IpAddressTypes.PUBLIC,
155157
limitRateInterval: opts.limitRateInterval || 30 * 1000, // 30 sec
156158
sqlAdminFetcher: this.sqlAdminFetcher,
159+
failoverPeriod: opts.failoverPeriod,
157160
});
158161
this.set(key, new CacheEntry(promise));
159162

@@ -257,6 +260,9 @@ export class Connector {
257260
tlsSocket.once('secureConnect', async () => {
258261
cloudSqlInstance.setEstablishedConnection();
259262
});
263+
264+
cloudSqlInstance.addSocket(tlsSocket);
265+
260266
return tlsSocket;
261267
}
262268

0 commit comments

Comments
 (0)