fix case of shorter length of inputString (#506)

当读取的inputString的长度小于5时,string([]byte(string))会出现bug
This commit is contained in:
blurty
2018-06-12 21:05:58 +08:00
committed by 无闻
parent 664624de09
commit ea414dbb85

View File

@@ -15,6 +15,7 @@ func main() {
defer outputFile.Close() defer outputFile.Close()
inputReader := bufio.NewReader(inputFile) inputReader := bufio.NewReader(inputFile)
outputWriter := bufio.NewWriter(outputFile) outputWriter := bufio.NewWriter(outputFile)
var outputString string
for { for {
// inputString, readerError := inputReader.ReadString('\n') // inputString, readerError := inputReader.ReadString('\n')
inputString, _, readerError := inputReader.ReadLine() inputString, _, readerError := inputReader.ReadLine()
@@ -23,7 +24,13 @@ func main() {
break break
} }
//fmt.Printf("The input was: --%s--", inputString) //fmt.Printf("The input was: --%s--", inputString)
outputString := string([]byte(inputString)[2:5]) + "\r\n" if len(inputString) < 3 {
outputString = "\r\n"
} else if len(inputString) < 5 {
outputString = string([]byte(inputString)[2:len(inputString)]) + "\r\n"
} else {
outputString = string([]byte(inputString)[2:5]) + "\r\n"
}
//fmt.Printf("The output was: --%s--", outputString) //fmt.Printf("The output was: --%s--", outputString)
_, err := outputWriter.WriteString(outputString) _, err := outputWriter.WriteString(outputString)
//fmt.Printf("Number of bytes written %d\n", n) //fmt.Printf("Number of bytes written %d\n", n)