Method loadTable

datasource/mockcsv/mockcsv.go:135–178

(tableName string)

}

func (m *Source) loadTable(tableName string) error {

	csvRaw, ok := m.raw[tableName]
	if !ok {
		return schema.ErrNotFound
	}

	// The expected format is that the csv data is a single string, with new-lines, etc.
	sr := strings.NewReader(csvRaw)
	csvSource, err := datasource.NewCsvSource(tableName, 0, sr, make(<-chan bool, 1))
	if err != nil {
		u.Warnf("Could not load csv table=%q %v", tableName, err)
		return err
	}
	if csvSource == nil {
		return fmt.Errorf("No csv-source created for %q", tableName)
	}
	ds := membtree.NewStaticData(tableName)
	ds.SetColumns(csvSource.Columns())
	m.tables[tableName] = ds

	// Now we are going to page through the Csv rows and Put into
	// Static Data Source, ie copy into memory btree structure
	for {
		msg := csvSource.Next()
		if msg == nil {
			break
		}
		dm, ok := msg.Body().(*datasource.SqlDriverMessageMap)
		if !ok {
			return fmt.Errorf("Expected *datasource.SqlDriverMessageMap but got %T", msg.Body())
		}

		// We don't know the Key
		ds.Put(nil, nil, dm.Values())
	}

	iter := &Table{StaticDataSource: ds}
	tbl, err := ds.Table(tableName)
	if err != nil {
		return err
	}
	return datasource.IntrospectTable(tbl, iter)
}

// Close csv source.
func (m *Source) Close() error { return nil }