如何使用上下文

大家好，我们又见面了啊~本文《如何使用上下文》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

对于个人项目，我正在生成一个托管 rest api 的外部进程。 我只想在外部进程初始化后将控制权传递回主线程，这可以通过读取 stdout 得知。

因此，我创建了一个 io.writer 实现，一旦满足特定条件，它就会关闭通道。

type channeliowriter struct {
    initializedchannel chan bool
    isinitialized      bool
    buffer             string
}

func newchanneliowriter() *channeliowriter {
    retval := new(channeliowriter)
    retval.initializedchannel = make(chan bool)

    return retval
}

func (writer *channeliowriter) write(data []byte) (int, error) {
    for _, b := range data {

        if b == '\n' {
            writer.buffer = *bytes.newbuffer(make([]byte, 0))
            continue
        }

        writer.buffer.writebyte(b)

        if strings.hasprefix(writer.buffer.string(), "chromedriver was started successfully.") {
            if !writer.isinitialized {
                close(writer.initializedchannel)
            }

            writer.isinitialized = true
        }
    }

    return len(data), nil
}

接下来，我有一个 main 函数，它在一个单独的 goroutine 中生成外部进程。 在应该由 io.writer 实现关闭的通道上执行等待。

func main() {
    writer := newChannelIOWriter()

    go func() {
        cmd := exec.Command("chromedriver", "--port=9009")
        cmd.Stdout = writer
        cmd.Start()
    }()

    <-writer.InitializedChannel

    fmt.Println("Control is passed back to the MAIN thread.")
}

根据对该问题的评论，使用上下文似乎更好。 谁能解释一下如何使用它？

解决方案

我删除了所有不必要的细节并演示了截止日期的上下文如何工作

package main

import (
    "context"
    "fmt"
    "os/exec"
    "time"
)

type example struct {
    cancel context.CancelFunc
}

func (e *example) Write(data []byte) (int, error) {
    defer e.cancel()
    return len(data), nil
}
func main() {
    deadline := time.Now().Add(5 * time.Second)
    ctx, cancel := context.WithDeadline(context.Background(), deadline)
    writer := &example{
        cancel: cancel,
    }

    go func() {
        cmd := exec.Command("ls", ".")
        cmd.Stdout = writer
        _ = cmd.Start()
    }()

    <-ctx.Done()

    fmt.Println("Control is passed back to the MAIN thread.", ctx.Err())
}

终于介绍完啦！小伙伴们，这篇关于《如何使用上下文》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！