39 lines
679 B
Go
39 lines
679 B
Go
|
package migrations
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
type columnInfo struct {
|
||
|
|
CID int
|
||
|
|
Name string
|
||
|
|
Type string
|
||
|
|
NotNull int
|
||
|
|
DefaultVal sql.NullString
|
||
|
|
PK int
|
||
|
|
}
|
||
|
|
|
||
|
|
func hasColumn(tx *sql.Tx, table string, column string) (bool, error) {
|
||
|
|
rows, err := tx.Query(fmt.Sprintf("PRAGMA table_info(%s)", table))
|
||
|
|
if err != nil {
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
defer rows.Close()
|
||
|
|
|
||
|
|
colExists := false
|
||
|
|
for rows.Next() {
|
||
|
|
var col columnInfo
|
||
|
|
if err := rows.Scan(&col.CID, &col.Name, &col.Type, &col.NotNull, &col.DefaultVal, &col.PK); err != nil {
|
||
|
|
return false, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if col.Name == column {
|
||
|
|
colExists = true
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return colExists, nil
|
||
|
|
}
|