Skip to content

Commit

Permalink
chore(instr-mysql): update semantic conventions
Browse files Browse the repository at this point in the history
  • Loading branch information
david-luna committed Apr 18, 2024
1 parent 9f5a867 commit 6d71a50
Show file tree
Hide file tree
Showing 5 changed files with 40 additions and 42 deletions.
4 changes: 2 additions & 2 deletions package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@
},
"dependencies": {
"@opentelemetry/instrumentation": "^0.50.0",
"@opentelemetry/semantic-conventions": "^1.0.0",
"@opentelemetry/semantic-conventions": "^1.22.0",
"@types/mysql": "2.15.22"
},
"homepage": "https://github.com/open-telemetry/opentelemetry-js-contrib/tree/main/plugins/node/opentelemetry-instrumentation-mysql#readme"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,9 @@ import {
isWrapped,
} from '@opentelemetry/instrumentation';
import {
DbSystemValues,
SemanticAttributes,
DBSYSTEMVALUES_MYSQL,
SEMATTRS_DB_STATEMENT,
SEMATTRS_DB_SYSTEM,
} from '@opentelemetry/semantic-conventions';
import type * as mysqlTypes from 'mysql';
import { AttributeNames } from './AttributeNames';
Expand All @@ -54,7 +55,7 @@ export class MySQLInstrumentation extends InstrumentationBase<
typeof mysqlTypes
> {
static readonly COMMON_ATTRIBUTES = {
[SemanticAttributes.DB_SYSTEM]: DbSystemValues.MYSQL,
[SEMATTRS_DB_SYSTEM]: DBSYSTEMVALUES_MYSQL,
};
private _connectionsUsage!: UpDownCounter;

Expand Down Expand Up @@ -331,10 +332,7 @@ export class MySQLInstrumentation extends InstrumentationBase<
},
});

span.setAttribute(
SemanticAttributes.DB_STATEMENT,
getDbStatement(query)
);
span.setAttribute(SEMATTRS_DB_STATEMENT, getDbStatement(query));

const instrumentationConfig: MySQLInstrumentationConfig =
thisPlugin.getConfig();
Expand Down
40 changes: 19 additions & 21 deletions plugins/node/opentelemetry-instrumentation-mysql/src/utils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,14 @@
* limitations under the License.
*/

import { SpanAttributes } from '@opentelemetry/api';
import { SemanticAttributes } from '@opentelemetry/semantic-conventions';
import { Attributes } from '@opentelemetry/api';
import {
SEMATTRS_DB_CONNECTION_STRING,
SEMATTRS_DB_NAME,
SEMATTRS_DB_USER,
SEMATTRS_NET_PEER_NAME,
SEMATTRS_NET_PEER_PORT,
} from '@opentelemetry/semantic-conventions';
import type {
ConnectionConfig,
PoolActualConfig,
Expand All @@ -25,37 +31,29 @@ import type {
import type * as mysqlTypes from 'mysql';

/**
* Get an SpanAttributes map from a mysql connection config object
* Get an Attributes map from a mysql connection config object
*
* @param config ConnectionConfig
*/
export function getConnectionAttributes(
config: ConnectionConfig | PoolActualConfig
): SpanAttributes {
): Attributes {
const { host, port, database, user } = getConfig(config);
const portNumber = parseInt(port, 10);
if (!isNaN(portNumber)) {
return {
[SemanticAttributes.NET_PEER_NAME]: host,
[SemanticAttributes.NET_PEER_PORT]: portNumber,
[SemanticAttributes.DB_CONNECTION_STRING]: getJDBCString(
host,
port,
database
),
[SemanticAttributes.DB_NAME]: database,
[SemanticAttributes.DB_USER]: user,
[SEMATTRS_NET_PEER_NAME]: host,
[SEMATTRS_NET_PEER_PORT]: portNumber,
[SEMATTRS_DB_CONNECTION_STRING]: getJDBCString(host, port, database),
[SEMATTRS_DB_NAME]: database,
[SEMATTRS_DB_USER]: user,
};
}
return {
[SemanticAttributes.NET_PEER_NAME]: host,
[SemanticAttributes.DB_CONNECTION_STRING]: getJDBCString(
host,
port,
database
),
[SemanticAttributes.DB_NAME]: database,
[SemanticAttributes.DB_USER]: user,
[SEMATTRS_NET_PEER_NAME]: host,
[SEMATTRS_DB_CONNECTION_STRING]: getJDBCString(host, port, database),
[SEMATTRS_DB_NAME]: database,
[SEMATTRS_DB_USER]: user,
};
}

Expand Down
24 changes: 13 additions & 11 deletions plugins/node/opentelemetry-instrumentation-mysql/test/index.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,13 @@
import { context, Context, trace, SpanStatusCode } from '@opentelemetry/api';
import { AsyncHooksContextManager } from '@opentelemetry/context-async-hooks';
import {
DbSystemValues,
SemanticAttributes,
DBSYSTEMVALUES_MYSQL,
SEMATTRS_DB_NAME,
SEMATTRS_DB_STATEMENT,
SEMATTRS_DB_SYSTEM,
SEMATTRS_DB_USER,
SEMATTRS_NET_PEER_NAME,
SEMATTRS_NET_PEER_PORT,
} from '@opentelemetry/semantic-conventions';
import * as testUtils from '@opentelemetry/contrib-test-utils';
import {
Expand Down Expand Up @@ -868,15 +873,12 @@ function assertSpan(
values?: any,
errorMessage?: string
) {
assert.strictEqual(
span.attributes[SemanticAttributes.DB_SYSTEM],
DbSystemValues.MYSQL
);
assert.strictEqual(span.attributes[SemanticAttributes.DB_NAME], database);
assert.strictEqual(span.attributes[SemanticAttributes.NET_PEER_PORT], port);
assert.strictEqual(span.attributes[SemanticAttributes.NET_PEER_NAME], host);
assert.strictEqual(span.attributes[SemanticAttributes.DB_USER], user);
assert.strictEqual(span.attributes[SemanticAttributes.DB_STATEMENT], sql);
assert.strictEqual(span.attributes[SEMATTRS_DB_SYSTEM], DBSYSTEMVALUES_MYSQL);
assert.strictEqual(span.attributes[SEMATTRS_DB_NAME], database);
assert.strictEqual(span.attributes[SEMATTRS_NET_PEER_PORT], port);
assert.strictEqual(span.attributes[SEMATTRS_NET_PEER_NAME], host);
assert.strictEqual(span.attributes[SEMATTRS_DB_USER], user);
assert.strictEqual(span.attributes[SEMATTRS_DB_STATEMENT], sql);
if (errorMessage) {
assert.strictEqual(span.status.message, errorMessage);
assert.strictEqual(span.status.code, SpanStatusCode.ERROR);
Expand Down

0 comments on commit 6d71a50

Please sign in to comment.