ch7/ch7-02 #150

2022-12-14T10:23:36Z

giscus[bot]
bot Dec 14, 2022

ch7/ch7-02

中文版

https://golang-china.github.io/gopl-zh/ch7/ch7-02.html

dugulingping · 2022-12-14T10:23:39Z

dugulingping
Dec 14, 2022 — with giscus

练习7.4 将第5章的某一些功能重新了一下

package main

import (
	"fmt"
	"golang.org/x/net/html"
	"io"
	"net/http"
	"strings"
)

type Html interface {
	io.ReadCloser
	Getdoc(url string) (err error)
	ForeachNode(n *html.Node, pre, post func(n *html.Node))
	SearchNode(tag string) []*html.Node
	PrintAll()
}

// 不对外暴露，一切操作均通过方法实现
// 除了Html接口中定义的方法, 还可以自己拓展方法, 例如Show方法
type htmlParser struct {
	Html
	str  string
	node *html.Node
	resp *http.Response
}

// 自定义读入
func (doc *htmlParser) Read(p []byte) (n int, err error) {
	if len(p) == 0 {
		return 0, fmt.Errorf("error, read data is empty")
	}
	n = copy(p, doc.str)
	if doc.node != nil {
		return 0, fmt.Errorf("error, node is not empty")
	}
	doc.node, err = html.Parse(strings.NewReader(doc.str))
	return n, err
}

// Clear 清空
func (doc *htmlParser) Clear() {
	doc.node = nil
	doc.resp = nil
}

// Close 关闭连接
func (doc *htmlParser) Close() error {
	return doc.resp.Body.Close()
}

// Getdoc 通过url获取网页内容
func (doc *htmlParser) Getdoc(url string) (err error) {
	doc.resp, err = http.Get(url)
	defer doc.Close() // 关闭连接
	if doc.resp.StatusCode != http.StatusOK {
		err = fmt.Errorf("http get error: %s", doc.resp.Status)
		return err
	}
	doc.node, err = html.Parse(doc.resp.Body)
	if err != nil {
		return fmt.Errorf("html parse error: %s", err)
	}
	return nil
}
// ForEachNode 遍历节点
func (doc *htmlParser) ForEachNode(n *html.Node, pre, post func(n *html.Node)) {
	if n == nil {
		return
	}
	if pre != nil {
		pre(n)
	}
	doc.ForEachNode(n.FirstChild, pre, post)
	doc.ForEachNode(n.NextSibling, pre, post)
	if post != nil {
		post(n)
	}
}
// SearchNode 搜索节点
func (doc *htmlParser) SearchNode(tag string) []*html.Node {
	var nodes []*html.Node
	doc.ForEachNode(doc.node, func(n *html.Node) {
		if n.Type == html.ElementNode && n.Data == tag {
			nodes = append(nodes, n)
		}
	}, nil)
	return nodes
}
// PrintAll 打印所有节点
func (doc *htmlParser) PrintAll() {
	doc.ForEachNode(doc.node, func(n *html.Node) {
		if n.Type == html.ElementNode {
			fmt.Printf("%s\n", n.Data)
		}
	}, nil)
}

// Show 是没有定义在接口中的方法，所以只能通过结构体实例调用
func (doc *htmlParser) Show() {
	fmt.Println(doc.resp)
	fmt.Println()
	fmt.Println(doc.node)
}
func main() {
	var doc htmlParser
	doc.Getdoc("https://go.dev/")
	doc.Show()
	// 如果使用接口类型，则不能使用Show方法
	//var doc Html
	//doc.Getdoc("https://go.dev/")
	// 因为接口类型中没有定义Show方法，所以不能调用
	//doc.Show() // error， Show undefined (type Html has no field or method Show)
}

0 replies

xianren68 · 2023-07-21T07:20:33Z

xianren68
Jul 21, 2023 — with giscus

7.4

package main

import (
	"fmt"
	"io"
	"os"

	"golang.org/x/net/html"
)

// 定义一个html结构体
type HtmlReader struct {
	html []byte
	i    int
}

// 跟着strings.NewReader写的
func (h *HtmlReader) Read(p []byte) (n int, err error) {
	// 存储的索引大于byte数组，表示都读完了
	if h.i >= len(h.html) {
		return 0, io.EOF
	}
	// 复制p长度/剩余长度的字节到p
	n = copy(p, h.html[h.i:])
	// 变化索引
	h.i += n
	return
}

// NewReader，将html转化为reader,以供解析器解析
func NewReader(html string) *HtmlReader {
	return &HtmlReader{html: []byte(html), i: 0}
}

func main() {

	doc, err := html.Parse(NewReader(`
	<!DOCTYPE html>
	<html lang="en">
		<head>
	    <meta charset="UTF-8">
	    <meta name="viewport" content="width=device-width, initial-scale=1.0">
	    <title>Document</title>
	</head>
	<body>
	   hahaha
	</body>
	</html>
		`))
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	fmt.Printf("%#v", doc)
}

0 replies

xianren68 · 2023-07-21T08:13:19Z

xianren68
Jul 21, 2023 — with giscus

7.5

package main

import (
	"bufio"
	"fmt"
	"io"
	"strings"
)

type reader struct {
	i    int
	read io.Reader
	n    int
}

func (r *reader) Read(p []byte) (n int, err error) {
	// 到达n,直接结束
	if r.i >= r.n {
		return 0, io.EOF
	}
	// 剩余的长度不够p
	if r.n-r.i < len(p) {
		n, _ = r.read.Read(p[:r.n-r.i])
		err = io.EOF
		return

	}
	// 通过嵌套读取另一个流
	n, err = r.read.Read(p)
	// 另一个流读到尾部了
	if err != nil {
		return
	}
	// 修改索引
	r.i += n
	return
}
func LimitReader(r io.Reader, n int) io.Reader {
	return &reader{i: 0, read: r, n: n}
}

func main() {
	// 创建一个读入流
	r := strings.NewReader("1234567890")
	r1 := LimitReader(r, 4)
	s := bufio.NewScanner(r1)
	s.Split(bufio.ScanBytes)
	for s.Scan() {
		fmt.Println(s.Text())
	}
}

0 replies

Apahno · 2024-11-07T05:55:09Z

Apahno
Nov 7, 2024 — with giscus

翻译看晕了，能好好表述吗，好绕

0 replies

hongjiala · 2025-04-11T08:36:52Z

hongjiala
Apr 11, 2025 — with giscus

// 练习 7.4： strings.NewReader函数通过读取一个string参数返回一个满足io.Reader接口类型的值（和其它值）。
// 实现一个简单版本的NewReader，用它来构造一个接收字符串输入的HTML解析器（§5.2）
package main

import (
	"fmt"
	"io"
	"os"

	"golang.org/x/net/html"
)

// 定义一个html结构体
type HtmlReader struct {
	html []byte
	i    int
}

// 用来检验是否读完了
func (h *HtmlReader) Read(p []byte) (n int, err error) {
	// 存储的索引大于byte数组，表示都读完了
	if h.i >= len(h.html) {
		return 0, io.EOF
	}
	// 复制p长度/剩余长度的字节到p
	n = copy(p, h.html[h.i:])
	// 变化索引
	h.i += n
	return
}

// NewReader函数返回一个满足io.Reader接口类型的值
func NewReader(html string) io.Reader {
	// 返回一个HtmlReader结构体的指针，传入html字符串转换成byte数组
	return &HtmlReader{[]byte(html), 0}
}

func main() {

	doc, err := html.Parse(NewReader(`
	<!DOCTYPE html>
	<html lang="en">
		<head>
	    <meta charset="UTF-8">
	    <meta name="viewport" content="width=device-width, initial-scale=1.0">
	    <title>Document</title>
	</head>
	<body>
	   hahaha
	</body>
	</html>
		`))
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	fmt.Printf("%#v", doc)
}

0 replies

hongjiala · 2025-04-11T09:04:09Z

hongjiala
Apr 11, 2025 — with giscus

// 练习 7.5： io包里面的LimitReader函数接收一个io.Reader接口类型的r和字节数n，
// 并且返回另一个从r中读取字节但是当读完n个字节后就表示读到文件结束的Reader。实现这个LimitReader函数：

package main

import (
	"fmt"
	"io"
)

// LimitReader 是一个包装类型，它限制了从原始 Reader 中最多只能读取 n 字节数据
type LimitReader struct {
	r io.Reader // 原始的 Reader
	n int64     // 读取的最大字节数
}

// LimitReader 函数返回一个新的 Reader，限制从原始 Reader 中最多读取 n 字节数据
func LimitRead(r io.Reader, n int64) io.Reader {
	return &LimitReader{r: r, n: n}
}

// 实现 io.Reader 接口的 Read 方法
func (lr *LimitReader) Read(p []byte) (n int, err error) {
	// 如果 n 为零，则返回 EOF，表示读取结束
	if lr.n <= 0 {
		return 0, io.EOF
	}

	// 如果 p 的长度大于剩余可读字节数，限制读取字节数为 lr.n
	if int64(len(p)) > lr.n {
		p = p[:lr.n]
	}

	// 从底层 Reader 中读取数据
	n, err = lr.r.Read(p)

	// 更新剩余可读取字节数
	lr.n -= int64(n)

	// 如果已读取字节数达到限制，则返回 EOF
	if lr.n <= 0 {
		err = io.EOF
	}

	return n, err
}

// 示例：使用 LimitReader 限制读取字节
func main() {
	// 模拟一个数据源
	data := "Hello, this is a test string for LimitReader function."

	// 创建一个字符串 Reader
	r := io.Reader(&CustomReader{data: data})

	// 创建一个 LimitReader，最多读取 20 个字节
	lr := LimitRead(r, 20)

	// 创建一个缓冲区来存储读取的内容
	buf := make([]byte, 10)

	// 读取数据
	for {
		n, err := lr.Read(buf)
		fmt.Printf("Read %d bytes: %s\n", n, buf[:n])
		if err == io.EOF {
			break
		}
		if err != nil {
			fmt.Println("Error:", err)
			break
		}
	}
}

// CustomReader 用于模拟一个简单的 io.Reader
type CustomReader struct {
	data string
}

func (r *CustomReader) Read(p []byte) (n int, err error) {
	if len(r.data) == 0 {
		return 0, io.EOF
	}

	n = copy(p, r.data)
	r.data = r.data[n:]

	return n, nil
}

0 replies

w2807 · 2025-04-30T09:32:26Z

w2807
Apr 30, 2025 — with giscus

6.5

type limitReader struct {
	r io.Reader
	n int64
}

func (r *limitReader) Read(p []byte) (int, error) {
	if r.n <= 0 {
		return 0, io.EOF
	}
	if int64(len(p)) > r.n {
		p = p[:r.n]
	}
	n, err := r.r.Read(p)
	r.n -= int64(n)
	return n, err
}

func LimitReader(r io.Reader, n int64) io.Reader {
	return &limitReader{r, n}
}

0 replies

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

ch7/ch7-02 #150

Uh oh!

{{title}}

Uh oh!

Replies: 7 comments

Uh oh!

{{title}}

Uh oh!

Uh oh!

{{title}}

Uh oh!

Uh oh!

{{editor}}'s edit

{{editor}}'s edit

Uh oh!

Uh oh!

{{title}}

Uh oh!

Uh oh!

{{title}}

Uh oh!

Uh oh!

{{title}}

Uh oh!

Uh oh!

{{title}}

Uh oh!

Uh oh!

{{title}}

Uh oh!

Select a reply

Uh oh!

ch7/ch7-02 #150

Uh oh!

giscus[bot] bot Dec 14, 2022

ch7/ch7-02

Replies: 7 comments

Uh oh!

dugulingping Dec 14, 2022 — with giscus

练习7.4 将第5章的某一些功能重新了一下

Uh oh!

Uh oh!

xianren68 Jul 21, 2023 — with giscus

7.4

Uh oh!

xianren68 Jul 21, 2023 — with giscus

7.5

Uh oh!

Apahno Nov 7, 2024 — with giscus

Uh oh!

hongjiala Apr 11, 2025 — with giscus

Uh oh!

hongjiala Apr 11, 2025 — with giscus

Uh oh!

w2807 Apr 30, 2025 — with giscus

giscus[bot]
bot Dec 14, 2022

dugulingping
Dec 14, 2022 — with giscus

xianren68
Jul 21, 2023 — with giscus

xianren68
Jul 21, 2023 — with giscus

Apahno
Nov 7, 2024 — with giscus

hongjiala
Apr 11, 2025 — with giscus

hongjiala
Apr 11, 2025 — with giscus

w2807
Apr 30, 2025 — with giscus