[pdata/pprofile] add ProfileCount() (#14239)

<!--Ex. Fixing a bug - Describe the bug and how this fixes the issue.
Ex. Adding a feature - Explain what this achieves.-->
#### Description


Similar to `(Logs) LogRecordCount() int`, `(Metrics) MetricCount() int`
and `(Traces) SpanCount() int` add `ProfileCount()`.
<!-- Issue number if applicable -->
#### Link to tracking issue
Fixes #

<!--Describe what testing was performed and which tests were added.-->
#### Testing

<!--Describe the documentation added.-->
#### Documentation

<!--Please delete paragraphs that you did not use before submitting.-->

Signed-off-by: Florian Lehner <[email protected]>

Author: florianl

.chloggen/profiles-count.yaml

@@ -0,0 +1,25 @@
+# Use this changelog template to create an entry for release notes.
+
+# One of 'breaking', 'deprecation', 'new_component', 'enhancement', 'bug_fix'
+change_type: 'enhancement'
+
+# The name of the component, or a single word describing the area of concern, (e.g. receiver/otlp)
+component: pdata/pprofile
+
+# A brief description of the change.  Surround your text with quotes ("") if it needs to start with a backtick (`).
+note: add ProfileCount()
+
+# One or more tracking issues or pull requests related to the change
+issues: [14239]
+
+# (Optional) One or more lines of additional information to render under the primary note.
+# These lines will be padded with 2 spaces and then inserted directly into the document.
+# Use pipe (|) for multiline entries.
+subtext:
+
+# Optional: The change log or logs in which this entry should be included.
+# e.g. '[user]' or '[user, api]'
+# Include 'user' if the change is relevant to end users.
+# Include 'api' if there is a change to a library API.
+# Default: '[user]'
+change_logs: [api]

pdata/pprofile/profiles.go

@@ -44,3 +44,17 @@ func (ms Profiles) switchDictionary(src, dst ProfilesDictionary) error {
 
 	return nil
 }
+
+// ProfileCount calculates the total number of profile records.
+func (ms Profiles) ProfileCount() int {
+	profileCount := 0
+	rps := ms.ResourceProfiles()
+	for i := 0; i < rps.Len(); i++ {
+		rp := rps.At(i)
+		sps := rp.ScopeProfiles()
+		for j := 0; j < sps.Len(); j++ {
+			profileCount += sps.At(j).Profiles().Len()
+		}
+	}
+	return profileCount
+}

pdata/pprofile/profiles_test.go

@@ -61,6 +61,42 @@ func TestSampleCount(t *testing.T) {
 	assert.Equal(t, 7, pd.SampleCount())
 }
 
+func TestProfileCount(t *testing.T) {
+	pd := NewProfiles()
+	assert.Equal(t, 0, pd.ProfileCount())
+
+	rs := pd.ResourceProfiles().AppendEmpty()
+	assert.Equal(t, 0, pd.ProfileCount())
+
+	ils := rs.ScopeProfiles().AppendEmpty()
+	assert.Equal(t, 0, pd.ProfileCount())
+
+	ps := ils.Profiles().AppendEmpty()
+	assert.Equal(t, 1, pd.ProfileCount())
+
+	ps.Samples().AppendEmpty()
+	assert.Equal(t, 1, pd.ProfileCount())
+
+	ils2 := rs.ScopeProfiles().AppendEmpty()
+	assert.Equal(t, 1, pd.ProfileCount())
+
+	ps2 := ils2.Profiles().AppendEmpty()
+	assert.Equal(t, 2, pd.ProfileCount())
+
+	ps2.Samples().AppendEmpty()
+	assert.Equal(t, 2, pd.ProfileCount())
+
+	rms := pd.ResourceProfiles()
+	rms.EnsureCapacity(3)
+	rms.AppendEmpty().ScopeProfiles().AppendEmpty()
+	ilss := rms.AppendEmpty().ScopeProfiles().AppendEmpty().Profiles().AppendEmpty().Samples()
+	for range 5 {
+		ilss.AppendEmpty()
+	}
+	// 5 + 2 (from rms.At(0) and rms.At(1) initialized first)
+	assert.Equal(t, 3, pd.ProfileCount())
+}
+
 func TestSampleCountWithEmpty(t *testing.T) {
 	assert.Equal(t, 0, newProfiles(&internal.ExportProfilesServiceRequest{
 		ResourceProfiles: []*internal.ResourceProfiles{{}},