small bug fix

This commit is contained in:
Remco
2014-09-23 11:31:42 +02:00
parent ed99e80d4f
commit 3d385f0b1e
2 changed files with 6 additions and 7 deletions

View File

@@ -28,6 +28,7 @@ package clamd
import (
"errors"
"fmt"
"log"
"io"
"strings"
)
@@ -243,15 +244,15 @@ func (c *Clamd) ScanStream(r io.Reader) (chan string, error) {
buf := make([]byte, CHUNK_SIZE)
nr, err := r.Read(buf)
if nr > 0 {
log.Printf("Error %v, %v, %v", buf[0:nr], nr, err)
conn.sendChunk(buf[0:nr])
}
if err != nil {
break
}
if nr == 0 {
break
}
conn.sendChunk(buf[:nr])
}
err = conn.sendEOF()

View File

@@ -86,9 +86,7 @@ func (c *CLAMDConn) readResponse() (chan string, sync.WaitGroup, error) {
// var dataArrays []string
go func() {
fmt.Println("Start")
defer func() {
fmt.Println("Cleaning up")
close(ch)
wg.Done()
}()