element.go

/*
 * Copyright (c) 2020 Firas M. Darwish ( https://firas.dev.sy )
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package temap

import "time"

type element struct {
	Value     interface{} `json:"value"`
	ExpiresAt int64       `json:"expires_at"` // nanoseconds
}

func (t *TimedMap) SetPermanent(key string, value interface{}) {
	t.mu.Lock()
	if t.tmap[key] != nil {
		t.tmap[key].ExpiresAt = ElementPermanent
		t.tmap[key].Value = value
	} else {
		t.tmap[key] = &element{
			Value:     value,
			ExpiresAt: ElementPermanent,
		}
	}
	t.mu.Unlock()
}

func (t *TimedMap) SetTemporary(key string, value interface{}, expiresAt time.Time) {
	t.mu.Lock()
	if t.tmap[key] != nil {
		t.tmap[key].ExpiresAt = expiresAt.UnixNano()
		t.tmap[key].Value = value
	} else {
		t.tmap[key] = &element{
			Value:     value,
			ExpiresAt: expiresAt.UnixNano(),
		}
	}
	t.mu.Unlock()
}

func (t *TimedMap) Get(key string) (interface{}, int64, bool) {
	t.mu.RLock()
	v := t.tmap[key]
	t.mu.RUnlock()
	if v == nil {
		return nil, ElementDoesntExist, false
	}
	return v.Value, v.ExpiresAt, true
}

func (t *TimedMap) Remove(key string) {
	t.mu.Lock()
	delete(t.tmap, key)
	t.mu.Unlock()
}

func (t *TimedMap) RemoveAll() {
	t.mu.Lock()
	defer t.mu.Unlock()
	for k := range t.tmap {
		delete(t.tmap, k)
	}
}

func (t *TimedMap) MakePermanent(key string) bool {
	t.mu.Lock()

	if t.tmap[key] == nil {
		t.mu.Unlock()
		return false
	}

	t.tmap[key].ExpiresAt = ElementPermanent
	t.mu.Unlock()
	return true
}

func (t *TimedMap) SetExpiry(key string, expiresAt time.Time) bool {
	t.mu.Lock()

	if t.tmap[key] == nil {
		t.mu.Unlock()
		return false
	}

	t.tmap[key].ExpiresAt = expiresAt.UnixNano()
	t.mu.Unlock()
	return true
}