frp/pkg/util/metric/counter.go

// Copyright 2017 fatedier, fatedier@gmail.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 metric

import (
	"sync/atomic"
)

type Counter interface {
	Count() int32
	Inc(int32)
	Dec(int32)
	Snapshot() Counter
	Clear()
}

func NewCounter() Counter {
	return &StandardCounter{
		count: 0,
	}
}

type StandardCounter struct {
	count int32
}

func (c *StandardCounter) Count() int32 {
	return atomic.LoadInt32(&c.count)
}

func (c *StandardCounter) Inc(count int32) {
	atomic.AddInt32(&c.count, count)
}

func (c *StandardCounter) Dec(count int32) {
	atomic.AddInt32(&c.count, -count)
}

func (c *StandardCounter) Snapshot() Counter {
	tmp := &StandardCounter{
		count: atomic.LoadInt32(&c.count),
	}
	return tmp
}

func (c *StandardCounter) Clear() {
	atomic.StoreInt32(&c.count, 0)
}