New immutable type for Flags structs

Signed-off-by: Bogdan <[email protected]>

pdata/pcommon/trace_flags.go

@@ -0,0 +1,25 @@
+package pcommon
+
+const isSampledMask = uint8(1)
+
+var DefaultTraceFlags = TraceFlags(0)
+
+// TraceFlags defines the trace flags as defined by the w3c trace-context, see
+// https://www.w3.org/TR/trace-context/#trace-flags
+type TraceFlags uint8
+
+// IsSampled returns true if the LogRecordFlags contains the IsSampled flag.
+func (ms TraceFlags) IsSampled() bool {
+	return uint8(ms)&isSampledMask != 0
+}
+
+// WithIsSampled returns a new TraceFlags, with the IsSampled flag set to the given value.
+func (ms TraceFlags) WithIsSampled(b bool) TraceFlags {
+	orig := uint8(ms)
+	if b {
+		orig |= isSampledMask
+	} else {
+		orig &^= isSampledMask
+	}
+	return TraceFlags(orig)
+}

pdata/pcommon/trace_flags_test.go

@@ -0,0 +1,49 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//       http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package pcommon
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDefaultTraceFlags(t *testing.T) {
+	flags := DefaultTraceFlags
+	assert.False(t, flags.IsSampled())
+	assert.EqualValues(t, uint8(0), flags)
+
+	flags = flags.WithIsSampled(true)
+	assert.True(t, flags.IsSampled())
+	assert.EqualValues(t, uint8(1), flags)
+
+	flags = flags.WithIsSampled(false)
+	assert.False(t, flags.IsSampled())
+	assert.EqualValues(t, uint8(0), flags)
+}
+
+func TestNewTraceFlagsFromRaw(t *testing.T) {
+	flags := TraceFlags(1)
+	assert.True(t, flags.IsSampled())
+	assert.EqualValues(t, uint8(1), flags)
+
+	flags = flags.WithIsSampled(false)
+	assert.False(t, flags.IsSampled())
+	assert.EqualValues(t, uint8(0), flags)
+
+	flags = flags.WithIsSampled(true)
+	assert.True(t, flags.IsSampled())
+	assert.EqualValues(t, uint8(1), flags)
+}

pdata/plog/log_record_flags.go

@@ -0,0 +1,40 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//       http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package plog
+
+import (
+	"go.opentelemetry.io/collector/pdata/pcommon"
+)
+
+const traceFlagsMask = uint32(0xFF)
+
+var DefaultLogRecordFlags = LogRecordFlags(0)
+
+// LogRecordFlags defines flags for the LogRecord. 8 least significant bits are the trace flags as
+// defined in W3C Trace Context specification. 24 most significant bits are reserved and must be set to 0.
+type LogRecordFlags uint32
+
+// TraceFlags copies all properties from the current struct to the dest.
+func (ms LogRecordFlags) TraceFlags() pcommon.TraceFlags {
+	return pcommon.TraceFlags(uint32(ms) & traceFlagsMask)
+}
+
+// WithTraceFlags moves all properties from the current struct to dest resetting the current instance to its zero value
+func (ms LogRecordFlags) WithTraceFlags(v pcommon.TraceFlags) LogRecordFlags {
+	orig := uint32(ms)
+	orig &= ^traceFlagsMask // cleanup old trace flags
+	orig |= uint32(v)       // set new trace flags
+	return LogRecordFlags(orig)
+}

pdata/plog/log_record_flags_test.go

@@ -0,0 +1,51 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//       http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package plog
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"go.opentelemetry.io/collector/pdata/pcommon"
+)
+
+func TestDefaultLogRecordFlags(t *testing.T) {
+	flags := DefaultLogRecordFlags
+	assert.Equal(t, pcommon.DefaultTraceFlags, flags.TraceFlags())
+	assert.EqualValues(t, uint32(0), flags)
+
+	flags = flags.WithTraceFlags(pcommon.DefaultTraceFlags.WithIsSampled(true))
+	assert.True(t, flags.TraceFlags().IsSampled())
+	assert.EqualValues(t, uint32(1), flags)
+
+	flags = flags.WithTraceFlags(pcommon.DefaultTraceFlags.WithIsSampled(false))
+	assert.False(t, flags.TraceFlags().IsSampled())
+	assert.EqualValues(t, uint32(0), flags)
+}
+
+func TestLogRecordFlags(t *testing.T) {
+	flags := LogRecordFlags(1)
+	assert.True(t, flags.TraceFlags().IsSampled())
+	assert.EqualValues(t, uint32(1), flags)
+
+	flags = flags.WithTraceFlags(pcommon.DefaultTraceFlags.WithIsSampled(false))
+	assert.False(t, flags.TraceFlags().IsSampled())
+	assert.EqualValues(t, uint32(0), flags)
+
+	flags = flags.WithTraceFlags(pcommon.DefaultTraceFlags.WithIsSampled(true))
+	assert.True(t, flags.TraceFlags().IsSampled())
+	assert.EqualValues(t, uint32(1), flags)
+}

pdata/plog/logs.go

@@ -191,64 +191,3 @@ func (ms LogRecord) Flags() uint32 {
 func (ms LogRecord) SetFlags(v uint32) {
 	ms.getOrig().Flags = v
 }
-
-const (
-	traceFlagsNone = uint32(0)
-	isSampledMask  = uint32(1)
-)
-
-// LogRecordFlags defines flags for the LogRecord. 8 least significant bits are the trace flags as
-// defined in W3C Trace Context specification. 24 most significant bits are reserved and must be set to 0.
-//
-// This is a reference type, if passed by value and callee modifies it the caller will see the modification.
-//
-// Must use NewLogRecordFlags function to create new instances.
-// Important: zero-initialized instance is not valid for use.
-type LogRecordFlags internal.LogRecordFlags
-
-func newLogRecordFlags(orig *uint32) LogRecordFlags {
-	return LogRecordFlags(internal.NewLogRecordFlags(orig))
-}
-
-func (ms LogRecordFlags) getOrig() *uint32 {
-	return internal.GetOrigLogRecordFlags(internal.LogRecordFlags(ms))
-}
-
-// NewLogRecordFlags creates a new empty LogRecordFlags.
-//
-// This must be used only in testing code. Users should use "AppendEmpty" when part of a Slice,
-// OR directly access the member if this is embedded in another struct.
-func NewLogRecordFlags() LogRecordFlags {
-	return newLogRecordFlags(new(uint32))
-}
-
-// MoveTo moves all properties from the current struct to dest resetting the current instance to its zero value
-func (ms LogRecordFlags) MoveTo(dest LogRecordFlags) {
-	*dest.getOrig() = *ms.getOrig()
-	*ms.getOrig() = traceFlagsNone
-}
-
-// CopyTo copies all properties from the current struct to the dest.
-func (ms LogRecordFlags) CopyTo(dest LogRecordFlags) {
-	*dest.getOrig() = *ms.getOrig()
-}
-
-// IsSampled returns true if the LogRecordFlags contains the IsSampled flag.
-func (ms LogRecordFlags) IsSampled() bool {
-	return *ms.getOrig()&isSampledMask != 0
-}
-
-// SetIsSampled sets the IsSampled flag if true and removes it if false.
-// Setting this Flag when it is already set is a no-op.
-func (ms LogRecordFlags) SetIsSampled(b bool) {
-	if b {
-		*ms.getOrig() |= isSampledMask
-	} else {
-		*ms.getOrig() &^= isSampledMask
-	}
-}
-
-// AsRaw converts LogRecordFlags to the OTLP uint32 representation.
-func (ms LogRecordFlags) AsRaw() uint32 {
-	return *ms.getOrig()
-}

pdata/plog/logs_test.go

@@ -132,33 +132,6 @@ func TestLogsClone(t *testing.T) {
 	assert.EqualValues(t, logs, logs.Clone())
 }
 
-func TestLogRecordFlags(t *testing.T) {
-	flags := LogRecordFlags(internal.GenerateTestLogRecordFlags())
-	assert.True(t, flags.IsSampled())
-	assert.Equal(t, uint32(1), flags.AsRaw())
-
-	flags.SetIsSampled(false)
-	flags.SetIsSampled(false)
-	assert.False(t, flags.IsSampled())
-	assert.Equal(t, uint32(0), flags.AsRaw())
-
-	flags.SetIsSampled(true)
-	flags.SetIsSampled(true)
-	assert.True(t, flags.IsSampled())
-	assert.Equal(t, uint32(1), flags.AsRaw())
-
-	moveFlags := NewLogRecordFlags()
-	assert.False(t, moveFlags.IsSampled())
-
-	flags.MoveTo(moveFlags)
-	assert.False(t, flags.IsSampled())
-	assert.True(t, moveFlags.IsSampled())
-
-	moveFlags.CopyTo(flags)
-	assert.True(t, flags.IsSampled())
-	assert.True(t, moveFlags.IsSampled())
-}
-
 func BenchmarkLogsClone(b *testing.B) {
 	logs := NewLogs()
 	internal.FillTestResourceLogsSlice(internal.ResourceLogsSlice(logs.ResourceLogs()))