试图在Go中测试来自goroutines的写文件

时间:2014-10-03 00:58:42

标签: go

好吧,我的部分代码在没有方法的情况下工作,我正在尝试测试 将文本附加到文件并从goroutines中读取,但是我在这里试图解决 写下来。

有什么问题?该文件已创建,但我无法在其中附加文字,也许是显而易见的事情,但似乎我是盲目的,也许我没有理解某些语言概念...

package main

import (
    "bufio"
    "fmt"
    "os"
    "sync"
    "time"
)

var w sync.WaitGroup

type Buffer struct {
    F *os.File
}

func (buff *Buffer) Open(pathName string) (err error) {
    buff.F, err = os.OpenFile(pathName, os.O_APPEND|os.O_CREATE, 0666)
    if err != nil {
        return
    }
    fmt.Println("Open() ok")
    return nil
}
func (buff *Buffer) Close() (err error) {
    err = buff.F.Close()
    if err != nil {
        return
    }
    fmt.Println("Close() ok")
    return nil
}
func (buff *Buffer) Push(data string) (err error) {
    w := bufio.NewWriter(buff.F)
    _, err = fmt.Fprintf(w, "data=%s", data)
    if err != nil {
        return
    }
    w.Flush()
    fmt.Println("Push() ok")
    return nil
}
func checkErr(err error) {
    if err != nil {
        fmt.Println(err.Error())
    }
}
func worker() {
    var err error
    buffer := new(Buffer)
    err = buffer.Open("test")
    checkErr(err)
    err = buffer.Push("data\n")
    checkErr(err)
    time.Sleep(5 * time.Second)
    err = buffer.Close()
    checkErr(err)
    w.Done()
}
func main() {
    w.Add(2)
    go worker()
    go worker()
    w.Wait()
}

谢谢

1 个答案:

答案 0 :(得分:3)

像这样打开文件:

buff.F, err = os.OpenFile(pathName, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)

写入标志需要写入标志。

您错过了写入错误,因为忽略了bufio Flush的返回。将推送更改为:

func (buff *Buffer) Push(data string) (err error) {
    w := bufio.NewWriter(buff.F)
    _, err = fmt.Fprintf(w, "data=%s", data)
    if err != nil {
        return
    }
    err = w.Flush()
    if err != nil {
        return err
    }
    fmt.Println("Push() ok")
    return nil
}

要在不与其他推送混合的情况下干净地附加数据,必须通过对文件Write方法的单次调用来写入数据。使用bytes.Buffer而不是bufio.Writer来确保单个调用文件Write方法:

func (buff *Buffer) Push(data string) (err error) {
    var b bytes.Buffer
    _, err = fmt.Fprintf(&b, "data=%s", data)
    if err != nil {
        return
    }
    _, err := buff.F.Write(b.Bytes())
    if err != nil {
        return err
    }
    fmt.Println("Push() ok")
    return nil
}