mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-01 11:29:48 +08:00
59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package storage
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zilliztech/milvus-distributed/internal/proto/schemapb"
|
|
"github.com/zilliztech/milvus-distributed/internal/util/tsoutil"
|
|
)
|
|
|
|
func TestPrintBinlogFilesInt64(t *testing.T) {
|
|
w, err := NewInsertBinlogWriter(schemapb.DataType_INT64, 10, 20, 30, 40)
|
|
assert.Nil(t, err)
|
|
|
|
curTS := time.Now().UnixNano() / int64(time.Millisecond)
|
|
|
|
e1, err := w.NextInsertEventWriter()
|
|
assert.Nil(t, err)
|
|
err = e1.AddDataToPayload([]int64{1, 2, 3})
|
|
assert.Nil(t, err)
|
|
err = e1.AddDataToPayload([]int32{4, 5, 6})
|
|
assert.NotNil(t, err)
|
|
err = e1.AddDataToPayload([]int64{4, 5, 6})
|
|
assert.Nil(t, err)
|
|
e1.SetStartTimestamp(tsoutil.ComposeTS(curTS+10*60*1000, 0))
|
|
e1.SetEndTimestamp(tsoutil.ComposeTS(curTS+20*60*1000, 0))
|
|
|
|
e2, err := w.NextInsertEventWriter()
|
|
assert.Nil(t, err)
|
|
err = e2.AddDataToPayload([]int64{7, 8, 9})
|
|
assert.Nil(t, err)
|
|
err = e2.AddDataToPayload([]bool{true, false, true})
|
|
assert.NotNil(t, err)
|
|
err = e2.AddDataToPayload([]int64{10, 11, 12})
|
|
assert.Nil(t, err)
|
|
e2.SetStartTimestamp(tsoutil.ComposeTS(curTS+30*60*1000, 0))
|
|
e2.SetEndTimestamp(tsoutil.ComposeTS(curTS+40*60*1000, 0))
|
|
|
|
w.SetStartTimeStamp(tsoutil.ComposeTS(curTS, 0))
|
|
w.SetEndTimeStamp(tsoutil.ComposeTS(curTS+3600*1000, 0))
|
|
|
|
_, err = w.GetBuffer()
|
|
assert.NotNil(t, err)
|
|
err = w.Close()
|
|
assert.Nil(t, err)
|
|
buf, err := w.GetBuffer()
|
|
assert.Nil(t, err)
|
|
|
|
fd, err := os.Create("/tmp/binlog_int64.db")
|
|
assert.Nil(t, err)
|
|
num, err := fd.Write(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, num, len(buf))
|
|
err = fd.Close()
|
|
assert.Nil(t, err)
|
|
}
|