STACKITGit/cli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Bump copilot telemetry sampling to 100%

Browse source 
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
William Martin 2026-05-07 17:20:03 +02:00
parent 3c162a78ef
commit 1caa3b7475
3 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
pkg/cmd/copilot/copilot.go
View file

@ -20,6 +20,7 @@ import (
"github.com/MakeNowJust/heredoc"
"github.com/cli/cli/v2/internal/ci"
"github.com/cli/cli/v2/internal/config"
"github.com/cli/cli/v2/internal/gh/ghtelemetry"
"github.com/cli/cli/v2/internal/prompter"
"github.com/cli/cli/v2/internal/safepaths"
ghzip "github.com/cli/cli/v2/internal/zip"
@ -37,7 +38,7 @@ type CopilotOptions struct {
Remove bool
}
func NewCmdCopilot(f *cmdutil.Factory, runF func(*CopilotOptions) error) *cobra.Command {
func NewCmdCopilot(f *cmdutil.Factory, telemetry ghtelemetry.CommandRecorder, runF func(*CopilotOptions) error) *cobra.Command {
opts := &CopilotOptions{
IO: f.IOStreams,
HttpClient: f.HttpClient,
@ -80,6 +81,8 @@ func NewCmdCopilot(f *cmdutil.Factory, runF func(*CopilotOptions) error) *cobra.
`),
DisableFlagParsing: true,
RunE: func(cmd *cobra.Command, args []string) error {
telemetry.SetSampleRate(ghtelemetry.SAMPLE_ALL)
stopParsePos := -1
for i, arg := range args {
if arg == "--" {

20
pkg/cmd/copilot/copilot_test.go
View file

@ -14,6 +14,8 @@ import (
"runtime"
"testing"
"github.com/cli/cli/v2/internal/gh/ghtelemetry"
"github.com/cli/cli/v2/internal/telemetry"
"github.com/cli/cli/v2/pkg/cmdutil"
"github.com/cli/cli/v2/pkg/httpmock"
"github.com/cli/cli/v2/pkg/iostreams"
@ -110,7 +112,7 @@ func TestNewCmdCopilot(t *testing.T) {
assert.NoError(t, err)
var gotOpts *CopilotOptions
cmd := NewCmdCopilot(f, func(opts *CopilotOptions) error {
cmd := NewCmdCopilot(f, &telemetry.CommandRecorderSpy{}, func(opts *CopilotOptions) error {
gotOpts = opts
return nil
})
@ -586,3 +588,19 @@ func TestDownloadCopilot(t *testing.T) {
require.Equal(t, localPath, path, "downloadCopilot() path mismatch")
})
}
func TestCopilotCommandIsSampledAt100(t *testing.T) {
spy := &telemetry.CommandRecorderSpy{}
factory := &cmdutil.Factory{}
cmd := NewCmdCopilot(factory, spy, func(opts *CopilotOptions) error {
return nil
})
cmd.SetArgs([]string{})
cmd.SetIn(&bytes.Buffer{})
cmd.SetOut(&bytes.Buffer{})
cmd.SetErr(&bytes.Buffer{})
_, err := cmd.ExecuteC()
require.NoError(t, err)
require.Equal(t, ghtelemetry.SAMPLE_ALL, spy.LastSampleRate)
}

2
pkg/cmd/root/root.go
View file

@ -152,7 +152,7 @@ func NewCmdRoot(f *cmdutil.Factory, telemetry ghtelemetry.CommandRecorder, versi
cmd.AddCommand(skillsCmd.NewCmdSkills(f, telemetry))
// Root commands with standalone functionality and no subcommands
cmd.AddCommand(copilotCmd.NewCmdCopilot(f, nil))
cmd.AddCommand(copilotCmd.NewCmdCopilot(f, telemetry, nil))
cmd.AddCommand(statusCmd.NewCmdStatus(f, nil))
cmd.AddCommand(creditsCmd.NewCmdCredits(f, nil))
cmd.AddCommand(licensesCmd.NewCmdLicenses(f))