Function getTables

backend/plugin/db/sqlite/sync.go:100–166

getTables gets all tables of a database.

(txn *sql.Tx)

// getTables gets all tables of a database.
func getTables(txn *sql.Tx) ([]*storepb.TableMetadata, error) {
	indexMap, err := getIndices(txn)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to get indices")
	}

	var tableNames []string
	query := `
		SELECT
			name
		FROM sqlite_schema
		WHERE type ='table' AND name NOT LIKE 'sqlite_%'
		ORDER BY name;`
	rows, err := txn.Query(query)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	for rows.Next() {
		var name string
		if err := rows.Scan(&name); err != nil {
			return nil, err
		}
		tableNames = append(tableNames, name)
	}
	if err := rows.Err(); err != nil {
		return nil, util.FormatErrorWithQuery(err, query)
	}

	var tables []*storepb.TableMetadata
	for _, name := range tableNames {
		table := &storepb.TableMetadata{
			Name: name,
		}
		if err := func() error {
			// Get columns: cid, name, type, notnull, dflt_value, pk.
			query := fmt.Sprintf("pragma table_info(%s);", name)
			rows, err := txn.Query(query)
			if err != nil {
				return err
			}
			defer rows.Close()
			for rows.Next() {
				column := &storepb.ColumnMetadata{}
				var notNull, unusedPk bool
				var defaultStr sql.NullString
				if err := rows.Scan(&column.Position, &column.Name, &column.Type, &notNull, &defaultStr, &unusedPk); err != nil {
					return err
				}
				column.Nullable = !notNull
				if defaultStr.Valid {
					// TODO: use correct default type
					column.Default = defaultStr.String
				}

				table.Columns = append(table.Columns, column)
				table.Indexes = indexMap[table.Name]
			}