Method SetModel

viewer/modelutil.go:122–163

SetModel creates a model to be used by bubbletea using some golang wizardry

(c *sql.Rows, db *sql.DB)

// SetModel creates a model to be used by bubbletea using some golang wizardry
func (m *TuiModel) SetModel(c *sql.Rows, db *sql.DB) error {
	var err error

	indexMap := 0

	// gets all the schema names of the database
	tableNamesQuery := m.Table().Database.GetTableNamesQuery()
	rows, err := db.Query(tableNamesQuery)
	if err != nil {
		return err
	}

	defer rows.Close()

	// for each schema
	for rows.Next() {
		var schemaName string
		rows.Scan(&schemaName)

		// couldn't get prepared statements working and gave up because it was very simple
		var statement strings.Builder
		statement.WriteString("select * from ")
		statement.WriteString(schemaName)
		getAll := statement.String()

		if c != nil {
			c.Close()
			c = nil
		}
		c, err = db.Query(getAll)
		if err != nil {
			panic(err)
		}

		m.PopulateDataForResult(c, &indexMap, schemaName)
	}

	// set the first table to be initial view
	m.UI.CurrentTable = 1

	return nil
}

func (m *TuiModel) PopulateDataForResult(c *sql.Rows, indexMap *int, schemaName string) {
	columnNames, _ := c.Columns()