uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-11 18:16:24 +05:00
Code Issues Packages Projects Releases Wiki Activity

First operational version of *cost_details migrator

Browse Source
This commit is contained in:
DanB
2016-12-08 15:45:18 +01:00
parent b2f1b07c8b
commit f789e181ba
3 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/cgr-loader/cgr-loader.go
View File

@@ -215,7 +215,7 @@ func main() {
if err != nil {
log.Fatal(err)
}
if err := migrator.NewMigrator(storDB).Migrate(*migrate); err != nil {
if err := migrator.NewMigrator(storDB, *stor_db_type).Migrate(*migrate); err != nil {
log.Fatal(err)
}
log.Print("Done migrating!")

24
migrator/migrator.go
View File

@@ -26,12 +26,13 @@ import (
"github.com/cgrates/cgrates/utils"
)
func NewMigrator(storDB engine.Storage) *Migrator {
return &Migrator{storDB: storDB}
func NewMigrator(storDB engine.Storage, storDBType string) *Migrator {
return &Migrator{storDB: storDB, storDBType: storDBType}
}
type Migrator struct {
storDB engine.Storage
storDB engine.Storage
storDBType string
}
func (m *Migrator) Migrate(taskID string) (err error) {
@@ -76,8 +77,19 @@ func (m *Migrator) migrateCostDetails() (err error) {
if vrs[utils.COST_DETAILS] != 1 { // Right now we only support migrating from version 1
return
}
storSQL := m.storDB.(*engine.SQLStorage)
rows, err := storSQL.Db.Query("SELECT id, tor, direction, tenant, category, account, subject, destination, cost, cost_details FROM cdrs WHERE run_id!= '*raw' and cost_details NOT NULL")
var storSQL *sql.DB
switch m.storDBType {
case utils.MYSQL:
storSQL = m.storDB.(*engine.MySQLStorage).Db
case utils.POSTGRES:
storSQL = m.storDB.(*engine.PostgresStorage).Db
default:
return utils.NewCGRError(utils.Migrator,
utils.MandatoryIEMissingCaps,
utils.UnsupportedDB,
fmt.Sprintf("unsupported database type: <%s>", m.storDBType))
}
rows, err := storSQL.Query("SELECT id, tor, direction, tenant, category, account, subject, destination, cost, cost_details FROM cdrs WHERE run_id!= '*raw' and cost_details IS NOT NULL AND deleted_at IS NULL")
if err != nil {
return utils.NewCGRError(utils.Migrator,
utils.ServerErrorCaps,
@@ -111,7 +123,7 @@ func (m *Migrator) migrateCostDetails() (err error) {
fmt.Sprintf("<Migrator> Error: <%s> when converting into CallCost CDR with id: <%d>", err.Error(), id))
continue
}
if _, err := storSQL.Db.Exec(fmt.Sprintf("UPDATE cdrs SET cost_details='%s' WHERE id=%d", cc.AsJSON(), id)); err != nil {
if _, err := storSQL.Exec(fmt.Sprintf("UPDATE cdrs SET cost_details='%s' WHERE id=%d", cc.AsJSON(), id)); err != nil {
utils.Logger.Warning(
fmt.Sprintf("<Migrator> Error: <%s> updating CDR with id <%d> into StorDB", err.Error(), id))
continue

1
utils/consts.go
View File

@@ -328,4 +328,5 @@ const (
NoStorDBConnection = "not connected to StorDB"
UndefinedVersion = "undefined version"
MetaSetVersions = "*set_versions"
UnsupportedDB = "unsupported database"
)