mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-21 09:42:11 +00:00
208 lines
6 KiB
Markdown
208 lines
6 KiB
Markdown
|
[![Go Reference](https://pkg.go.dev/badge/github.com/wk8/go-ordered-map/v2.svg)](https://pkg.go.dev/github.com/wk8/go-ordered-map/v2)
|
||
|
[![Build Status](https://circleci.com/gh/wk8/go-ordered-map.svg?style=svg)](https://app.circleci.com/pipelines/github/wk8/go-ordered-map)
|
||
|
|
||
|
# Golang Ordered Maps
|
||
|
|
||
|
Same as regular maps, but also remembers the order in which keys were inserted, akin to [Python's `collections.OrderedDict`s](https://docs.python.org/3.7/library/collections.html#ordereddict-objects).
|
||
|
|
||
|
It offers the following features:
|
||
|
* optimal runtime performance (all operations are constant time)
|
||
|
* optimal memory usage (only one copy of values, no unnecessary memory allocation)
|
||
|
* allows iterating from newest or oldest keys indifferently, without memory copy, allowing to `break` the iteration, and in time linear to the number of keys iterated over rather than the total length of the ordered map
|
||
|
* supports any generic types for both keys and values. If you're running go < 1.18, you can use [version 1](https://github.com/wk8/go-ordered-map/tree/v1) that takes and returns generic `interface{}`s instead of using generics
|
||
|
* idiomatic API, akin to that of [`container/list`](https://golang.org/pkg/container/list)
|
||
|
* support for JSON and YAML marshalling
|
||
|
|
||
|
## Documentation
|
||
|
|
||
|
[The full documentation is available on pkg.go.dev](https://pkg.go.dev/github.com/wk8/go-ordered-map/v2).
|
||
|
|
||
|
## Installation
|
||
|
```bash
|
||
|
go get -u github.com/wk8/go-ordered-map/v2
|
||
|
```
|
||
|
|
||
|
Or use your favorite golang vendoring tool!
|
||
|
|
||
|
## Supported go versions
|
||
|
|
||
|
Go >= 1.23 is required to use version >= 2.2.0 of this library, as it uses generics and iterators.
|
||
|
|
||
|
if you're running go < 1.23, you can use [version 2.1.8](https://github.com/wk8/go-ordered-map/tree/v2.1.8) instead.
|
||
|
|
||
|
If you're running go < 1.18, you can use [version 1](https://github.com/wk8/go-ordered-map/tree/v1) instead.
|
||
|
|
||
|
## Example / usage
|
||
|
|
||
|
```go
|
||
|
package main
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/wk8/go-ordered-map/v2"
|
||
|
)
|
||
|
|
||
|
func main() {
|
||
|
om := orderedmap.New[string, string]()
|
||
|
|
||
|
om.Set("foo", "bar")
|
||
|
om.Set("bar", "baz")
|
||
|
om.Set("coucou", "toi")
|
||
|
|
||
|
fmt.Println(om.Get("foo")) // => "bar", true
|
||
|
fmt.Println(om.Get("i dont exist")) // => "", false
|
||
|
|
||
|
// iterating pairs from oldest to newest:
|
||
|
for pair := om.Oldest(); pair != nil; pair = pair.Next() {
|
||
|
fmt.Printf("%s => %s\n", pair.Key, pair.Value)
|
||
|
} // prints:
|
||
|
// foo => bar
|
||
|
// bar => baz
|
||
|
// coucou => toi
|
||
|
|
||
|
// iterating over the 2 newest pairs:
|
||
|
i := 0
|
||
|
for pair := om.Newest(); pair != nil; pair = pair.Prev() {
|
||
|
fmt.Printf("%s => %s\n", pair.Key, pair.Value)
|
||
|
i++
|
||
|
if i >= 2 {
|
||
|
break
|
||
|
}
|
||
|
} // prints:
|
||
|
// coucou => toi
|
||
|
// bar => baz
|
||
|
}
|
||
|
```
|
||
|
|
||
|
An `OrderedMap`'s keys must implement `comparable`, and its values can be anything, for example:
|
||
|
|
||
|
```go
|
||
|
type myStruct struct {
|
||
|
payload string
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
om := orderedmap.New[int, *myStruct]()
|
||
|
|
||
|
om.Set(12, &myStruct{"foo"})
|
||
|
om.Set(1, &myStruct{"bar"})
|
||
|
|
||
|
value, present := om.Get(12)
|
||
|
if !present {
|
||
|
panic("should be there!")
|
||
|
}
|
||
|
fmt.Println(value.payload) // => foo
|
||
|
|
||
|
for pair := om.Oldest(); pair != nil; pair = pair.Next() {
|
||
|
fmt.Printf("%d => %s\n", pair.Key, pair.Value.payload)
|
||
|
} // prints:
|
||
|
// 12 => foo
|
||
|
// 1 => bar
|
||
|
}
|
||
|
```
|
||
|
|
||
|
Also worth noting that you can provision ordered maps with a capacity hint, as you would do by passing an optional hint to `make(map[K]V, capacity`):
|
||
|
```go
|
||
|
om := orderedmap.New[int, *myStruct](28)
|
||
|
```
|
||
|
|
||
|
You can also pass in some initial data to store in the map:
|
||
|
```go
|
||
|
om := orderedmap.New[int, string](orderedmap.WithInitialData[int, string](
|
||
|
orderedmap.Pair[int, string]{
|
||
|
Key: 12,
|
||
|
Value: "foo",
|
||
|
},
|
||
|
orderedmap.Pair[int, string]{
|
||
|
Key: 28,
|
||
|
Value: "bar",
|
||
|
},
|
||
|
))
|
||
|
```
|
||
|
|
||
|
`OrderedMap`s also support JSON serialization/deserialization, and preserves order:
|
||
|
|
||
|
```go
|
||
|
// serialization
|
||
|
data, err := json.Marshal(om)
|
||
|
...
|
||
|
|
||
|
// deserialization
|
||
|
om := orderedmap.New[string, string]() // or orderedmap.New[int, any](), or any type you expect
|
||
|
err := json.Unmarshal(data, &om)
|
||
|
...
|
||
|
```
|
||
|
|
||
|
Similarly, it also supports YAML serialization/deserialization using the yaml.v3 package, which also preserves order:
|
||
|
|
||
|
```go
|
||
|
// serialization
|
||
|
data, err := yaml.Marshal(om)
|
||
|
...
|
||
|
|
||
|
// deserialization
|
||
|
om := orderedmap.New[string, string]() // or orderedmap.New[int, any](), or any type you expect
|
||
|
err := yaml.Unmarshal(data, &om)
|
||
|
...
|
||
|
```
|
||
|
|
||
|
## Iterator support (go >= 1.23)
|
||
|
|
||
|
The `FromOldest`, `FromNewest`, `KeysFromOldest`, `KeysFromNewest`, `ValuesFromOldest` and `ValuesFromNewest` methods return iterators over the map's pairs, starting from the oldest or newest pair, respectively.
|
||
|
|
||
|
For example:
|
||
|
|
||
|
```go
|
||
|
om := orderedmap.New[int, string]()
|
||
|
om.Set(1, "foo")
|
||
|
om.Set(2, "bar")
|
||
|
om.Set(3, "baz")
|
||
|
|
||
|
for k, v := range om.FromOldest() {
|
||
|
fmt.Printf("%d => %s\n", k, v)
|
||
|
}
|
||
|
|
||
|
// prints:
|
||
|
// 1 => foo
|
||
|
// 2 => bar
|
||
|
// 3 => baz
|
||
|
|
||
|
for k := range om.KeysNewest() {
|
||
|
fmt.Printf("%d\n", k)
|
||
|
}
|
||
|
|
||
|
// prints:
|
||
|
// 3
|
||
|
// 2
|
||
|
// 1
|
||
|
```
|
||
|
|
||
|
`From` is a convenience function that creates a new `OrderedMap` from an iterator over key-value pairs.
|
||
|
|
||
|
```go
|
||
|
om := orderedmap.New[int, string]()
|
||
|
om.Set(1, "foo")
|
||
|
om.Set(2, "bar")
|
||
|
om.Set(3, "baz")
|
||
|
|
||
|
om2 := orderedmap.From(om.FromOldest())
|
||
|
|
||
|
for k, v := range om2.FromOldest() {
|
||
|
fmt.Printf("%d => %s\n", k, v)
|
||
|
}
|
||
|
|
||
|
// prints:
|
||
|
// 1 => foo
|
||
|
// 2 => bar
|
||
|
// 3 => baz
|
||
|
```
|
||
|
|
||
|
## Alternatives
|
||
|
|
||
|
There are several other ordered map golang implementations out there, but I believe that at the time of writing none of them offer the same functionality as this library; more specifically:
|
||
|
* [iancoleman/orderedmap](https://github.com/iancoleman/orderedmap) only accepts `string` keys, its `Delete` operations are linear
|
||
|
* [cevaris/ordered_map](https://github.com/cevaris/ordered_map) uses a channel for iterations, and leaks goroutines if the iteration is interrupted before fully traversing the map
|
||
|
* [mantyr/iterator](https://github.com/mantyr/iterator) also uses a channel for iterations, and its `Delete` operations are linear
|
||
|
* [samdolan/go-ordered-map](https://github.com/samdolan/go-ordered-map) adds unnecessary locking (users should add their own locking instead if they need it), its `Delete` and `Get` operations are linear, iterations trigger a linear memory allocation
|