Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions experimental/CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ For notes on migrating to 2.x / 0.200.x see [the upgrade guide](doc/upgrade-to-2
### :rocket: Features

* feat(otlp-transformer): add custom protobuf logs serializer [#6228](https://github.com/open-telemetry/opentelemetry-js/pull/6228) @pichlermarc
* feat(otlp-transformer): add custom protobuf logs export response deserializer [#6530](https://github.com/open-telemetry/opentelemetry-js/pull/6530) @pichlermarc

### :bug: Bug Fixes

Expand Down
7 changes: 5 additions & 2 deletions experimental/packages/otlp-transformer/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,11 @@
"compile": "tsc --build tsconfig.json tsconfig.esm.json tsconfig.esnext.json",
"clean": "tsc --build --clean tsconfig.json tsconfig.esm.json tsconfig.esnext.json",
"protos": "npm run submodule && npm run protos:generate",
"protos:generate:js": "pbjs -t static-module -p ./protos -w commonjs --null-defaults -o ./src/generated/root.js ./protos/opentelemetry/proto/common/v1/common.proto ./protos/opentelemetry/proto/resource/v1/resource.proto ./protos/opentelemetry/proto/trace/v1/trace.proto ./protos/opentelemetry/proto/collector/trace/v1/trace_service.proto ./protos/opentelemetry/proto/metrics/v1/metrics.proto ./protos/opentelemetry/proto/collector/metrics/v1/metrics_service.proto ./protos/opentelemetry/proto/logs/v1/logs.proto ./protos/opentelemetry/proto/collector/logs/v1/logs_service.proto && pbjs -t static-module -p ./test/fixtures -w commonjs --null-defaults -o ./test/generated/testbed.js ./test/fixtures/testbed.proto",
"protos:generate:ts": "pbts -o ./src/generated/root.d.ts ./src/generated/root.js && pbts -o ./test/generated/testbed.d.ts ./test/generated/testbed.js",
"protos:generate:js": "pbjs -t static-module -p ./protos -w commonjs --null-defaults -o ./src/generated/root.js ./protos/opentelemetry/proto/common/v1/common.proto ./protos/opentelemetry/proto/resource/v1/resource.proto ./protos/opentelemetry/proto/trace/v1/trace.proto ./protos/opentelemetry/proto/collector/trace/v1/trace_service.proto ./protos/opentelemetry/proto/metrics/v1/metrics.proto ./protos/opentelemetry/proto/collector/metrics/v1/metrics_service.proto && npm run protos:generate:js:tests",
"protos:generate:js:tests": "npm run protos:generate:js:testbed && npm run protos:generate:js:test-signals",
"protos:generate:js:testbed": "pbjs -t static-module -p ./test/fixtures -w commonjs --null-defaults -o ./test/generated/testbed.js ./test/fixtures/testbed.proto",
"protos:generate:js:test-signals": "pbjs -r signals -t static-module -p ./protos -w commonjs --null-defaults -o ./test/generated/signals.js ./protos/opentelemetry/proto/common/v1/common.proto ./protos/opentelemetry/proto/resource/v1/resource.proto ./protos/opentelemetry/proto/logs/v1/logs.proto ./protos/opentelemetry/proto/collector/logs/v1/logs_service.proto",
"protos:generate:ts": "pbts -o ./src/generated/root.d.ts ./src/generated/root.js && pbts -o ./test/generated/testbed.d.ts ./test/generated/testbed.js && pbts -o ./test/generated/signals.d.ts ./test/generated/signals.js",
"protos:generate": "npm run protos:generate:js && npm run protos:generate:ts",
"lint": "eslint . --ext .ts",
"lint:fix": "eslint . --ext .ts --fix",
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,109 @@
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/

/**
* Minimal binary protobuf reader.
* Only implements the wire-types that we currently need; this is not intended
* to be a general-purpose protobuf reader.
*
* Since the values we parse are generally small and not very nested, it's public
* interface does not enforce the same low-allocation philosophy that ProtobufWriter does.
* If this is needed in the future, we should refactor this to fit the use-case.
*/
export class ProtobufReader {
pos: number = 0;
private readonly _buf: Uint8Array;
private readonly _textDecoder: {
decode: (input?: Uint8Array | null) => string;
};

constructor(buf: Uint8Array) {
this._buf = buf;
this._textDecoder = new TextDecoder();
}

isAtEnd(): boolean {
return this.pos >= this._buf.length;
}

/** Read a varint and decode it as a tag, returning field number and wire type. */
readTag(): { fieldNumber: number; wireType: number } {
const raw = this.readVarint();
return { fieldNumber: raw >>> 3, wireType: raw & 0x7 };
}

/**
* Read a base-128 varint.
* Returns a JS `number`; precision above 2^53 is silently lost.
*/
readVarint(): number {
let result = 0;
let shift = 0;
while (this.pos < this._buf.length) {
const b = this._buf[this.pos++];
result += (b & 0x7f) * Math.pow(2, shift);
shift += 7;
if ((b & 0x80) === 0) break;
}
return result;
}

/** Read a length-delimited byte sequence (bytes field or embedded message). */
readBytes(): Uint8Array {
const len = this.readVarint();
const slice = this._buf.subarray(this.pos, this.pos + len);
this.pos += len;
return slice;
}

/** Read a length-delimited UTF-8 string. */
readString(): string {
return this._textDecoder.decode(this.readBytes());
}

/**
* Skip an unknown field.
* Handles wire types 0 (varint), 1 (64-bit), 2 (length-delimited),
* 3 (start-group), 4 (end-group), and 5 (32-bit).
*/
skip(wireType: number): void {
switch (wireType) {
case 0: // varint
this.readVarint();
break;
case 1: // 64-bit fixed
this.pos += 8;
break;
case 2: // length-delimited
this.readBytes();
break;
case 3: // start group (deprecated)
// We should never encounter this, but let's handle it gracefully in case we do:
// Read nested tags until matching end-group (wire type 4) is found.
// Groups can be nested, so continue until the end-group for this
// start-group is encountered.
while (!this.isAtEnd()) {
const { wireType: nestedWireType } = this.readTag();
if (nestedWireType === 4) {
// matched end-group for this start-group
break;
}
// recursive skip also handles nested groups
this.skip(nestedWireType);
}
break;
case 4: // end group
// End-group should be handled by the start-group logic above.
// When encountered directly in skip, treat it as a no-op (it signals
// termination of the enclosing group).
break;
case 5: // 32-bit fixed
this.pos += 4;
break;
default:
throw new Error(`Unknown wire type ${wireType}, cannot safely skip`);
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,12 @@
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
import * as root from '../../generated/root';
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

And almost 1MiB of JS just left the process.
... or almost (need trace and metrics as well).


import type { IExportLogsServiceResponse } from '../export-response';

import type { ReadableLogRecord } from '@opentelemetry/sdk-logs';
import type { ExportType } from '../../common/protobuf/protobuf-export-type';
import type { IExportLogsServiceResponse } from '../export-response';
import type { ISerializer } from '../../i-serializer';
import { serializeLogsExportRequest } from './logs-serializer';

const logsResponseType = root.opentelemetry.proto.collector.logs.v1
.ExportLogsServiceResponse as ExportType<IExportLogsServiceResponse>;
import { deserializeExportLogsServiceResponse } from './response-deserializer';

/*
* @experimental this serializer may receive breaking changes in minor versions, pin this package's version when using this constant
Expand All @@ -25,6 +20,6 @@ export const ProtobufLogsSerializer: ISerializer<
return serializeLogsExportRequest(arg);
},
deserializeResponse: (arg: Uint8Array) => {
return logsResponseType.decode(arg);
return deserializeExportLogsServiceResponse(arg);
},
};
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/

import type {
IExportLogsServiceResponse,
IExportLogsPartialSuccess,
} from '../export-response';
import { ProtobufReader } from '../../common/protobuf/protobuf-reader';

/**
* Parse an ExportLogsPartialSuccess embedded message from raw bytes.
*
* Field map (opentelemetry/proto/collector/logs/v1/logs_service.proto):
* 1 rejected_log_records int64 (varint)
* 2 error_message string (length-delimited)
*/
function deserializePartialSuccess(
data: Uint8Array
): IExportLogsPartialSuccess {
const reader = new ProtobufReader(data);
const result: IExportLogsPartialSuccess = {};

while (!reader.isAtEnd()) {
const { fieldNumber, wireType } = reader.readTag();
switch (fieldNumber) {
case 1: // rejected_log_records (int64, varint)
// expected wire type 0 (varint)
if (wireType === 0) {
result.rejectedLogRecords = reader.readVarint();
} else {
// unexpected wire type for this field; skip it safely
reader.skip(wireType);
}
break;
case 2: // error_message (string, length-delimited)
// expected wire type 2 (length-delimited)
if (wireType === 2) {
result.errorMessage = reader.readString();
} else {
// unexpected wire type for this field; skip it safely
reader.skip(wireType);
}
break;
default:
reader.skip(wireType);
break;
}
}

return result;
}

/**
* Parse an ExportLogsServiceResponse protobuf message from raw bytes.
*
* Field map (opentelemetry/proto/collector/logs/v1/logs_service.proto):
* 1 partial_success ExportLogsPartialSuccess (length-delimited)
*/
export function deserializeExportLogsServiceResponse(
data: Uint8Array
): IExportLogsServiceResponse {
const reader = new ProtobufReader(data);
const result: IExportLogsServiceResponse = {};

while (!reader.isAtEnd()) {
const { fieldNumber, wireType } = reader.readTag();
switch (fieldNumber) {
case 1: // partial_success (ExportLogsPartialSuccess, length-delimited)
// expected wire type 2 (length-delimited / embedded message)
if (wireType === 2) {
result.partialSuccess = deserializePartialSuccess(reader.readBytes());
} else {
// unexpected wire type for this field; skip it safely
reader.skip(wireType);
}
break;
default:
reader.skip(wireType);
break;
}
}

return result;
}
58 changes: 53 additions & 5 deletions experimental/packages/otlp-transformer/test/logs.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,16 @@ import { SeverityNumber } from '@opentelemetry/api-logs';
import type { Encoder } from '../src/common/utils';
import { JSON_ENCODER, PROTOBUF_ENCODER } from '../src/common/utils';
import { toBase64 } from './utils';
import * as root from '../src/generated/root';
import * as signals from '../test/generated/signals';
import type { IExportLogsServiceRequest } from '../src/logs/internal-types';
import { ESeverityNumber } from '../src/logs/internal-types';
import { createExportLogsServiceRequest } from '../src/logs/internal';
import { ProtobufLogsSerializer } from '../src/logs/protobuf';
import { JsonLogsSerializer } from '../src/logs/json';
import { GROWING_BUFFER_DEBUG_MESSAGE } from '../src/common/protobuf/protobuf-writer';
import {
GROWING_BUFFER_DEBUG_MESSAGE,
ProtobufWriter,
} from '../src/common/protobuf/protobuf-writer';

function createExpectedLogJson(encoder: Encoder): IExportLogsServiceRequest {
const timeUnixNano = encoder.encodeHrTime([1680253513, 123241635]);
Expand Down Expand Up @@ -360,13 +363,13 @@ describe('Logs', () => {
const serialized = ProtobufLogsSerializer.serializeRequest([log_1_1_1]);
assert.ok(serialized, 'serialized response is undefined');
const decoded =
root.opentelemetry.proto.collector.logs.v1.ExportLogsServiceRequest.decode(
signals.opentelemetry.proto.collector.logs.v1.ExportLogsServiceRequest.decode(
serialized
);

const expected = createExpectedLogProtobuf();
const decodedObj =
root.opentelemetry.proto.collector.logs.v1.ExportLogsServiceRequest.toObject(
signals.opentelemetry.proto.collector.logs.v1.ExportLogsServiceRequest.toObject(
decoded,
{
// This incurs some precision loss that's taken into account in createExpectedLogsProtobuf()
Expand All @@ -385,7 +388,7 @@ describe('Logs', () => {

it('deserializes a response', () => {
const protobufSerializedResponse =
root.opentelemetry.proto.collector.logs.v1.ExportLogsServiceResponse.encode(
signals.opentelemetry.proto.collector.logs.v1.ExportLogsServiceResponse.encode(
{
partialSuccess: {
errorMessage: 'foo',
Expand Down Expand Up @@ -415,6 +418,51 @@ describe('Logs', () => {
);
sinon.assert.neverCalledWith(diagStub, GROWING_BUFFER_DEBUG_MESSAGE);
});

it('does not throw when encountering unexpected wiretypes during deserialization', function () {
const writer = new ProtobufWriter(50);
// Construct an ExportLogsServiceResponse where the embedded
// ExportLogsPartialSuccess has fields encoded with incorrect wire types.
// ExportLogsServiceResponse { 1: partial_success (length-delimited) }
// ExportLogsPartialSuccess expects:
// 1: rejected_log_records (varint)
// 2: error_message (length-delimited string)

// first pretend the field number 1 is a varint (type 0, correct format expects a length delimited field)
writer.writeTag(1, 0);
writer.writeVarint(3);

// also pretend we have an extra field that we don't know yet what to do with
writer.writeTag(99, 0);
writer.writeVarint(42);

// now write field 1 again, but this time as length-delimited, as expected.
writer.writeTag(1, 2);
const lengthVarintPosition = writer.startLengthDelimited();
const innerStartPos = writer.pos;
// instead of putting the correct data here, we put unexpected wire-types for each field, ensuring it's handled gracefully.
// Write field 1 but with wire type 2 (length-delimited) and a string (correct format expects a varint)
writer.writeTag(1, 2);
writer.writeString('not-a-number');
// Write field 2 but with wire type 0 (varint) instead of length-delimited (corrent format expects a string, which is length delimited)
writer.writeTag(2, 0);
writer.writeVarint(12345);
// Write field 99, which is completely unknown to us; pretend it's a varint (type 0)
writer.writeTag(99, 0);
writer.writeVarint(42);

// finish up
writer.finishLengthDelimited(
lengthVarintPosition,
writer.pos - innerStartPos
);

// Ensure deserialization does not throw when encountering these
// unexpected wire types.
assert.doesNotThrow(() =>
ProtobufLogsSerializer.deserializeResponse(writer.finish())
);
});
});

describe('JsonLogsSerializer', function () {
Expand Down
Loading
Loading