initial inventory service

This commit is contained in:
matst80
2025-11-04 12:48:41 +01:00
parent 42e38504a3
commit c67ebd7a5f
7 changed files with 237 additions and 2 deletions

41
cmd/inventory/main.go Normal file
View File

@@ -0,0 +1,41 @@
package main
import (
"context"
"fmt"
"github.com/redis/go-redis/v9"
"github.com/redis/go-redis/v9/maintnotifications"
)
func main() {
var ctx = context.Background()
rdb := redis.NewClient(&redis.Options{
Addr: "10.10.3.18:6379",
Password: "slaskredis", // no password set
DB: 0, // use default DB
MaintNotificationsConfig: &maintnotifications.Config{
Mode: maintnotifications.ModeDisabled,
},
})
err := rdb.Set(ctx, "key", "value", 0).Err()
if err != nil {
panic(err)
}
val, err := rdb.Get(ctx, "key").Result()
if err != nil {
panic(err)
}
fmt.Println("key", val)
val2, err := rdb.Get(ctx, "key2").Result()
if err == redis.Nil {
fmt.Println("key2 does not exist")
} else if err != nil {
panic(err)
} else {
fmt.Println("key2", val2)
}
}

2
go.mod
View File

@@ -8,6 +8,7 @@ require (
github.com/matst80/slask-finder v0.0.0-20251023104024-f788e5a51d68
github.com/prometheus/client_golang v1.23.2
github.com/rabbitmq/amqp091-go v1.10.0
github.com/redis/go-redis/v9 v9.16.0
google.golang.org/grpc v1.76.0
google.golang.org/protobuf v1.36.10
k8s.io/api v0.34.1
@@ -21,6 +22,7 @@ require (
github.com/bits-and-blooms/bitset v1.24.1 // indirect
github.com/cespare/xxhash/v2 v2.3.0 // indirect
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
github.com/dprotaso/go-yit v0.0.0-20220510233725-9ba8df137936 // indirect
github.com/emicklei/go-restful/v3 v3.13.0 // indirect
github.com/fxamacker/cbor/v2 v2.9.0 // indirect

8
go.sum
View File

@@ -4,6 +4,10 @@ github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
github.com/bits-and-blooms/bitset v1.24.1 h1:hqnfFbjjk3pxGa5E9Ho3hjoU7odtUuNmJ9Ao+Bo8s1c=
github.com/bits-and-blooms/bitset v1.24.1/go.mod h1:7hO7Gc7Pp1vODcmWvKMRA9BNmbv6a/7QIWpPxHddWR8=
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c=
github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0=
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
@@ -13,6 +17,8 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=
github.com/dprotaso/go-yit v0.0.0-20191028211022-135eb7262960/go.mod h1:9HQzr9D/0PGwMEbC3d5AB7oi67+h4TsQqItC1GVYG58=
github.com/dprotaso/go-yit v0.0.0-20220510233725-9ba8df137936 h1:PRxIJD8XjimM5aTknUK9w6DHLDox2r2M3DI4i2pnd3w=
github.com/dprotaso/go-yit v0.0.0-20220510233725-9ba8df137936/go.mod h1:ttYvX5qlB+mlV1okblJqcSMtR4c52UKxDiX9GRBS8+Q=
@@ -168,6 +174,8 @@ github.com/prometheus/procfs v0.18.0 h1:2QTA9cKdznfYJz7EDaa7IiJobHuV7E1WzeBwcrhk
github.com/prometheus/procfs v0.18.0/go.mod h1:M0aotyiemPhBCM0z5w87kL22CxfcH05ZpYlu+b4J7mw=
github.com/rabbitmq/amqp091-go v1.10.0 h1:STpn5XsHlHGcecLmMFCtg7mqq0RnD+zFr4uzukfVhBw=
github.com/rabbitmq/amqp091-go v1.10.0/go.mod h1:Hy4jKW5kQART1u+JkDTF9YYOQUHXqMuhrgxOEeS7G4o=
github.com/redis/go-redis/v9 v9.16.0 h1:OotgqgLSRCmzfqChbQyG1PHC3tLNR89DG4jdOERSEP4=
github.com/redis/go-redis/v9 v9.16.0/go.mod h1:u410H11HMLoB+TP67dz8rL9s6QW2j76l0//kSOd3370=
github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII=
github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
github.com/sergi/go-diff v1.1.0 h1:we8PVUC3FE2uYfodKH/nBHMSetSfHDR6scGdBi+erh0=

View File

@@ -21,8 +21,8 @@ func TestRegisteredMutationBasics(t *testing.T) {
func(state *cartState, msg *messages.AddItem) error {
state.calls++
// copy to avoid external mutation side-effects (not strictly necessary for the test)
cp := *msg
state.lastAdded = &cp
cp := msg
state.lastAdded = cp
return nil
},
func() *messages.AddItem { return &messages.AddItem{} },

View File

@@ -0,0 +1,47 @@
package inventory
import (
"errors"
"sync"
)
type MemoryInventoryService struct {
warehouses map[LocationID]*Warehouse
mu sync.RWMutex
}
func NewMemoryInventoryService() *MemoryInventoryService {
return &MemoryInventoryService{
warehouses: make(map[LocationID]*Warehouse),
}
}
func (s *MemoryInventoryService) AddWarehouse(warehouse *Warehouse) {
s.mu.Lock()
defer s.mu.Unlock()
s.warehouses[warehouse.ID] = warehouse
}
func (s *MemoryInventoryService) GetInventory(sku SKU, locationID LocationID) (*InventoryItem, error) {
s.mu.RLock()
defer s.mu.RUnlock()
warehouse, ok := s.warehouses[locationID]
if !ok {
return nil, errors.New("warehouse not found")
}
for _, item := range warehouse.Inventory {
if item.SKU == sku {
return &item, nil
}
}
return nil, errors.New("sku not found in warehouse")
}
func (s *MemoryInventoryService) ReserveInventory(req ...ReserveRequest) error {
// We'll implement the reservation logic using Lua script execution here
// For now, let's just return an error indicating it's not implemented
return errors.New("reservation not implemented yet")
}

View File

@@ -0,0 +1,107 @@
package inventory
import (
"context"
"encoding/json"
"errors"
"github.com/redis/go-redis/v9"
)
type RedisInventoryService struct {
client *redis.Client
ctx context.Context
luaScripts map[string]*redis.Script
}
func NewRedisInventoryService(client *redis.Client, ctx context.Context) (*RedisInventoryService, error) {
rdb := client
// Ping Redis to check connection
_, err := rdb.Ping(ctx).Result()
if err != nil {
return nil, err
}
return &RedisInventoryService{
client: rdb,
ctx: ctx,
luaScripts: make(map[string]*redis.Script),
}, nil
}
func (s *RedisInventoryService) LoadLuaScript(key string) error {
// Get the script from Redis
script, err := s.client.Get(s.ctx, key).Result()
if err != nil {
return err
}
// Load the script into the luaScripts cache
s.luaScripts[key] = redis.NewScript(script)
return nil
}
func (s *RedisInventoryService) AddWarehouse(warehouse *Warehouse) error {
// Convert warehouse to Redis-friendly format
data := map[string]interface{}{
"id": string(warehouse.ID),
"name": warehouse.Name,
"inventory": warehouse.Inventory,
}
// Store in Redis with a key pattern like "warehouse:<ID>"
key := "warehouse:" + string(warehouse.ID)
_, err := s.client.HMSet(s.ctx, key, data).Result()
return err
}
func (s *RedisInventoryService) GetInventory(sku SKU, locationID LocationID) (*InventoryItem, error) {
// Get the warehouse from Redis
key := "warehouse:" + string(locationID)
result, err := s.client.HGetAll(s.ctx, key).Result()
if err != nil {
return nil, err
}
// Parse the inventory items
var inventoryItems []InventoryItem
for _, itemData := range result {
var item InventoryItem
if err := json.Unmarshal([]byte(itemData), &item); err == nil {
inventoryItems = append(inventoryItems, item)
}
}
// Find the requested SKU
for _, item := range inventoryItems {
if item.SKU == sku {
return &item, nil
}
}
return nil, errors.New("sku not found in warehouse")
}
func (s *RedisInventoryService) ReserveInventory(req ReserveRequest) error {
return nil
// Get the Lua script from Redis
// key := "lua:reserve_inventory"
// script, err := s.client.Get(s.ctx, key).Result()
// if err != nil {
// return err
// }
// luaScript := redis.NewScript(script)
// // Prepare arguments for the Lua script
// args := []interface{}{
// string(req.LocationID),
// string(req.SKU),
// req.Quantity,
// }
// // Execute the Lua script
// cmd := s.client.Eval(s.ctx, luaScript, len(args), args...)
//return err
}

30
pkg/inventory/types.go Normal file
View File

@@ -0,0 +1,30 @@
package inventory
import "time"
type SKU string
type LocationID string
type InventoryItem struct {
SKU SKU
Quantity int
LastUpdate time.Time
}
type Warehouse struct {
ID LocationID
Name string
Inventory []InventoryItem
}
type InventoryService interface {
GetInventory(sku SKU, locationID LocationID) (uint32, error)
ReserveInventory(req ...ReserveRequest) error
}
type ReserveRequest struct {
SKU SKU
LocationID LocationID
Quantity uint32
}