clickhouse_send_external_data.go 971 B

1234567891011121314151617181920212223242526272829303132333435
  1. package clickhouse
  2. import "github.com/ClickHouse/clickhouse-go/lib/data"
  3. func (ch *clickhouse) sendExternalTables(externalTables []ExternalTable) error {
  4. ch.logf("[send external tables] count %d", len(externalTables))
  5. if externalTables == nil || len(externalTables) == 0 {
  6. return nil
  7. }
  8. block := &data.Block{}
  9. sentTables := make(map[string]bool, 0)
  10. for _, externalTable := range externalTables {
  11. if _, ok := sentTables[externalTable.Name]; ok {
  12. continue
  13. }
  14. ch.logf("[send external table] name %s", externalTable.Name)
  15. sentTables[externalTable.Name] = true
  16. block.Columns = externalTable.Columns
  17. block.NumColumns = uint64(len(externalTable.Columns))
  18. for _, row := range externalTable.Values {
  19. err := block.AppendRow(row)
  20. if err != nil {
  21. return err
  22. }
  23. }
  24. if err := ch.writeBlock(block, externalTable.Name); err != nil {
  25. return err
  26. }
  27. if err := ch.encoder.Flush(); err != nil {
  28. return err
  29. }
  30. block.Reset()
  31. }
  32. return nil
  33. }