mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-26 07:52:32 +00:00
ae25ec2e6b
* [NOD-303] Implement get transaction by id for api server * [NOD-303] Make routeParamTxID a constant * [NOD-303] Change database is not current error. * [NOD-303] Add ID to TransactionInput and TransactionOutput models * [NOD-303] Change transactions_outputs table name to transaction_outputs and transactions_inputs to transaction_inputs * [NOD-303] Add json annotations to transaction response types * [NOD-303] Split server package * [NOD-303] Add GetTransactionByHashHandler * [NOD-303] Add comments to exported functions and variables * [NOD-303] Put response types in a separate file * [NOD-303] Rename functions
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/daglabs/btcd/apiserver/config"
|
|
"github.com/golang-migrate/migrate/v4/source"
|
|
"github.com/jinzhu/gorm"
|
|
"os"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
)
|
|
|
|
// DB is the API server database.
|
|
var DB *gorm.DB
|
|
|
|
// Connect connects to the database mentioned in
|
|
// config variable.
|
|
func Connect(cfg *config.Config) error {
|
|
connectionString := buildConnectionString(cfg)
|
|
isCurrent, err := isCurrent(connectionString)
|
|
if err != nil {
|
|
return fmt.Errorf("Error checking whether the database is current: %s", err)
|
|
}
|
|
if !isCurrent {
|
|
return fmt.Errorf("Database is not current. Please migrate" +
|
|
" the database and start again.")
|
|
}
|
|
|
|
DB, err = gorm.Open("mysql", connectionString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func buildConnectionString(cfg *config.Config) string {
|
|
return fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=True",
|
|
cfg.DBUser, cfg.DBPassword, cfg.DBAddress, cfg.DBName)
|
|
}
|
|
|
|
// isCurrent resolves whether the database is on the latest
|
|
// version of the schema.
|
|
func isCurrent(connectionString string) (bool, error) {
|
|
driver, err := source.Open("file://migrations")
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
migrator, err := migrate.NewWithSourceInstance(
|
|
"migrations", driver, "mysql://"+connectionString)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
// Get the current version
|
|
version, isDirty, err := migrator.Version()
|
|
if err == migrate.ErrNilVersion {
|
|
return false, nil
|
|
}
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if isDirty {
|
|
return false, fmt.Errorf("Database is dirty")
|
|
}
|
|
|
|
// The database is current if Next returns ErrNotExist
|
|
_, err = driver.Next(version)
|
|
if pathErr, ok := err.(*os.PathError); ok {
|
|
if pathErr.Err == os.ErrNotExist {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, err
|
|
}
|