|
| 1 | +package asyncstore |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "sync" |
| 6 | + |
| 7 | + "github.com/brianvoe/gofakeit/v7" |
| 8 | + |
| 9 | + v1 "github.com/onexstack/b-dms/pkg/api/apiserver/v1" |
| 10 | +) |
| 11 | + |
| 12 | +// FakeStore implements the storage mechanism for FakeData. |
| 13 | +type FakeStore struct { |
| 14 | + // Use RWMutex to allow concurrent reads. |
| 15 | + mu sync.RWMutex |
| 16 | + items map[string]*v1.FakeData |
| 17 | +} |
| 18 | + |
| 19 | +// NewFakeStore creates a new instance of FakeStore. |
| 20 | +func NewFakeStore() *FakeStore { |
| 21 | + return &FakeStore{ |
| 22 | + items: make(map[string]*v1.FakeData), |
| 23 | + } |
| 24 | +} |
| 25 | + |
| 26 | +// Sync simulates data synchronization by generating random fake data. |
| 27 | +func (s *FakeStore) Sync(ctx context.Context) error { |
| 28 | + s.mu.Lock() |
| 29 | + defer s.mu.Unlock() |
| 30 | + |
| 31 | + // Simulate generating 10 random items. |
| 32 | + newItems := make(map[string]*v1.FakeData) |
| 33 | + for i := 0; i < 10; i++ { |
| 34 | + // Generate a random UUID as the key |
| 35 | + id := gofakeit.UUID() |
| 36 | + |
| 37 | + newItems[id] = &v1.FakeData{ |
| 38 | + ID: id, |
| 39 | + Name: gofakeit.AppName(), |
| 40 | + Category: gofakeit.CarMaker(), |
| 41 | + Description: gofakeit.Phrase(), |
| 42 | + Status: gofakeit.RandomString([]string{"active", "pending", "failed"}), |
| 43 | + Score: gofakeit.Float64Range(0.0, 100.0), |
| 44 | + } |
| 45 | + } |
| 46 | + |
| 47 | + // Example: Add a fixed item for deterministic testing. |
| 48 | + fixedID := "fixed-item-001" |
| 49 | + newItems[fixedID] = &v1.FakeData{ |
| 50 | + ID: fixedID, |
| 51 | + Name: "Fixed Test Item", |
| 52 | + Category: "Testing", |
| 53 | + Description: "This item always exists.", |
| 54 | + Status: "active", |
| 55 | + Score: 99.9, |
| 56 | + } |
| 57 | + |
| 58 | + // Atomically replace the entire map. |
| 59 | + s.items = newItems |
| 60 | + |
| 61 | + return nil |
| 62 | +} |
| 63 | + |
| 64 | +// Get retrieves an item by its ID. |
| 65 | +// It returns the item pointer and a boolean indicating if the item was found. |
| 66 | +func (s *FakeStore) Get(id string) (*v1.FakeData, bool) { |
| 67 | + s.mu.RLock() // Acquire read lock |
| 68 | + defer s.mu.RUnlock() |
| 69 | + |
| 70 | + item, ok := s.items[id] |
| 71 | + return item, ok |
| 72 | +} |
| 73 | + |
| 74 | +// List retrieves all items. |
| 75 | +func (s *FakeStore) List() []*v1.FakeData { |
| 76 | + s.mu.RLock() |
| 77 | + defer s.mu.RUnlock() |
| 78 | + |
| 79 | + list := make([]*v1.FakeData, 0, len(s.items)) |
| 80 | + for _, item := range s.items { |
| 81 | + list = append(list, item) |
| 82 | + } |
| 83 | + return list |
| 84 | +} |
0 commit comments