cerc-io/go-ethereum
18
0
Fork 0
Code Issues 9 Pull Requests Packages Projects Releases 66 Wiki Activity

Include block number in csv filename

Browse Source
This commit is contained in:
Elizabeth Engelman 2019-01-02 16:12:50 -06:00
parent 74bcf49aa4
commit ecd34978d3
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
statediff/publisher/csv.go
View File

@ -22,16 +22,17 @@ var (
updatedAccountAction = "updated" updatedAccountAction = "updated"
) )
func createCSVFilePath(path string) string { func createCSVFilePath(path, blockNumber string) string {
now := time.Now() now := time.Now()
timeStamp := now.Format(timeStampFormat) timeStamp := now.Format(timeStampFormat)
filePath := filepath.Join(path, timeStamp) suffix := timeStamp + "-" + blockNumber
filePath := filepath.Join(path, suffix)
filePath = filePath + ".csv" filePath = filePath + ".csv"
return filePath return filePath
} }
func (p *publisher) publishStateDiffToCSV(sd builder.StateDiff) (string, error) { func (p *publisher) publishStateDiffToCSV(sd builder.StateDiff) (string, error) {
filePath := createCSVFilePath(p.Config.Path) filePath := createCSVFilePath(p.Config.Path, strconv.FormatInt(sd.BlockNumber, 10))
file, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) file, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil { if err != nil {

4
statediff/publisher/publisher_test.go
View File

@ -116,6 +116,10 @@ func testFileName(t *testing.T) {
if !strings.HasPrefix(fileName, dir) { if !strings.HasPrefix(fileName, dir) {
t.Errorf(testhelpers.TestFailureFormatString, t.Name(), dir, fileName) t.Errorf(testhelpers.TestFailureFormatString, t.Name(), dir, fileName)
} }
blockNumberWithFileExt := strconv.FormatInt(testhelpers.BlockNumber, 10) + ".csv"
if !strings.HasSuffix(fileName, blockNumberWithFileExt) {
t.Errorf(testhelpers.TestFailureFormatString, t.Name(), blockNumberWithFileExt, fileName)
}
} }
func testColumnHeaders(t *testing.T) { func testColumnHeaders(t *testing.T) {