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

added rating plans history

Browse Source
This commit is contained in:
Radu Ioan Fericean
2013-10-22 14:13:28 +03:00
parent 60c8469477
commit b343d89293
3 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
history/file_scribe.go
View File

@@ -33,6 +33,7 @@ import (
const (
DESTINATIONS_FILE = "destinations.json"
RATING_PLANS_FILE = "rating_plans.json"
RATING_PROFILES_FILE = "rating_profiles.json"
)
@@ -41,6 +42,7 @@ type FileScribe struct {
fileRoot string
gitCommand string
destinations records
ratingPlans records
ratingProfiles records
loopChecker chan int
waitingFile string
@@ -70,6 +72,9 @@ func (s *FileScribe) Record(rec *Record, out *int) error {
case strings.HasPrefix(rec.Key, DESTINATION_PREFIX):
s.destinations = s.destinations.SetOrAdd(&Record{rec.Key[len(DESTINATION_PREFIX):], rec.Object})
fileToSave = DESTINATIONS_FILE
case strings.HasPrefix(rec.Key, RATING_PLAN_PREFIX):
s.ratingPlans = s.ratingPlans.SetOrAdd(&Record{rec.Key[len(RATING_PLAN_PREFIX):], rec.Object})
fileToSave = RATING_PLANS_FILE
case strings.HasPrefix(rec.Key, RATING_PROFILE_PREFIX):
s.ratingProfiles = s.ratingProfiles.SetOrAdd(&Record{rec.Key[len(RATING_PROFILE_PREFIX):], rec.Object})
fileToSave = RATING_PROFILES_FILE
@@ -120,6 +125,11 @@ func (s *FileScribe) gitInit() error {
} else {
f.Close()
}
if f, err := os.Create(filepath.Join(s.fileRoot, RATING_PLANS_FILE)); err != nil {
return errors.New("<History> Error writing rating plans file: " + err.Error())
} else {
f.Close()
}
if f, err := os.Create(filepath.Join(s.fileRoot, RATING_PROFILES_FILE)); err != nil {
return errors.New("<History> Error writing rating profiles file: " + err.Error())
} else {
@@ -159,6 +169,11 @@ func (s *FileScribe) load(filename string) error {
return errors.New("<History> Error loading destinations: " + err.Error())
}
s.destinations.Sort()
case RATING_PLANS_FILE:
if err := d.Decode(&s.ratingPlans); err != nil && err != io.EOF {
return errors.New("<History> Error loading rating plans: " + err.Error())
}
s.ratingPlans.Sort()
case RATING_PROFILES_FILE:
if err := d.Decode(&s.ratingProfiles); err != nil && err != io.EOF {
return errors.New("<History> Error loading rating profiles: " + err.Error())
@@ -182,6 +197,10 @@ func (s *FileScribe) save(filename string) error {
if err := s.format(b, s.destinations); err != nil {
return err
}
case RATING_PLANS_FILE:
if err := s.format(b, s.ratingPlans); err != nil {
return err
}
case RATING_PROFILES_FILE:
if err := s.format(b, s.ratingProfiles); err != nil {
return err

12
history/mock_scribe.go
View File

@@ -30,8 +30,10 @@ import (
type MockScribe struct {
sync.Mutex
destinations records
ratingPlans records
ratingProfiles records
DestBuf bytes.Buffer
RplBuf bytes.Buffer
RpBuf bytes.Buffer
}
@@ -44,6 +46,9 @@ func (s *MockScribe) Record(rec *Record, out *int) error {
case strings.HasPrefix(rec.Key, DESTINATION_PREFIX):
s.destinations = s.destinations.SetOrAdd(&Record{rec.Key[len(DESTINATION_PREFIX):], rec.Object})
s.save(DESTINATIONS_FILE)
case strings.HasPrefix(rec.Key, RATING_PLAN_PREFIX):
s.ratingPlans = s.ratingPlans.SetOrAdd(&Record{rec.Key[len(RATING_PLAN_PREFIX):], rec.Object})
s.save(RATING_PLANS_FILE)
case strings.HasPrefix(rec.Key, RATING_PROFILE_PREFIX):
s.ratingProfiles = s.ratingProfiles.SetOrAdd(&Record{rec.Key[len(RATING_PROFILE_PREFIX):], rec.Object})
s.save(RATING_PROFILES_FILE)
@@ -63,6 +68,13 @@ func (s *MockScribe) save(filename string) error {
if err := s.format(b, s.destinations); err != nil {
return err
}
case RATING_PLANS_FILE:
s.RplBuf.Reset()
b := bufio.NewWriter(&s.RplBuf)
defer b.Flush()
if err := s.format(b, s.ratingPlans); err != nil {
return err
}
case RATING_PROFILES_FILE:
s.RpBuf.Reset()
b := bufio.NewWriter(&s.RpBuf)

1
history/scribe.go
View File

@@ -23,6 +23,7 @@ import (
)
const (
RATING_PLAN_PREFIX = "rpl_"
RATING_PROFILE_PREFIX = "rpf_"
DESTINATION_PREFIX = "dst_"
)