Golang bufio.NewScanner函数代码示例

func populateGridFromStdin() [GRID_ROW_SIZE][GRID_COL_SIZE]int {

	var grid [GRID_ROW_SIZE][GRID_COL_SIZE]int

	i := 0

	inputScanner := bufio.NewScanner(os.Stdin)

	for inputScanner.Scan() {

		j := 0

		lineScanner := bufio.NewScanner(strings.NewReader(inputScanner.Text()))
		lineScanner.Split(bufio.ScanWords)

		for lineScanner.Scan() {

			grid[i][j], _ = strconv.Atoi(lineScanner.Text())
			j++

		}

		if j == 0 {
			break
		}

		i++
	}

	return grid
}

// RecordResponse writes response's status, headers and body.
func (r *APIB) RecordResponse(resp *http.Response, status, headers, body []byte, wc io.WriteCloser) (err error) {
	indent := strings.Repeat(" ", 12)

	// indent headers
	var headersS []string
	s := bufio.NewScanner(bytes.NewReader(headers))
	for s.Scan() {
		headersS = append(headersS, indent+s.Text())
	}
	if err = s.Err(); err != nil {
		return
	}

	// indent body
	var bodyS []string
	s = bufio.NewScanner(bytes.NewReader(body))
	for s.Scan() {
		bodyS = append(bodyS, indent+s.Text())
	}
	if err = s.Err(); err != nil {
		return
	}

	err = apibResponseTemplate.Execute(wc, map[string]interface{}{
		"StatusCode": resp.StatusCode,
		"Headers":    strings.Join(headersS, "\n"),
		"Body":       strings.Join(bodyS, "\n"),
	})
	if err == nil {
		err = wc.Close()
	}
	return
}

func MergeColumns() {
	f1, err := os.Open(COL1_FILENAME)
	if err != nil {
		log.Fatalln("open col1.txt error:", err)
	}
	defer f1.Close()
	s1 := bufio.NewScanner(f1)
	f2, err := os.Open(COL2_FILENAME)
	if err != nil {
		log.Fatalln("open col2.txt error:", err)
	}
	defer f2.Close()
	s2 := bufio.NewScanner(f2)

	mf, err := os.Create(MERGED_FILENAME)
	if err != nil {
		log.Fatalln("create merged.txt error:", err)
	}
	defer mf.Close()

	for s1.Scan() && s2.Scan() {
		c1 := s1.Text()
		c2 := s2.Text()
		fmt.Fprintf(mf, "%s\t%s\n", c1, c2)
	}
}

func LogCommand(cmd *exec.Cmd, asDebug bool) {
	logrus.WithField("command", cmd.Args).Debugf("running command")
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		return
	}
	stderr, err := cmd.StderrPipe()
	if err != nil {
		return
	}
	go func() {
		in := bufio.NewScanner(stdout)
		for in.Scan() {
			if asDebug {
				logrus.Debugf(in.Text())
			} else {
				logrus.Infof(in.Text())
			}
		}
	}()
	go func() {
		in := bufio.NewScanner(stderr)
		for in.Scan() {
			logrus.Debugf(in.Text())
		}
	}()
}

func runClient(addr string) {
	conn, err := net.Dial("tcp", addr)
	if err != nil {
		log.Fatal(err)
	}

	defer conn.Close()

	// Messages from the conn to stdout.
	// The scanner will exit when the conn is closed.
	// We close the conn via the defer.
	go func() {
		s := bufio.NewScanner(conn)
		for s.Scan() {
			fmt.Fprintf(os.Stdout, s.Text()+"\n")
		}
	}()

	// Messages from stdin to the conn.
	// The scanner will exit when stdin is closed.
	// We close stdin via ctrl-D.
	s := bufio.NewScanner(os.Stdin)
	for s.Scan() {
		fmt.Fprintf(conn, s.Text()+"\n")
	}
}

func readNames(names []string) ([]string, error) {
	if len(inputFile) == 0 {
		return names, nil
	}

	var scanner *bufio.Scanner
	if inputFile != "-" {
		in, err := os.Open(inputFile)
		if err != nil {
			return nil, err
		}

		defer in.Close()
		scanner = bufio.NewScanner(in)
	} else {
		scanner = bufio.NewScanner(os.Stdin)
	}

	for scanner.Scan() {
		names = append(names, scanner.Text())
	}
	if err := scanner.Err(); err != nil {
		return nil, err
	}

	return names, nil
}

func (c *Client) listen() {
	r := bufio.NewScanner(c)
	res := time.Now()
	for {
		select {
		case <-c.quit:
			c.Close()
			return

		default:
			c.SetReadDeadline(time.Now().Add(120 * time.Second))
			if !r.Scan() {
				if ne, ok := r.Err().(net.Error); ok {
					if ne.Temporary() && ne.Timeout() && time.Since(res) < 300*time.Second {
						r = bufio.NewScanner(c)
						c.sout("PING %s", c.Host)
						continue
					}
				}
				c.err <- r.Err()
				return
			}
			c.psrv(r.Text())
			res = time.Now()
		}
	}
}

// I actually wanted to have a test containing all the words used in the
// original paper but Martin Porter (author of the original algorithm)
// has provided a massive collection of ~23000 words with the expected
// output of the algorithm so we can test our implementation against
// the original paper implementation
func TestCorpus(t *testing.T) {
	input, errIn := os.Open("corpus/test_input.txt")
	output, errOut := os.Open("corpus/test_output.txt")

	defer input.Close()
	defer output.Close()

	if errIn != nil || errOut != nil {
		t.Fatalf("Could not read input or output test files [%s, %s]", errIn, errOut)
	}

	inScan := bufio.NewScanner(input)
	outScan := bufio.NewScanner(output)

	for inScan.Scan() && outScan.Scan() {
		in := inScan.Text()
		out := outScan.Text()
		stemmed := Stem(in)

		t.Logf("[PASS] Input: %s → Expected: %s, Stemmed: %s\n", in, out, stemmed)

		if !strings.EqualFold(out, stemmed) {
			t.Errorf("[FAIL] Expected %s but got %s for input %s\n", out, stemmed, in)
		}
	}

	if inScan.Err() != nil || outScan.Err() != nil {
		t.Fatalf("Could not open scanner for input or output test files [%s, %s]", inScan.Err(), outScan.Err())
	}
}

// Secure chat client
func ExampleSecureConn_client() error {
	// connect to remote server
	c, err := net.Dial("tcp", "localhost:8080")
	if err != nil {
		return err
	}
	defer c.Close()

	// wrap insecure client, perform key exchange and retrieve
	// a secured connection
	sc, err := secureconn.New(c)
	if err != nil {
		return err
	}
	defer sc.Close()

	// read messages from the server line by line
	go func() {
		scanner := bufio.NewScanner(sc)
		for scanner.Scan() {
			fmt.Println(scanner.Text())
		}
	}()

	// read messages from command line to send to server
	scanner := bufio.NewScanner(os.Stdin)
	for scanner.Scan() {
		_, err := sc.Write([]byte(scanner.Text() + "\n"))
		if err != nil {
			return err
		}
	}

	return nil
}

func (tm *TestKeeper) Start() error {
	if tm.cmd != nil {
		panic(fmt.Errorf("tm: %s, cmd not cleanly stopped", tm.id))
	}
	tm.cmd = exec.Command(tm.keeperBin, tm.args...)
	stdoutPipe, err := tm.cmd.StdoutPipe()
	if err != nil {
		return err
	}
	go func() {
		scanner := bufio.NewScanner(stdoutPipe)
		for scanner.Scan() {
			fmt.Printf("[%s]: %s\n", tm.id, scanner.Text())
		}
	}()

	stderrPipe, err := tm.cmd.StderrPipe()
	if err != nil {
		return err
	}
	go func() {
		scanner := bufio.NewScanner(stderrPipe)
		for scanner.Scan() {
			fmt.Printf("[%s]: %s\n", tm.id, scanner.Text())
		}
	}()
	err = tm.cmd.Start()
	if err != nil {
		return err
	}
	return nil
}

// Creates new Trie struct loaded from filename
func Load(filename string, maxLength int) (*Trie, error) {
	file, err := os.Open(filename)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	cfs := make(map[rune]int)
	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		w := scanner.Text()
		if maxLength < 1 || len(w) < maxLength {
			CharFreqs(&cfs, w)
		}
	}

	t := CustomNew(cfs)

	if _, err := file.Seek(0, 0); err != nil {
		return nil, err
	}

	scanner = bufio.NewScanner(file)
	for scanner.Scan() {
		w := scanner.Text()
		if maxLength < 1 || len(w) < maxLength {
			t.AddWord(w)
		}
	}

	return t, nil
}

func gatherCmdOutput(taskName string, cmd *exec.Cmd, eventsChan chan *types.Event) error {
	stdoutReader, err := cmd.StdoutPipe()
	if err != nil {
		return err
	}

	stderrReader, err := cmd.StderrPipe()
	if err != nil {
		return err
	}

	stdoutScanner := bufio.NewScanner(stdoutReader)
	stderrScanner := bufio.NewScanner(stderrReader)
	go func() {
		for stdoutScanner.Scan() {
			eventsChan <- types.NewOutputEvent(taskName, stdoutScanner.Bytes())
		}
	}()

	go func() {
		for stderrScanner.Scan() {
			eventsChan <- types.NewOutputEvent(taskName, stderrScanner.Bytes())
		}
	}()

	return nil
}

func main() {
	// EXAMPLE 1
	reader := StringPump("Test")

	scanner := bufio.NewScanner(reader)
	for scanner.Scan() {
		fmt.Printf("Received: %s \n", scanner.Text())
	}

	// EXAMPLE 2
	reader = Capitalize(StringPump("Test"))

	scanner = bufio.NewScanner(reader)
	for scanner.Scan() {
		fmt.Printf("Received: %s \n", scanner.Text())
	}

	// EXAMPLE 3
	ToLower := WrapStringWordFilter(strings.ToLower)
	reader = ToLower(Capitalize(StringPump("Test THIS bit")))

	scanner = bufio.NewScanner(reader)
	for scanner.Scan() {
		fmt.Printf("Received: %s \n", scanner.Text())
	}

	fmt.Printf("Finished.\n")
}

func (t *task) logCmd(cmd *exec.Cmd) {
	stderr, _ := cmd.StderrPipe()
	stdout, _ := cmd.StdoutPipe()

	go t.logPipe(bufio.NewScanner(stderr))
	go t.logPipe(bufio.NewScanner(stdout))
}

func scannerFromFile(reader io.Reader) (*bufio.Scanner, error) {

	var scanner *bufio.Scanner
	//create a bufio.Reader so we can 'peek' at the first few bytes
	bReader := bufio.NewReader(reader)

	testBytes, err := bReader.Peek(64) //read a few bytes without consuming
	if err != nil {
		return nil, err
	}
	//Detect if the content is gzipped
	contentType := http.DetectContentType(testBytes)

	fmt.Printf("Content Type:%s\n", contentType)

	//If we detect gzip, then make a gzip reader, then wrap it in a scanner
	if strings.Contains(contentType, "x-gzip") {
		gzipReader, err := gzip.NewReader(bReader)
		if err != nil {
			return nil, err
		}

		scanner = bufio.NewScanner(gzipReader)

	} else {
		//Not gzipped, just make a scanner based on the reader
		scanner = bufio.NewScanner(bReader)
	}

	return scanner, nil
}