隔离字符串输出

I currently have a script that performs an os command, that returns a great deal of data, at the end of the data it gives a total such that:

N Total.

N can be any number from 0 upward.

I want to perform this command, and take N then put it into a value. I have the command running and I'm storing it in a bytes.Buffer, however I'm unsure how to scrape this so that I only get the number. The "N Total." string is always at the end of the output. Any help would be appreciated as I've seen various different methods but they all seem quite convoluted.

You can split the string by and get the last line.

package main

import (
    "fmt"
    "strconv"
    "strings"
)

func main() {
    output := `
Some os output
Some more os output
Again some os output
1001 Total`

    // If you're getting the string from the bytes.Buffer do this:
    // output := myBytesBuffer.String()

    outputSplit := strings.Split(output, "
") // Break into lines

    // Get last line from the end.
    // -1 assumes the numbers in the last line. Change it if its not.
    lastLine := outputSplit[len(outputSplit)-1]

    lastLine = strings.Replace(lastLine, " Total", "", -1) // Remove text

    number, _ := strconv.Atoi(lastLine) // Convert from text to number

    fmt.Println(number)
}

peterSO points out that for big output the above may be slow.

Here's another way that uses a compiled regexp expression to match against a small subset of bytes.

package main

import (
    "bytes"
    "fmt"
    "os/exec"
    "regexp"
    "strconv"
)

func main() {
    // Create regular expression. You only create this once.
    // Would be regexpNumber := regexp.MustCompile(`(\d+) Total`) for you
    regexpNumber := regexp.MustCompile(`(\d+) bits physical`)

    // Whatever your os command is
    command := exec.Command("cat", "/proc/cpuinfo")
    output, _ := command.Output()

    // Your bytes.Buffer
    var b bytes.Buffer
    b.Write(output)

    // Get end of bytes slice
    var end []byte
    if b.Len()-200 > 0 {
        end = b.Bytes()[b.Len()-200:]
    } else {
        end = b.Bytes()
    }

    // Get matches. matches[1] contains your number
    matches := regexpNumber.FindSubmatch(end)

    // Convert bytes to int
    number, _ := strconv.Atoi(string(matches[1])) // Convert from text to number
    fmt.Println(number)
}

You can use a bufio.Scanner to read the command's output line-wise. Then just remember the last line and parse it once the command has finished.

package main

import (
    "bufio"
    "fmt"
    "io"
    "os/exec"
    "strings"
)

func main() {
    r, w := io.Pipe()

    cmd := exec.Command("fortune")
    cmd.Stdout = w
    go func() {
            cmd.Run()
            r.Close()
            w.Close()
    }()

    sc := bufio.NewScanner(r)
    var lastLine string
    for sc.Scan() {
            line := sc.Text()
            fmt.Println("debug:", line)

            if strings.TrimSpace(line) != "" {
                    lastLine = line
            }
    }

    fmt.Println(lastLine)
}

Sample output:

debug: "Get back to your stations!"
debug: "We're beaming down to the planet, sir."
debug: -- Kirk and Mr. Leslie, "This Side of Paradise",
debug: stardate 3417.3
stardate 3417.3

Parsing lastLine is left as an excercise for the reader.