Skip to content

Commit 05e1edd

Browse files
authored
feat: Add WriteInserts.GetRecords() (#1053)
1 parent ef222df commit 05e1edd

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

message/write_message.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,14 @@ func (m WriteInserts) Exists(tableName string) bool {
8686
})
8787
}
8888

89+
func (m WriteInserts) GetRecords() []arrow.Record {
90+
res := make([]arrow.Record, len(m))
91+
for i := range m {
92+
res[i] = m[i].Record
93+
}
94+
return res
95+
}
96+
8997
func (m WriteInserts) GetRecordsForTable(table *schema.Table) []arrow.Record {
9098
res := make([]arrow.Record, 0, len(m))
9199
for _, insert := range m {

0 commit comments

Comments
 (0)