Golang bufio WriteByte не пишет - PullRequest
0 голосов
/ 27 апреля 2020

У меня небольшая проблема. Я пытаюсь, чтобы программа читала из файла и записывала ту же самую вещь в другой файл. Я заставил это работать, но я полагал, что это должно go быстрее, если я немного изменил это, используя goroutines. Я разделил чтение и запись на разные функции и попытался заставить их общаться. Кажется, они нормально общаются, но на самом деле функция записи ничего не записывает в файл назначения. Дело в том, что когда я выбираю гораздо больший файл, чем простой короткий текстовый файл, он записывает, но не все, пропускает что-то. Может кто-нибудь сказать мне точно, как это может go неправильно? Мой код здесь:

func main() {
    // Creating channels
    messages := make(chan byte)
    status := make(chan bool)
    finished := make(chan bool)
    fmt.Printf("Reporting from main thread. Created the channels, go routines will follow.\n")
    go readFileRoutine("file", messages, status, finished)
    fmt.Printf("Reporting from main thread, readFileRoutine has been let go.\n")
    go writeFileRoutine("destination", messages, status)
    fmt.Printf("Reporting from main thread, writeFileRoutine has been let go. Waiting on status message\n")
    <-finished
    close(status)
    fmt.Printf("Success!\n")
}

func readFileRoutine(filename string, messages chan byte, status chan bool, finished chan bool) {
    file, err := os.Open(filename)
    check(err)
    buff := bufio.NewReader(file)
    fmt.Printf("Reporting from readFileRoutine, initialised. Will start reading now!\n")
    for {
        byti, err := buff.ReadByte()
        fmt.Printf("Read character %c\n", byti)
        if err == io.EOF { //If EOF break for loop and finish off
            readFinished = true
            break
        }
        <-status         // Waiting for status message from writeFileRoutine
        messages <- byti //Putting next message on channel, then buffering next character

    }
    fmt.Printf("Reporting from readFileRoutine, finished reading and putting messages on channel, will now wait on writer to finish\n")
    //fmt.Printf("Reporting from readFileRoutine, finished with reading. Closing file and messages channel now\n")
    file.Close()
    <-status
    fmt.Printf("Reporting from readFileRoutine, moving on to closing messages channel. Received status from writer correctly!\n")
    close(messages)
    <-status
    fmt.Printf("Reporting from readFileRoutine, closed messages channel, will report status on the finished channel.\n")
    finished <- true
}

func writeFileRoutine(filename string, messages chan byte, status chan bool) {
    file, err := os.Create(filename) // Creates a new file
    check(err)
    buf := bufio.NewWriter(file) //New file writer
    fmt.Printf("Reporting from writeFileRoutine, initialised. Will start writing now!\n")
    status <- true
    for readFinished == false {
        bit := <-messages // Wait for receiving
        fmt.Printf("Received message %c\n", bit)
        buf.WriteByte(bit)
        fmt.Printf("Written byte %c\n", bit)
        status <- true //Sending status to readFileRoutine, done writing and waiting for next character
    }
    fmt.Printf("Reporting from writeFileRoutine, read all messages from reader and put them in a file\n")
    file.Close()
    fmt.Printf("Reporting from writeFileRoutine, closed file, will put true on status channel and exit\n")
    status <- true
}

1 Ответ

0 голосов
/ 27 апреля 2020

Как отметил Мартин в комментариях, я просто забыл функцию Writer.Flu sh (). Я поставил buf.Flu sh () сразу после buf.WriteByte (bit), и это исправило это.

...