mirror of
https://github.com/tiennm99/miti99bot.git
synced 2026-09-16 16:24:49 +00:00
tiennm99
642fccb7b7
Rename: - Go module github.com/tiennm99/miti99bot-go → github.com/tiennm99/miti99bot - CloudFormation stack miti99bot-aws-port → miti99bot - Drop "port", "Cloud Run", "GCP", "cutover", "Phase NN" framing from active code and docs — project reads as canonical AWS-Lambda from now on. AWS deploy guide + flow fix: - New docs/deploy-aws-free-tier-guide.md — Ubuntu 24.04 ARM64 onboarding with project-local venv (pip awscli + sam-cli), SSM secrets via read -s, idempotent OIDC provider + role creation, $1 budget alarm. - Drop sam build from the pipeline — provided.al2023 + makefile builder expects a Makefile in CodeUri (build/lambda/, the output dir), so the step always fails. sam deploy --template-file template.yaml now reads the raw template and zips build/lambda/ directly. - Rollback section rewritten — use continue-update-rollback / cancel-update-stack / git-SHA redeploy. Drop the broken --use-previous-template recipe. - DynamoDB free-tier row corrected (on-demand is 2.5M read / 1M write request units, not 25 RCU/WCU). Updated: - README.md fully rewritten (drops port/legacy framing, lists modules, points new users at the free-tier guide). - aws/README.md retitled "AWS account setup", phase numbers stripped. - Makefile / .github/workflows/deploy.yml — sam deploy flow. - samconfig.toml — stack_name = "miti99bot". - Go comments — Cloud Run → Lambda, Cloud Scheduler → EventBridge Scheduler, Cloud Logging → CloudWatch Logs. - Struct field GCPProject → FirestoreProject (env GOOGLE_CLOUD_PROJECT unchanged). Plus advisory reports under plans/reports/ from the code-reviewer + researcher passes that informed the fixes. Verified: go vet ./..., go build ./..., go test ./... all green.
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package storage
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// MemoryKVStore is an in-process KVStore for tests and local smoke runs.
|
|
// Data is lost on restart; production uses the DynamoDB provider.
|
|
type MemoryKVStore struct {
|
|
mu sync.RWMutex
|
|
m map[string][]byte
|
|
}
|
|
|
|
// NewMemoryKVStore returns an empty in-memory store.
|
|
func NewMemoryKVStore() *MemoryKVStore {
|
|
return &MemoryKVStore{m: make(map[string][]byte)}
|
|
}
|
|
|
|
func (s *MemoryKVStore) Get(_ context.Context, key string) ([]byte, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
v, ok := s.m[key]
|
|
if !ok {
|
|
return nil, ErrNotFound
|
|
}
|
|
out := make([]byte, len(v))
|
|
copy(out, v)
|
|
return out, nil
|
|
}
|
|
|
|
func (s *MemoryKVStore) GetJSON(ctx context.Context, key string, dst any) error {
|
|
raw, err := s.Get(ctx, key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return json.NewDecoder(bytes.NewReader(raw)).Decode(dst)
|
|
}
|
|
|
|
func (s *MemoryKVStore) Put(_ context.Context, key string, val []byte) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
stored := make([]byte, len(val))
|
|
copy(stored, val)
|
|
s.m[key] = stored
|
|
return nil
|
|
}
|
|
|
|
func (s *MemoryKVStore) PutJSON(ctx context.Context, key string, val any) error {
|
|
raw, err := json.Marshal(val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.Put(ctx, key, raw)
|
|
}
|
|
|
|
func (s *MemoryKVStore) Delete(_ context.Context, key string) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
delete(s.m, key)
|
|
return nil
|
|
}
|
|
|
|
func (s *MemoryKVStore) List(_ context.Context, prefix string) ([]string, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
keys := make([]string, 0)
|
|
for k := range s.m {
|
|
if strings.HasPrefix(k, prefix) {
|
|
keys = append(keys, k)
|
|
}
|
|
}
|
|
sort.Strings(keys)
|
|
return keys, nil
|
|
}
|