kern/v4/lev1/local_ctx/local_ctx.go

// package local_ctx -- локальный контекст
package local_ctx

import (
	"context"
	"fmt"
	"sync"

	"gitp78su.ipnodns.ru/svi/kern/v4/kc/log_buf"
	. "gitp78su.ipnodns.ru/svi/kern/v4/lev0/ktypes"
	"gitp78su.ipnodns.ru/svi/kern/v4/lev1/ctx_value"
	. "gitp78su.ipnodns.ru/svi/kern/v4/lev1/helpers"
	"gitp78su.ipnodns.ru/svi/kern/v4/lev1/lst_sort"
	. "gitp78su.ipnodns.ru/svi/kern/v4/lev1/result"
)

// LocalCtx -- локальный контекст
type LocalCtx struct {
	sync.RWMutex
	ctx      context.Context // Отменяемый контекст
	fnCancel func()          // Функция отмены контекста

	dictVal map[string]ICtxValue // Словарь различных значений
	lstSort *lst_sort.LstSort    // Сортированный список значений
	log     ILogBuf              // Локальный буфер
}

// NewLocalCtx -- возвращает новый локальный контекст
func NewLocalCtx(ctx context.Context) ILocalCtx {
	Hassert(ctx != nil, "NewLocalCtx(): ctx==nil")
	_ctx, fnCancel := context.WithCancel(ctx)
	sf := &LocalCtx{
		ctx:      _ctx,
		fnCancel: fnCancel,
		dictVal:  map[string]ICtxValue{},
		lstSort:  lst_sort.NewLstSort(),
		log:      log_buf.NewLogBuf(log_buf.OptIsTerm(true), log_buf.OptPrefix("LocalCtx")),
	}
	return sf
}

// Ctx -- возвращает отменяемый контекст
func (sf *LocalCtx) Ctx() context.Context {
	return sf.ctx
}

// Size -- возвращает размер контекста
func (sf *LocalCtx) Size() int {
	sf.RLock()
	defer sf.RUnlock()
	return len(sf.dictVal)
}

// SortedList -- возвращает сортированный список значений
func (sf *LocalCtx) SortedList() []ICtxValue {
	return sf.lstSort.List()
}

// Log -- возвращает локальный буферный лог
func (sf *LocalCtx) Log() ILogBuf {
	return sf.log
}

// Get -- возвращает хранимое значение
func (sf *LocalCtx) Get(key string) IResult[ICtxValue] {
	sf.RLock()
	defer sf.RUnlock()
	Hassert(key != "", "localCtx.Get(): key is empty")
	sf.log.Debug("Get(): key='%v'", key)
	val, isOk := sf.dictVal[key]
	if !isOk {
		err := fmt.Errorf("LocalCtx.Get(): key='%v' not found", key)
		return NewErr[ICtxValue](err)
	}
	return NewRes(val)
}

// Del -- удаляет значение из контекста
func (sf *LocalCtx) Del(key string) {
	sf.Lock()
	defer sf.Unlock()
	sf.log.Debug("Del(): key='%v'", key)
	val := sf.dictVal[key]
	delete(sf.dictVal, key)
	sf.lstSort.Del(val)
}

// Set -- добавляет значение в контекст
func (sf *LocalCtx) Set(key string, val any, comment string) IResult[bool] {
	sf.Lock()
	defer sf.Unlock()
	sf.log.Debug("Set(): key='%v'", key)
	_val, isOk := sf.dictVal[key]
	if isOk {
		val0 := _val.(*ctx_value.CtxValue)
		val0.Lock()
		val0.UpdateAt_ = TimeNowStr()
		val0.Val_ = val
		val0.Unlock()
		return NewRes(true)
	}
	_val = ctx_value.NewCtxValue(key, val, comment)
	sf.dictVal[key] = _val
	sf.lstSort.Add(_val)
	return NewRes(true)
}

// Done -- блокирующий вызов ожидания отмены контекста
func (sf *LocalCtx) Done() {
	<-sf.ctx.Done()
}

// Cancel -- отменяет контекст
func (sf *LocalCtx) Cancel() {
	sf.log.Warn("Cancel()")
	sf.fnCancel()
}