Go语言 strings包NewReader方法及ReadRune查找字符串

一、函数说明

strings.NewReader创建一个从s读取数据的Reader

type Reader struct {
	s        string //对应的字符串
	i        int64  // 当前读取到的位置
	prevRune int   
}
// 读取器实了 io.Reader, io.ReaderAt, io.Seeker, io.WriterTo,
// io.ByteScanner, 和 io.RuneScanner 通过从字符串中读取
// Reader的零值类似于空字符串的读取器
type Reader struct {
	s        string // 对应的字符串
	i        int64 // 当前读取到的位置
	prevRune int   // prevRune索引; or < 0
}

二、相关方法

2.1 ReadByte

func (r *Reader) ReadByte() (byte, error)

r *Readerr *Reader

2.2 UnreadByte

func (r *Reader) UnreadByte() error

func (r *Reader) ReadByte() (byte, error)

代码示例：

    read := strings.NewReader("abcd")
    c, _ := read.ReadByte()
    fmt.Printf("%c\n", c) // a
    c, _ = read.ReadByte()
    fmt.Printf("%c\n", c) // b

    _ = read.UnreadByte()
    c, _ = read.ReadByte() // b
    fmt.Printf("%c\n", c)

2.3 ReadRune

func (r *Reader) ReadRune() (ch rune, size int, err error)

r *ReaderUTF8size interr errorUTF8Unicoder *Reader

2.4 UnreadRune

func (r *Reader) UnreadRune() error

UnreadRuneUnicodeReadRuneUnreadRuneUnreadByte

代码示例：

    reader := strings.NewReader("你好，世界！")
    s, size, _ := reader.ReadRune()
    fmt.Printf("%c %v \n", s, size) // 你 3

    _ = reader.UnreadRune()
    s, size, _ = reader.ReadRune()
    fmt.Printf("%c %v \n", s, size) // 你 3

三、查询UTF-8字符位置

mb_strpos

package main

import (
    "fmt"
    "strings"
)

func main() {
    fmt.Println(mb_strpos("聚合数据科技公司", "科技")) // 4
    fmt.Println(mb_strpos("聚合数据--科技公司", "科技")) // 6
    fmt.Println(mb_strpos("juhe聚合数据-科技公司", "数据")) // 6
}

func mb_strpos(haystack, needle string) int {
    index := strings.Index(haystack, needle)
    if index == -1 || index == 0 {
        return index
    }
    pos := 0
    total := 0
    reader := strings.NewReader(haystack)
    for {
        _, size, err := reader.ReadRune()
        if err != nil {
            return -1
        }
        total += size
        pos++
        // got it
        if total == index {
            return pos
        }
    }
}

推荐查询 PHP 转 Go 函数的工具： https://php2golang.com