Golang: Schreiben gleichzeitig mit Kanälen

Question

Ich schrieb ein kurzes Skript, um Dateien gleichzeitig zu schreiben. Eine Goroutine soll Strings in eine Datei schreiben, während die anderen die Nachrichten über einen Kanal dorthin senden sollen. Aus irgendeinem wirklich seltsamen Grund wird die Datei erstellt, aber über den Kanal wird keine Nachricht hinzugefügt.

package main 

import (
    "fmt" 
    "os" 
    "sync" 
) 

var wg sync.WaitGroup 
var output = make(chan string) 

func concurrent(n uint64) { 
    output <- fmt.Sprint(n) 
    defer wg.Done() 
} 

func printOutput() { 
    f, err := os.OpenFile("output.txt", os.O_CREATE|os.O_RDWR|os.O_APPEND, 0666); 
    if err != nil { 
      panic(err) 
    } 
    defer f.Close() 

    for msg := range output { 
      f.WriteString(msg+"\n") 
    } 
} 

func main() { 
    wg.Add(2) 
    go concurrent(1) 
    go concurrent(2) 
    wg.Wait() 
    close(output) 
    printOutput() 
} 

Die printOutput() goroutine vollständig ausgeführt, wenn ich etwas zu schreiben versucht, nach dem for-Schleife würde es tatsächlich in die Datei. Das führt mich zu der Annahme, dass die Bereichsausgabe Null sein könnte.

Answer 1

Einer der Gründe, warum Sie eine Null bekommen output ist, weil Kanäle sind blocking für beide senden/empfangen.

Je nach Ihrem Fluss erreicht der Code-Ausschnitt unten niemals wg.Done(), da der Sendekanal ein Empfangsende erwartet, um die Daten herauszuziehen. Dies ist ein typisches Deadlock-Beispiel.

func concurrent(n uint64) { 
    output <- fmt.Sprint(n) // go routine is blocked until data in channel is fetched. 
    defer wg.Done() 
} 

Lassen Sie uns die Haupt func untersuchen:

func main() { 
    wg.Add(2) 
    go concurrent(1) 
    go concurrent(2) 
    wg.Wait()  // the main thread will be waiting indefinitely here. 
    close(output) 
    printOutput() 
} 

---

Answer 2

Sie müssen etwas vom Ausgangskanal nehmen, da es blockiert, bis etwas entfernt, was Sie darauf setzen.

Nicht der einzige/beste Weg, es zu tun, aber: Ich bewegte printOutput() über die anderen Funktionen und führen Sie es als eine Go-Routine und es verhindert den Deadlock.

package main 

import (
    "fmt" 
    "os" 
    "sync" 
) 

var wg sync.WaitGroup 
var output = make(chan string) 

func concurrent(n uint64) { 
    output <- fmt.Sprint(n) 
    defer wg.Done() 
} 

func printOutput() { 
    f, err := os.OpenFile("output.txt", os.O_CREATE|os.O_RDWR|os.O_APPEND, 0666) 
    if err != nil { 
     panic(err) 
    } 
    defer f.Close() 

    for msg := range output { 
     f.WriteString(msg + "\n") 
    } 
} 

func main() { 
    go printOutput() 
    wg.Add(2) 
    go concurrent(1) 
    go concurrent(2) 
    wg.Wait() 
    close(output) 
}