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

Add Get/Set/Remove Resource in DataManager

Browse Source
This commit is contained in:
TeoV
2017-10-17 17:47:21 +03:00
committed by Dan Christian Bogos
parent 717ab57886
commit 3c896d6181
8 changed files with 60 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
engine/storage_map.go
View File

@@ -1189,32 +1189,22 @@ func (ms *MapStorage) RemoveResourceProfile(tenant, id string, transactionID str
return nil
}
func (ms *MapStorage) GetResource(tenant, id string, skipCache bool, transactionID string) (r *Resource, err error) {
func (ms *MapStorage) GetResourceDrv(tenant, id string) (r *Resource, err error) {
ms.mu.RLock()
defer ms.mu.RUnlock()
key := utils.ResourcesPrefix + utils.ConcatenatedKey(tenant, id)
if !skipCache {
if x, ok := cache.Get(key); ok {
if x != nil {
return x.(*Resource), nil
}
return nil, utils.ErrNotFound
}
}
values, ok := ms.dict[key]
if !ok {
cache.Set(key, nil, cacheCommit(transactionID), transactionID)
return nil, utils.ErrNotFound
}
err = ms.ms.Unmarshal(values, r)
err = ms.ms.Unmarshal(values, &r)
if err != nil {
return nil, err
}
cache.Set(key, r, cacheCommit(transactionID), transactionID)
return
}
func (ms *MapStorage) SetResource(r *Resource) (err error) {
func (ms *MapStorage) SetResourceDrv(r *Resource) (err error) {
ms.mu.Lock()
defer ms.mu.Unlock()
result, err := ms.ms.Marshal(r)
@@ -1225,12 +1215,11 @@ func (ms *MapStorage) SetResource(r *Resource) (err error) {
return
}
func (ms *MapStorage) RemoveResource(tenant, id string, transactionID string) (err error) {
func (ms *MapStorage) RemoveResourceDrv(tenant, id string) (err error) {
ms.mu.Lock()
defer ms.mu.Unlock()
key := utils.ResourcesPrefix + utils.ConcatenatedKey(tenant, id)
delete(ms.dict, key)
cache.RemKey(key, cacheCommit(transactionID), transactionID)
return
}