You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
2.3 KiB
122 lines
2.3 KiB
// ------------------------------------------------------------------------
|
|
// Project atila
|
|
// Active Thing (activething.com) git.activething.com/go
|
|
//
|
|
// File name packer_state.go
|
|
// Created by DEV
|
|
// Modified 23/01/2024
|
|
//
|
|
// Copyright 2024 activething.com
|
|
// ------------------------------------------------------------------------
|
|
// 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 packs
|
|
|
|
import (
|
|
"activething.com/go/gogo/core/bins"
|
|
"activething.com/go/gogo/core/syncs"
|
|
"sync/atomic"
|
|
)
|
|
|
|
|
|
|
|
|
|
type (
|
|
|
|
|
|
State struct {
|
|
/*Embedded*/*bins.Buffer
|
|
Temp [64]byte
|
|
Level int
|
|
checks map[any]struct{}
|
|
Error error
|
|
}
|
|
|
|
)
|
|
|
|
|
|
func NewState() *State {
|
|
if atomic.CompareAndSwapUint32(&G.poolsSem,0,1) {
|
|
G.statePool = syncs.NewPool[State](
|
|
syncs.PoolWithNew[State](func() *State {
|
|
return &State{
|
|
Buffer: bins.NewBuffer(),
|
|
checks: make(map[any]struct{}),
|
|
}
|
|
}),
|
|
)
|
|
}
|
|
return G.statePool.Get()
|
|
}
|
|
|
|
|
|
func ReleaseState (state *State) {
|
|
if state == nil {
|
|
state.Buffer.Reset()
|
|
state.Error = nil
|
|
if len(state.checks) > 0 {
|
|
for k,_ := range state.checks {
|
|
delete(state.checks,k)
|
|
}
|
|
}
|
|
if atomic.LoadUint32(&G.poolsSem) == 1 {
|
|
G.statePool.Put(state)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
func (s *State) SetError(err error) bool {
|
|
if s.Error != nil {
|
|
return true
|
|
}
|
|
if err == nil {
|
|
return false
|
|
}
|
|
s.Error = err
|
|
s.Reset()
|
|
s.Level = 0
|
|
return true
|
|
}
|
|
|
|
|
|
|
|
func (s *State) CheckCycle (ptr any) (func(),bool) {
|
|
if _,hs := s.checks[ptr]; hs {
|
|
s.SetError(G.ErrRedundantPack)
|
|
return nil,false
|
|
}
|
|
s.checks[ptr]=struct{}{}
|
|
return func(){ delete(s.checks,ptr) },true
|
|
}
|
|
|
|
|
|
|
|
func (s *State) Open(opener byte) {
|
|
s.WriteByte(opener)
|
|
s.Level++
|
|
}
|
|
|
|
|
|
func (s *State) Close(closer byte) {
|
|
s.WriteByte(closer)
|
|
s.Level--
|
|
if s.Level < 0 {
|
|
s.SetError(G.ErrInvalidNode)
|
|
}
|
|
}
|
|
|
|
|