Created
January 1, 2017 16:46
-
-
Save ice1000/781c58a4a81058481a1c5d790d3c55be to your computer and use it in GitHub Desktop.
Java-like StringBuffer for Golang
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package sb | |
import ( | |
"fmt" | |
"strings" | |
) | |
const maxLength = 150 | |
/// a java style string buffer | |
type stringBuffer struct { | |
data []string | |
index int | |
indent int | |
} | |
func NewStringBuffer() *stringBuffer { | |
ret := new(stringBuffer) | |
ret.index = 0 | |
ret.indent = 0 | |
ret.data = make([]string, maxLength) | |
return ret | |
} | |
/// append a string to the tail of this buffer | |
func (sb *stringBuffer) Append(str string) *stringBuffer { | |
sb.data[sb.index] = strings.Repeat("\t", sb.indent) + str | |
sb.index++ | |
if sb.index >= maxLength { | |
sb.slice() | |
} | |
return sb | |
} | |
func (sb *stringBuffer) AppendFormat(str string, args ...interface{}) *stringBuffer { | |
return sb.Append(fmt.Sprintf(str, args...)) | |
} | |
func (sb *stringBuffer) AppendLine(str string) *stringBuffer { | |
return sb.Append(str + "\n") | |
} | |
func (sb *stringBuffer) AppendLineIndent(str string) *stringBuffer { | |
sb.AppendLine(str) | |
sb.indent++ | |
return sb | |
} | |
func (sb *stringBuffer) AppendLineClose(str string) *stringBuffer { | |
sb.indent-- | |
return sb.AppendLine(str) | |
} | |
func (sb *stringBuffer) AppendClose(str string) *stringBuffer { | |
sb.indent-- | |
return sb.Append(str) | |
} | |
func (sb *stringBuffer) slice() *stringBuffer { | |
sb.data[0] = sb.ToString() | |
sb.index = 1 | |
return sb | |
} | |
/// append a line as comment. | |
func (sb *stringBuffer) AppendComment(str string) *stringBuffer { | |
return sb.AppendLine("/// " + str) | |
} | |
func (sb *stringBuffer) ToString() string { | |
return strings.Join(sb.data, "") | |
} | |
/// clear elements. | |
func (sb *stringBuffer) Clear() { | |
sb.indent = 0 | |
} |
Missing the setlength method for easy truncating, isn't it?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi, I like it, I just want to suggest a fix to the code: in the ToString method, replace with this:
func (sb *stringBuffer) ToString() string {
data := sb.data[0:sb.index]
return strings.Join(data, "")
}
otherwise, you collect all lines even with strings that are left after slice() was ran