Function setStatsCallback

cli/ssh.go:1540–1617

(
	ctx context.Context,
	agentConn workspacesdk.AgentConn,
	logger slog.Logger,
	networkInfoDir string,
	networkInfoInterval time.Duration,
)

}

func setStatsCallback(
	ctx context.Context,
	agentConn workspacesdk.AgentConn,
	logger slog.Logger,
	networkInfoDir string,
	networkInfoInterval time.Duration,
) (<-chan error, error) {
	fs, ok := ctx.Value("fs").(afero.Fs)
	if !ok {
		fs = afero.NewOsFs()
	}
	if err := fs.MkdirAll(networkInfoDir, 0o700); err != nil {
		return nil, xerrors.Errorf("mkdir: %w", err)
	}

	// The VS Code extension obtains the PID of the SSH process to
	// read files to display logs and network info.
	//
	// We get the parent PID because it's assumed `ssh` is calling this
	// command via the ProxyCommand SSH option.
	pid := os.Getppid()

	// The VS Code extension obtains the PID of the SSH process to
	// read the file below which contains network information to display.
	//
	// We get the parent PID because it's assumed `ssh` is calling this
	// command via the ProxyCommand SSH option.
	networkInfoFilePath := filepath.Join(networkInfoDir, fmt.Sprintf("%d.json", pid))

	var (
		firstErrTime time.Time
		errCh        = make(chan error, 1)
	)
	cb := func(start, end time.Time, virtual, _ map[netlogtype.Connection]netlogtype.Counts) {
		sendErr := func(tolerate bool, err error) {
			logger.Error(ctx, "collect network stats", slog.Error(err))
			// Tolerate up to 1 minute of errors.
			if tolerate {
				if firstErrTime.IsZero() {
					logger.Info(ctx, "tolerating network stats errors for up to 1 minute")
					firstErrTime = time.Now()
				}
				if time.Since(firstErrTime) < time.Minute {
					return
				}
			}

			select {
			case errCh <- err:
			default:
			}
		}

		stats, err := collectNetworkStats(ctx, agentConn, start, end, virtual)
		if err != nil {
			sendErr(true, err)
			return
		}