Skip to content

Commit

Permalink
Merge branch 'main' into index_more_types
Browse files Browse the repository at this point in the history
  • Loading branch information
znd4 authored Nov 16, 2024
2 parents d2c1db3 + fbad29c commit 598ba77
Show file tree
Hide file tree
Showing 11 changed files with 130 additions and 13 deletions.
27 changes: 27 additions & 0 deletions .chloggen/feat_githubgen_skipmembercheck.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# 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. filelogreceiver)
component: cmd/githubgen

# A brief description of the change. Surround your text with quotes ("") if it needs to start with a backtick (`).
note: Adds a flag to skip checking GitHub organization membership for CODEOWNERS

# Mandatory: One or more tracking issues related to the change. You can use the PR number here if no issue exists.
issues: [36263]

# (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:

# If your change doesn't affect end users or the exported elements of any package,
# you should instead start your pull request title with [chore] or use the "Skip Changelog" label.
# 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: [user]
27 changes: 27 additions & 0 deletions .chloggen/span-rpocessor-keep-original-name.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# 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. filelogreceiver)
component: processor/spanprocessor

# A brief description of the change. Surround your text with quotes ("") if it needs to start with a backtick (`).
note: "Add a new configuration option to keep the original span name when extracting attributes from the span name."

# Mandatory: One or more tracking issues related to the change. You can use the PR number here if no issue exists.
issues: [36120]

# (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:

# If your change doesn't affect end users or the exported elements of any package,
# you should instead start your pull request title with [chore] or use the "Skip Changelog" label.
# 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: [user]
2 changes: 1 addition & 1 deletion cmd/githubgen/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ $> GITHUB_TOKEN=<mypattoken> githubgen --folder . [--allowlist cmd/githubgen/all

## Checking codeowners against OpenTelemetry membership via Github API

To authenticate, set the environment variable `GITHUB_TOKEN` to a PAT token.
To authenticate, set the environment variable `GITHUB_TOKEN` to a PAT token. If a PAT is not available you can use the `--skipgithub` flag to avoid checking for membership in the GitHub organization.

For each codeowner, the script will check if the user is registered as a member of the OpenTelemetry organization.

Expand Down
11 changes: 8 additions & 3 deletions cmd/githubgen/codeowners.go
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,7 @@ const distributionCodeownersHeader = `
`

type codeownersGenerator struct {
skipGithub bool
}

func (cg codeownersGenerator) generate(data *githubData) error {
Expand All @@ -91,7 +92,7 @@ func (cg codeownersGenerator) generate(data *githubData) error {
}
var missingCodeowners []string
var duplicateCodeowners []string
members, err := getGithubMembers()
members, err := cg.getGithubMembers()
if err != nil {
return err
}
Expand All @@ -109,7 +110,7 @@ func (cg codeownersGenerator) generate(data *githubData) error {
duplicateCodeowners = append(duplicateCodeowners, codeowner)
}
}
if len(missingCodeowners) > 0 {
if len(missingCodeowners) > 0 && !cg.skipGithub {
sort.Strings(missingCodeowners)
return fmt.Errorf("codeowners are not members: %s", strings.Join(missingCodeowners, ", "))
}
Expand Down Expand Up @@ -189,7 +190,11 @@ LOOP:
return nil
}

func getGithubMembers() (map[string]struct{}, error) {
func (cg codeownersGenerator) getGithubMembers() (map[string]struct{}, error) {
if cg.skipGithub {
// don't try to get organization members if no token is expected
return map[string]struct{}{}, nil
}
githubToken := os.Getenv("GITHUB_TOKEN")
if githubToken == "" {
return nil, fmt.Errorf("Set the environment variable `GITHUB_TOKEN` to a PAT token to authenticate")
Expand Down
5 changes: 3 additions & 2 deletions cmd/githubgen/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -32,22 +32,23 @@ type generator interface {
func main() {
folder := flag.String("folder", ".", "folder investigated for codeowners")
allowlistFilePath := flag.String("allowlist", "cmd/githubgen/allowlist.txt", "path to a file containing an allowlist of members outside the OpenTelemetry organization")
skipGithubCheck := flag.Bool("skipgithub", false, "skip checking GitHub membership check for CODEOWNERS generator")
flag.Parse()
var generators []generator
for _, arg := range flag.Args() {
switch arg {
case "issue-templates":
generators = append(generators, issueTemplatesGenerator{})
case "codeowners":
generators = append(generators, codeownersGenerator{})
generators = append(generators, codeownersGenerator{skipGithub: *skipGithubCheck})
case "distributions":
generators = append(generators, distributionsGenerator{})
default:
panic(fmt.Sprintf("Unknown generator: %s", arg))
}
}
if len(generators) == 0 {
generators = []generator{issueTemplatesGenerator{}, codeownersGenerator{}}
generators = []generator{issueTemplatesGenerator{}, codeownersGenerator{skipGithub: *skipGithubCheck}}
}
if err := run(*folder, *allowlistFilePath, generators); err != nil {
log.Fatal(err)
Expand Down
14 changes: 13 additions & 1 deletion processor/spanprocessor/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,9 @@ span name that is the output after processing the previous rule.
- `break_after_match` (default = false): specifies if processing of rules should stop after the first
match. If it is false rule processing will continue to be performed over the
modified span name.
- `keep_original_name` (default = false): specifies if the original span name should be kept after
processing the rules. If it is true, the original span name will be kept,
otherwise it will be replaced with the placeholders of the captured attributes.

```yaml
span/to_attributes:
Expand All @@ -92,7 +95,7 @@ span/to_attributes:
- regexp-rule3
...
break_after_match: <true|false>
keep_original_name: <true|false>
```

Example:
Expand All @@ -106,6 +109,15 @@ span/to_attributes:
to_attributes:
rules:
- ^\/api\/v1\/document\/(?P<documentId>.*)\/update$
# This example will add the same new "documentId"="12345678" attribute,
# but now resulting in an unchanged span name (/api/v1/document/12345678/update).
span/to_attributes_keep_original_name:
name:
to_attributes:
keep_original_name: true
rules:
- ^\/api\/v1\/document\/(?P<documentId>.*)\/update$
```

### Set status for span
Expand Down
5 changes: 5 additions & 0 deletions processor/spanprocessor/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,11 @@ type ToAttributes struct {
// match. If it is false rule processing will continue to be performed over the
// modified span name.
BreakAfterMatch bool `mapstructure:"break_after_match"`

// KeepOriginalName specifies if the original span name should be kept after
// processing the rules. If it is true the original span name will be kept,
// otherwise it will be replaced with the placeholders of the captured attributes.
KeepOriginalName bool `mapstructure:"keep_original_name"`
}

type Status struct {
Expand Down
14 changes: 13 additions & 1 deletion processor/spanprocessor/config_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,19 @@ func TestLoadingConfig(t *testing.T) {
expected: &Config{
Rename: Name{
ToAttributes: &ToAttributes{
Rules: []string{`^\/api\/v1\/document\/(?P<documentId>.*)\/update$`},
Rules: []string{`^\/api\/v1\/document\/(?P<documentId>.*)\/update$`},
KeepOriginalName: false,
},
},
},
},
{
id: component.MustNewIDWithName("span", "to_attributes_keep_original_name"),
expected: &Config{
Rename: Name{
ToAttributes: &ToAttributes{
Rules: []string{`^\/api\/v1\/document\/(?P<documentId>.*)\/update$`},
KeepOriginalName: true,
},
},
},
Expand Down
4 changes: 3 additions & 1 deletion processor/spanprocessor/span.go
Original file line number Diff line number Diff line change
Expand Up @@ -213,7 +213,9 @@ func (sp *spanProcessor) processToAttributes(span ptrace.Span) {
}

// Set new span name.
span.SetName(sb.String())
if !sp.config.Rename.ToAttributes.KeepOriginalName {
span.SetName(sb.String())
}

if sp.config.Rename.ToAttributes.BreakAfterMatch {
// Stop processing, break after first match is requested.
Expand Down
25 changes: 21 additions & 4 deletions processor/spanprocessor/span_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -414,8 +414,9 @@ func TestSpanProcessor_NilName(t *testing.T) {
// TestSpanProcessor_ToAttributes
func TestSpanProcessor_ToAttributes(t *testing.T) {
testCases := []struct {
rules []string
breakAfterMatch bool
rules []string
breakAfterMatch bool
keepOriginalName bool
testCase
}{
{
Expand Down Expand Up @@ -460,9 +461,24 @@ func TestSpanProcessor_ToAttributes(t *testing.T) {

{
rules: []string{
`^\/api\/v1\/document\/(?P<documentId>.*)\/update\/4$`,
`^\/api\/(?P<version>.*)\/document\/(?P<documentId>.*)\/update\/4$`,
`^\/api\/.*\/document\/(?P<documentId>.*)\/update\/3$`,
`^\/api\/(?P<version>.*)\/document\/.*\/update\/3$`},
testCase: testCase{
inputName: "/api/v1/document/321083210/update/3",
outputName: "/api/v1/document/321083210/update/3",
outputAttributes: map[string]any{
"documentId": "321083210",
"version": "v1",
},
},
breakAfterMatch: false,
keepOriginalName: true,
},

{
rules: []string{
`^\/api\/v1\/document\/(?P<documentId>.*)\/update\/4$`,
`^\/api\/(?P<version>.*)\/document\/(?P<documentId>.*)\/update\/4$`},
testCase: testCase{
inputName: "/api/v1/document/321083210/update/4",
outputName: "/api/v1/document/{documentId}/update/4",
Expand Down Expand Up @@ -491,6 +507,7 @@ func TestSpanProcessor_ToAttributes(t *testing.T) {
for _, tc := range testCases {
oCfg.Rename.ToAttributes.Rules = tc.rules
oCfg.Rename.ToAttributes.BreakAfterMatch = tc.breakAfterMatch
oCfg.Rename.ToAttributes.KeepOriginalName = tc.keepOriginalName
tp, err := factory.CreateTraces(context.Background(), processortest.NewNopSettings(), oCfg, consumertest.NewNop())
require.NoError(t, err)
require.NotNil(t, tp)
Expand Down
9 changes: 9 additions & 0 deletions processor/spanprocessor/testdata/config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,15 @@ span/to_attributes:
rules:
- ^\/api\/v1\/document\/(?P<documentId>.*)\/update$

# This example will add the same new "documentId"="12345678" attribute,
# but now resulting in an unchanged span name (/api/v1/document/12345678/update).
span/to_attributes_keep_original_name:
name:
to_attributes:
keep_original_name: true
rules:
- ^\/api\/v1\/document\/(?P<documentId>.*)\/update$

# The following demonstrates renaming the span name to `{operation_website}`
# and adding the attribute {Key: operation_website, Value: <old span name> }
# when the span has the following properties
Expand Down

0 comments on commit 598ba77

Please sign in to comment.