refactory: read,write use channel

2024-05-24 17:27:52 +08:00
parent c194bec3e3
commit e4d5a1be76
9 changed files with 150 additions and 168 deletions
--- a/internal/interfaces/dumpio.go
+++ b/internal/interfaces/dumpio.go
@ -7,7 +7,7 @@ import (

 type DumpIO interface {
 	ReadData(ctx context.Context, size int, query map[string]any, includeFields []string) (<-chan []*model.ESSource, <-chan error)
-	WriteData(ctx context.Context, docs []*model.ESSource) (int, error)
+	WriteData(ctx context.Context, docsCh <-chan []*model.ESSource) error

 	ReadMapping(context.Context) (map[string]any, error)
 	WriteMapping(context.Context, map[string]any) error