supported string keys and custom sort, like a v2

Author: go-numb

v2s/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 _numbP
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

v2s/books.go

@@ -0,0 +1,192 @@
+package obm
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/emirpasic/gods/maps/treemap"
+	"github.com/shopspring/decimal"
+)
+
+type Books struct {
+	isBid   bool
+	cap     int
+	remover Remover
+	tree    *treemap.Map
+	Books   []Book
+}
+
+func (p *Books) Len() int {
+	return len(p.Books)
+}
+
+func (p *Books) Less(i, j int) bool {
+	return p.Books[j].Price.GreaterThan(p.Books[i].Price)
+}
+
+func (p *Books) Swap(i, j int) {
+	p.Books[i], p.Books[j] = p.Books[j], p.Books[i]
+}
+
+type Book struct {
+	Price decimal.Decimal
+	Size  decimal.Decimal
+}
+
+func (p *Books) String() string {
+	c := make([]Book, len(p.Books))
+	copy(c, p.Books)
+
+	s := make([]string, len(c))
+	for i := range c {
+		s[i] = fmt.Sprintf("%d - %s:%s", i, c[i].Price.String(), c[i].Size.String())
+	}
+	return strings.Join(s, "\n")
+}
+
+func (p *Books) Size() int {
+	return p.tree.Size()
+}
+
+// Get depth default:10 with sort
+// bids: reverse, asks: sort
+func (p *Books) Get(depth int) *Books {
+	l := p.tree.Size()
+
+	if depth < 10 {
+		depth = 10
+	}
+	if depth > l {
+		depth = l
+	}
+
+	// sorted
+	keys := p.tree.Keys()
+	// Convert keys to decimal and sort
+	decimalKeys := make([]decimal.Decimal, len(keys))
+	for i, key := range keys {
+		decimalKeys[i], _ = decimal.NewFromString(key.(string))
+	}
+
+	b := make([]Book, depth)
+	for i := 0; i < depth; i++ {
+		keyStr := decimalKeys[i].String()
+		b[i] = p.getBookValues(keyStr)
+	}
+
+	// sort
+	p.Books = b
+
+	return p
+}
+
+func (p *Books) Put(book Book) {
+	// インプット情報はすべて入力保存
+	p.tree.Put(book.Price.String(), book.Size.String())
+
+	// If map size is the upper limit, delete data from the upper or lower limits.
+	if p.tree.Size() >= p.cap {
+		switch p.remover {
+		case MAX: // asks
+			maxPrice, _ := p.tree.Max()
+			p.tree.Remove(maxPrice)
+
+		case MIN: // bids
+			minPrice, _ := p.tree.Min()
+			p.tree.Remove(minPrice)
+
+		}
+	}
+}
+
+func (p *Books) Remove(key any) {
+	p.tree.Remove(key)
+}
+
+func (p *Books) Each(fn func(key, val string)) {
+	p.tree.Each(func(key, val any) {
+		fn(key.(string), val.(string))
+	})
+}
+
+func (p *Books) _all() []Book {
+	var books []Book
+
+	keys := p.tree.Keys()
+
+	for i := range keys {
+		books = append(books, p.getBookValues(keys[i]))
+	}
+
+	return books
+}
+
+func (p *Books) getBookValues(key any) Book {
+	val, isFound := p.tree.Get(key)
+	if !isFound {
+		return Book{}
+	}
+
+	return Converter(key, val)
+}
+
+// Converter is a function that converts the price and size of the book.
+func Converter(price, size any) Book {
+	switch v := price.(type) {
+	case int:
+		return converterI(v, size.(int))
+	case float32:
+		return converterF32(v, size.(float32))
+	case float64:
+		return converterF(v, size.(float64))
+	case string:
+		return converterS(v, size.(string))
+
+	case decimal.Decimal:
+		if s, ok := size.(decimal.Decimal); ok {
+			return Book{
+				Price: v,
+				Size:  s,
+			}
+		}
+	}
+
+	return Book{}
+}
+
+func converterI(price, size int) Book {
+	return Book{
+		Price: decimal.NewFromInt(int64(price)),
+		Size:  decimal.NewFromInt(int64(size)),
+	}
+}
+
+func converterF32(price, size float32) Book {
+	return Book{
+		Price: decimal.NewFromFloat32(price),
+		Size:  decimal.NewFromFloat32(size),
+	}
+}
+
+func converterF(price, size float64) Book {
+	return Book{
+		Price: decimal.NewFromFloat(price),
+		Size:  decimal.NewFromFloat(size),
+	}
+}
+
+func converterS(price, size string) Book {
+	p, err := decimal.NewFromString(price)
+	if err != nil {
+		p = decimal.Zero
+	}
+	s, err := decimal.NewFromString(size)
+	if err != nil {
+		s = decimal.Zero
+	}
+
+	return Book{
+		Price: p,
+		Size:  s,
+	}
+}

v2s/books_test.go

@@ -0,0 +1,67 @@
+package obm_test
+
+import (
+	"fmt"
+	"math/rand"
+	"sort"
+	"testing"
+	"time"
+
+	obmv2s "github.com/go-numb/go-obm/v2s"
+	"github.com/shopspring/decimal"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSort(t *testing.T) {
+	now := time.Now()
+	defer func() {
+		// exec time: 2.114597 s
+		fmt.Printf("exec time: %f s\n", time.Since(now).Seconds())
+	}()
+
+	count := 10000
+	o := obmv2s.New("test").SetCap(count, count)
+
+	prices := make([]float64, count)
+
+	s := rand.NewSource(time.Now().Unix())
+	r := rand.New(s)
+
+	for i := 0; i < count; i++ {
+		price, _ := decimal.NewFromString(fmt.Sprintf("%f", r.NormFloat64()*200))
+		size, _ := decimal.NewFromString(fmt.Sprintf("%f", r.NormFloat64()))
+
+		prices = append(prices, price.InexactFloat64())
+		o.Asks.Put(obmv2s.Book{
+			Price: price,
+			Size:  size,
+		})
+
+		price, _ = decimal.NewFromString(fmt.Sprintf("%f", r.NormFloat64()*100))
+		size, _ = decimal.NewFromString(fmt.Sprintf("%f", r.NormFloat64()))
+		o.Bids.Put(obmv2s.Book{
+			Price: price,
+			Size:  size,
+		})
+	}
+
+	assert.Equal(t, count, o.Asks.Size())
+
+	for i := 0; i < len(prices); i++ {
+		fmt.Printf("array: %f\n", prices[i])
+	}
+	fmt.Println("")
+
+	// depth default 10
+	books := o.Asks.Get(2)
+	fmt.Println(books)
+	assert.Equal(t, 10, len(books.Books))
+	fmt.Println("")
+	sort.Sort(sort.Reverse(books))
+	fmt.Println("reverse sort")
+	fmt.Println(books)
+
+	fmt.Println("sort")
+	sort.Sort(books)
+	fmt.Println(books)
+}

v2s/go.mod

@@ -0,0 +1,15 @@
+module github.com/go-numb/go-obm/v2s
+
+go 1.22
+
+require (
+	github.com/emirpasic/gods v1.18.1
+	github.com/shopspring/decimal v1.4.0
+	github.com/stretchr/testify v1.9.0
+)
+
+require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

v2s/go.sum

@@ -0,0 +1,14 @@
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/emirpasic/gods v1.18.1 h1:FXtiHYKDGKCW2KzwZKx0iC0PQmdlorYgdFG9jPXJ1Bc=
+github.com/emirpasic/gods v1.18.1/go.mod h1:8tpGGwCnJ5H4r6BWwaV6OrWmMoPhUl5jm/FMNAnJvWQ=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k=
+github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=