我有多個(gè)二進(jìn)制傳入 tcp 流，不斷地隨機(jī)到達(dá)。我想將它們中的每一個(gè)寫(xiě)入單獨(dú)的文件。這些文件應(yīng)該包含從連接的初始打開(kāi)到遠(yuǎn)程客戶端關(guān)閉連接的所有內(nèi)容。我在下面有這個(gè)，它似乎有效，但這是一種有效的方法嗎？//error handling to be added laterfunc main() { listen, _ := net.Listen("tcp4", ":8080") defer listen.Close() for { connection, _ := listen.Accept() defer connection.Close() //"myfile.dat" to be replaced later with variable creating unique names go handleConnection(connection, "myfile.dat") }}func handleConnection(connection net.Conn, myFile string) { outputFile, _ := os.Create(myFile) defer outputFile.Close() fileWriter := bufio.NewWriter(outputFile) for { serverConnReader := bufio.NewReader(connection) io.Copy(fileWriter, serverConnReader) break }}
1 回答
長(zhǎng)風(fēng)秋雁
TA貢獻(xiàn)1757條經(jīng)驗(yàn) 獲得超7個(gè)贊
關(guān)于代碼的一些注釋：
處理錯(cuò)誤。
defer connection.Close()在handleConnection. 否則，連接不會(huì)關(guān)閉，直到程序終止。將兩行代碼
handleConnection移出for循環(huán)，丟棄for循環(huán)。循環(huán)沒(méi)有做任何事情。添加
defer fileWriter.Flush()以刷新寫(xiě)入器中緩沖的任何最后一位數(shù)據(jù)（但請(qǐng)參閱下一點(diǎn)）。要求文件直接從連接中復(fù)制。
這是代碼：
func main() {
listen, err := net.Listen("tcp4", ":8080")
if err != nil {
log.Fatal(err)
}
defer listen.Close()
for {
connection, err := listen.Accept()
if err != nil {
log.Fatal(err)
}
go handleConnection(connection, "myfile.dat")
}
}
func handleConnection(connection net.Conn, myFile string) {
defer connection.Close()
outputFile, err := os.Create(myFile)
if err != nil {
log.Fatal(err)
}
defer outputFile.Close()
_, err = outputFile.ReadFrom(connection)
if err != nil {
log.Fatal(err)
}
}
- 1 回答
- 0 關(guān)注
- 96 瀏覽
添加回答
舉報(bào)
0/150
提交
取消