в статус number, добавлена проверка на слеш в текущей итерации
This commit is contained in:
parent
201ed716ee
commit
095a8daab2
|
@ -45,7 +45,8 @@ func Unpack(str string) (string, error) {
|
||||||
if unicode.IsDigit(char) {
|
if unicode.IsDigit(char) {
|
||||||
currentState = number
|
currentState = number
|
||||||
numOfRepeat, _ = strconv.Atoi(string(char))
|
numOfRepeat, _ = strconv.Atoi(string(char))
|
||||||
repStr, err := repeatRune(runeArray[i-1], numOfRepeat)
|
n := strconv.Itoa(numOfRepeat)
|
||||||
|
repStr, err := repeatRune(runeArray[i-len(n)], numOfRepeat)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// Если функция repeatRune возвращает ошибку, удаляем последний символ из результата.
|
// Если функция repeatRune возвращает ошибку, удаляем последний символ из результата.
|
||||||
res := result.String()
|
res := result.String()
|
||||||
|
@ -60,7 +61,9 @@ func Unpack(str string) (string, error) {
|
||||||
currentState = start
|
currentState = start
|
||||||
}
|
}
|
||||||
case number:
|
case number:
|
||||||
if unicode.IsDigit(char) {
|
if char == '\\' {
|
||||||
|
currentState = escape
|
||||||
|
} else if unicode.IsDigit(char) {
|
||||||
var n = strconv.Itoa(numOfRepeat) + string(char)
|
var n = strconv.Itoa(numOfRepeat) + string(char)
|
||||||
numOfRepeat, _ = strconv.Atoi(n)
|
numOfRepeat, _ = strconv.Atoi(n)
|
||||||
repStr, err := repeatRune(runeArray[i-len(n)], numOfRepeat)
|
repStr, err := repeatRune(runeArray[i-len(n)], numOfRepeat)
|
||||||
|
@ -71,6 +74,7 @@ func Unpack(str string) (string, error) {
|
||||||
result.WriteString(res[0 : len(res)-1])
|
result.WriteString(res[0 : len(res)-1])
|
||||||
}
|
}
|
||||||
result.WriteString(repStr)
|
result.WriteString(repStr)
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
result.WriteRune(char)
|
result.WriteRune(char)
|
||||||
currentState = start
|
currentState = start
|
||||||
|
|
|
@ -26,6 +26,8 @@ func TestUnpack(t *testing.T) {
|
||||||
{input: `qwe\45`, expected: `qwe44444`},
|
{input: `qwe\45`, expected: `qwe44444`},
|
||||||
{input: `qwe\\5`, expected: `qwe\\\\\`},
|
{input: `qwe\\5`, expected: `qwe\\\\\`},
|
||||||
{input: `qwe\\\3`, expected: `qwe\3`},
|
{input: `qwe\\\3`, expected: `qwe\3`},
|
||||||
|
{input: `qwe2\1`, expected: `qwee1`},
|
||||||
|
{input: `qwe2\\`, expected: `qwee\`},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tc := range tests {
|
for _, tc := range tests {
|
||||||
|
|
Loading…
Reference in New Issue