adding iterator interface and removing GetAll() method from attribute list

instrumentation/opentelemetry/span.go

@@ -31,13 +31,12 @@
 	return nil
 }
 
-func (l *AttributeList) GetAll() []sdk.Attribute {
-	size := len(l.attrs)
-	attributes := make([]sdk.Attribute, size)
-	for i := 0; i < size; i++ {
-		attributes[i] = sdk.Attribute{Key: string(l.attrs[i].Key), Value: l.attrs[i].Value.AsInterface()}
+func (l *AttributeList) Iterate(yield func(key string, value interface{}) bool) {
+	for _, attr := range l.attrs {
+		if !yield(string(attr.Key), attr.Value.AsInterface()) {
+			return
+		}
 	}
-	return attributes
 }
 
 var _ sdk.Span = (*Span)(nil)

instrumentation/opentelemetry/span_test.go

@@ -110,7 +110,7 @@ func TestGetAttributes(t *testing.T) {
 	assert.Equal(t, nil, attrs.GetValue("non_existent"))
 }
 
-func TestGetAllAttributes(t *testing.T) {
+func TestIterate(t *testing.T) {
 	sampler := sdktrace.AlwaysSample()
 	tp := sdktrace.NewTracerProvider(
 		sdktrace.WithSampler(sampler),
@@ -119,15 +119,17 @@ func TestGetAllAttributes(t *testing.T) {
 	_, s, _ := StartSpan(context.Background(), "test_span", &sdk.SpanOptions{})
 	s.SetAttribute("k1", "v1")
 	s.SetAttribute("k2", 200)
-	attrs := s.GetAttributes().GetAll()
 
-	// service.instance.id is added implicitly in StartSpan so 3 attributes will be present.
-	assert.Equal(t, 3, len(attrs))
-	for _, attr := range attrs {
-		if attr.Key == "k1" {
-			assert.Equal(t, "v1", fmt.Sprintf("%v", attr.Value))
-		} else if attr.Key == "k2" {
-			assert.Equal(t, "200", fmt.Sprintf("%v", attr.Value))
+	numAttrs := 0
+	s.GetAttributes().Iterate(func(key string, value interface{}) bool {
+		if key == "k1" {
+			assert.Equal(t, "v1", fmt.Sprintf("%v", value))
+		} else if key == "k2" {
+			assert.Equal(t, "200", fmt.Sprintf("%v", value))
 		}
-	}
+		numAttrs++
+		return true
+	})
+	// service.instance.id is added implicitly in StartSpan so 3 attributes will be present.
+	assert.Equal(t, 3, numAttrs)
 }

sdk/internal/mock/span.go

@@ -29,15 +29,12 @@ func (l *AttributeList) GetValue(key string) interface{} {
 	return l.attrs[key]
 }
 
-func (l *AttributeList) GetAll() []sdk.Attribute {
-
-	attributes := make([]sdk.Attribute, len(l.attrs))
-	i := 0
+func (l *AttributeList) Iterate(yield func(key string, value interface{}) bool) {
 	for key, value := range l.attrs {
-		attributes[i] = sdk.Attribute{Key: key, Value: value}
-		i++
+		if !yield(key, value) {
+			return
+		}
 	}
-	return attributes
 }
 
 var _ sdk.Span = &Span{}

sdk/span.go

@@ -5,14 +5,12 @@ import (
 	"time"
 )
 
-type Attribute struct {
-	Key   string
-	Value interface{}
-}
-
 type AttributeList interface {
 	GetValue(key string) interface{}
-	GetAll() []Attribute
+
+	// Iterate loops through the attributes list and applies the yield function on each attribute.
+	// If the yield function returns false, we exit the loop.
+	Iterate(yield func(key string, value interface{}) bool)
 }
 
 // Span is an interface that accepts attributes and can be