mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-13 11:06:25 +05:00
125 lines
3.9 KiB
Go
125 lines
3.9 KiB
Go
/*
|
|
Rating system designed to be used in VoIP Carriers World
|
|
Copyright (C) 2012 Radu Ioan Fericean
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/cgrates/cgrates/timespans"
|
|
"log"
|
|
"encoding/csv"
|
|
"strings"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
separator = flag.String("separator", ",", "Default field separator")
|
|
redissrv = flag.String("redissrv", "127.0.0.1:6379", "redis server address (tcp:127.0.0.1:6379)")
|
|
redisdb = flag.Int("redisdb", 10, "redis database number (10)")
|
|
redispass = flag.String("pass", "", "redis database password")
|
|
flush = flag.Bool("flush", false, "Flush the database before importing")
|
|
monthsFn = flag.String("month", "", "Months file")
|
|
monthdaysFn = flag.String("monthdays", "", "Month days file")
|
|
weekdaysFn = flag.String("weekdays", "", "Week days file")
|
|
destinationsFn = flag.String("destinations", "", "Destinations file")
|
|
ratesFn = flag.String("rates", "", "Rates file")
|
|
timingsFn = flag.String("timings", "", "Timings file")
|
|
ratetimingsFn = flag.String("ratetimings", "", "Rates timings file")
|
|
ratingprofilesFn = flag.String("ratingprofiles", "", "Rating profiles file")
|
|
actionsFn = flag.String("actions", "", "Actions file")
|
|
actiontimingsFn = flag.String("actiontimings", "", "Actions timings file")
|
|
actiontriggersFn = flag.String("actiontriggers", "", "Actions triggers file")
|
|
accountactionsFn = flag.String("accountactions", "", "Account actions file")
|
|
sep rune
|
|
)
|
|
|
|
func writeToDatabase() {
|
|
storage, err := timespans.NewRedisStorage(*redissrv, *redisdb)
|
|
if err != nil {
|
|
log.Fatalf("Could not open database connection: %v", err)
|
|
}
|
|
if *flush {
|
|
storage.Flush()
|
|
}
|
|
log.Print("Destinations")
|
|
for _, d := range destinations {
|
|
storage.SetDestination(d)
|
|
log.Print(d.Id, " : ", d.Prefixes)
|
|
}
|
|
log.Print("Rating profiles")
|
|
for _, cds := range ratingProfiles {
|
|
for _, cd := range cds {
|
|
err = storage.SetActivationPeriodsOrFallback(cd.GetKey(), cd.ActivationPeriods, cd.FallbackKey)
|
|
log.Print(cd.GetKey())
|
|
}
|
|
}
|
|
log.Print("Action timings")
|
|
for k, ats := range actionsTimings {
|
|
storage.SetActionTimings(timespans.ACTION_TIMING_PREFIX+":"+k, ats)
|
|
log.Println(k)
|
|
}
|
|
log.Print("Actions")
|
|
for k, as := range actions {
|
|
storage.SetActions(k, as)
|
|
log.Println(k)
|
|
}
|
|
log.Print("Account actions")
|
|
for _, ub := range accountActions {
|
|
storage.SetUserBalance(ub)
|
|
log.Println(ub.Id)
|
|
}
|
|
}
|
|
|
|
func openFileCSVReader(fn string) (csvReader *csv.Reader, fp *os.File, err error) {
|
|
fp, err = os.Open(fn)
|
|
if err != nil {
|
|
return
|
|
}
|
|
csvReader = csv.NewReader(fp)
|
|
csvReader.Comma = sep
|
|
csvReader.TrailingComma = true
|
|
return
|
|
}
|
|
|
|
func openStringCSVReader(data string) (csvReader *csv.Reader, fp *os.File, err error) {
|
|
csvReader = csv.NewReader(strings.NewReader(data))
|
|
csvReader.Comma = ','
|
|
csvReader.TrailingComma = true
|
|
return
|
|
}
|
|
|
|
type CSVReader struct {
|
|
readerFunc func(string) (*csv.Reader, *os.File, error)
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
sep = []rune(*separator)[0]
|
|
csvr := &CSVReader{openFileCSVReader}
|
|
csvr.loadDestinations(*destinationsFn)
|
|
csvr.loadRates(*ratesFn)
|
|
csvr.loadTimings(*timingsFn)
|
|
csvr.loadRateTimings(*ratetimingsFn)
|
|
csvr.loadRatingProfiles(*ratingprofilesFn)
|
|
csvr.loadActions(*actionsFn)
|
|
csvr.loadActionTimings(*actiontimingsFn)
|
|
csvr.loadActionTriggers(*actiontriggersFn)
|
|
csvr.loadAccountActions(*accountactionsFn)
|
|
writeToDatabase()
|
|
}
|