git.iko.soy/crush
on main help sign in
fix: race in agent.go (#1853)

Signed-off-by: Carlos Alexandro Becker <caarlos0@users.noreply.github.com>

commitf66762b917c3ac133ca8f1fc430c5cbd524412f0 authorCarlos Alexandro Becker <caarlos0@users.noreply.github.com> dateWed 14 Jan 2026, 12:16 (9 months ago) parent8f2ae5c
6 files changed+218−145
compare with main
unifiedsplit
internal/agent/agent.go+59−48 internal/agent/event.go+6−6 internal/csync/slices.go+10−34 internal/csync/slices_test.go−57 internal/csync/value.go+44 internal/csync/value_test.go+99
internal/agent/agent.go+59−48
@@ -87,12 +87,13 @@ type Model struct {
8787}
8888
8989type sessionAgent struct {
90− largeModel Model
91− smallModel Model
92− systemPromptPrefix string
93− systemPrompt string
90+ largeModel *csync.Value[Model]
91+ smallModel *csync.Value[Model]
92+ systemPromptPrefix *csync.Value[string]
93+ systemPrompt *csync.Value[string]
94+ tools *csync.Slice[fantasy.AgentTool]
95+
9496 isSubAgent bool
95− tools []fantasy.AgentTool
9697 sessions session.Service
9798 messages message.Service
9899 disableAutoSummarize bool
@@ -119,15 +120,15 @@ func NewSessionAgent(
119120 opts SessionAgentOptions,
120121) SessionAgent {
121122 return &sessionAgent{
122− largeModel: opts.LargeModel,
123− smallModel: opts.SmallModel,
124− systemPromptPrefix: opts.SystemPromptPrefix,
125− systemPrompt: opts.SystemPrompt,
123+ largeModel: csync.NewValue(opts.LargeModel),
124+ smallModel: csync.NewValue(opts.SmallModel),
125+ systemPromptPrefix: csync.NewValue(opts.SystemPromptPrefix),
126+ systemPrompt: csync.NewValue(opts.SystemPrompt),
126127 isSubAgent: opts.IsSubAgent,
127128 sessions: opts.Sessions,
128129 messages: opts.Messages,
129130 disableAutoSummarize: opts.DisableAutoSummarize,
130− tools: opts.Tools,
131+ tools: csync.NewSliceFrom(opts.Tools),
131132 isYolo: opts.IsYolo,
132133 messageQueue: csync.NewMap[string, []SessionAgentCall](),
133134 activeRequests: csync.NewMap[string, context.CancelFunc](),
@@ -153,15 +154,21 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
153154 return nil, nil
154155 }
155156
156− if len(a.tools) > 0 {
157+ // Copy mutable fields under lock to avoid races with SetTools/SetModels.
158+ agentTools := a.tools.Copy()
159+ largeModel := a.largeModel.Get()
160+ systemPrompt := a.systemPrompt.Get()
161+ promptPrefix := a.systemPromptPrefix.Get()
162+
163+ if len(agentTools) > 0 {
157164 // Add Anthropic caching to the last tool.
158− a.tools[len(a.tools)-1].SetProviderOptions(a.getCacheControlOptions())
165+ agentTools[len(agentTools)-1].SetProviderOptions(a.getCacheControlOptions())
159166 }
160167
161168 agent := fantasy.NewAgent(
162− a.largeModel.Model,
163− fantasy.WithSystemPrompt(a.systemPrompt),
164− fantasy.WithTools(a.tools...),
169+ largeModel.Model,
170+ fantasy.WithSystemPrompt(systemPrompt),
171+ fantasy.WithTools(agentTools...),
165172 )
166173
167174 sessionLock := sync.Mutex{}
@@ -234,7 +241,7 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
234241 prepared.Messages = append(prepared.Messages, userMessage.ToAIMessage()...)
235242 }
236243
237− prepared.Messages = a.workaroundProviderMediaLimitations(prepared.Messages)
244+ prepared.Messages = a.workaroundProviderMediaLimitations(prepared.Messages, largeModel)
238245
239246 lastSystemRoleInx := 0
240247 systemMessageUpdated := false
@@ -252,7 +259,7 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
252259 }
253260 }
254261
255− if promptPrefix := a.promptPrefix(); promptPrefix != "" {
262+ if promptPrefix != "" {
256263 prepared.Messages = append([]fantasy.Message{fantasy.NewSystemMessage(promptPrefix)}, prepared.Messages...)
257264 }
258265
@@ -260,15 +267,15 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
260267 assistantMsg, err = a.messages.Create(callContext, call.SessionID, message.CreateMessageParams{
261268 Role: message.Assistant,
262269 Parts: []message.ContentPart{},
263− Model: a.largeModel.ModelCfg.Model,
264− Provider: a.largeModel.ModelCfg.Provider,
270+ Model: largeModel.ModelCfg.Model,
271+ Provider: largeModel.ModelCfg.Provider,
265272 })
266273 if err != nil {
267274 return callContext, prepared, err
268275 }
269276 callContext = context.WithValue(callContext, tools.MessageIDContextKey, assistantMsg.ID)
270− callContext = context.WithValue(callContext, tools.SupportsImagesContextKey, a.largeModel.CatwalkCfg.SupportsImages)
271− callContext = context.WithValue(callContext, tools.ModelNameContextKey, a.largeModel.CatwalkCfg.Name)
277+ callContext = context.WithValue(callContext, tools.SupportsImagesContextKey, largeModel.CatwalkCfg.SupportsImages)
278+ callContext = context.WithValue(callContext, tools.ModelNameContextKey, largeModel.CatwalkCfg.Name)
272279 currentAssistant = &assistantMsg
273280 return callContext, prepared, err
274281 },
@@ -362,7 +369,7 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
362369 sessionLock.Unlock()
363370 return getSessionErr
364371 }
365− a.updateSessionUsage(a.largeModel, &updatedSession, stepResult.Usage, a.openrouterCost(stepResult.ProviderMetadata))
372+ a.updateSessionUsage(largeModel, &updatedSession, stepResult.Usage, a.openrouterCost(stepResult.ProviderMetadata))
366373 _, sessionErr := a.sessions.Save(genCtx, updatedSession)
367374 sessionLock.Unlock()
368375 if sessionErr != nil {
@@ -372,7 +379,7 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
372379 },
373380 StopWhen: []fantasy.StopCondition{
374381 func(_ []fantasy.StepResult) bool {
375− cw := int64(a.largeModel.CatwalkCfg.ContextWindow)
382+ cw := int64(largeModel.CatwalkCfg.ContextWindow)
376383 tokens := currentSession.CompletionTokens + currentSession.PromptTokens
377384 remaining := cw - tokens
378385 var threshold int64
@@ -474,7 +481,7 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy
474481 currentAssistant.AddFinish(
475482 message.FinishReasonError,
476483 "Copilot model not enabled",
477− fmt.Sprintf("%q is not enabled in Copilot. Go to the following page to enable it. Then, wait 5 minutes before trying again. %s", a.largeModel.CatwalkCfg.Name, link),
484+ fmt.Sprintf("%q is not enabled in Copilot. Go to the following page to enable it. Then, wait 5 minutes before trying again. %s", largeModel.CatwalkCfg.Name, link),
478485 )
479486 } else {
480487 currentAssistant.AddFinish(message.FinishReasonError, cmp.Or(stringext.Capitalize(providerErr.Title), defaultTitle), providerErr.Message)
@@ -529,6 +536,10 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
529536 return ErrSessionBusy
530537 }
531538
539+ // Copy mutable fields under lock to avoid races with SetModels.
540+ largeModel := a.largeModel.Get()
541+ systemPromptPrefix := a.systemPromptPrefix.Get()
542+
532543 currentSession, err := a.sessions.Get(ctx, sessionID)
533544 if err != nil {
534545 return fmt.Errorf("failed to get session: %w", err)
@@ -549,13 +560,13 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
549560 defer a.activeRequests.Del(sessionID)
550561 defer cancel()
551562
552− agent := fantasy.NewAgent(a.largeModel.Model,
563+ agent := fantasy.NewAgent(largeModel.Model,
553564 fantasy.WithSystemPrompt(string(summaryPrompt)),
554565 )
555566 summaryMessage, err := a.messages.Create(ctx, sessionID, message.CreateMessageParams{
556567 Role: message.Assistant,
557− Model: a.largeModel.Model.Model(),
558− Provider: a.largeModel.Model.Provider(),
568+ Model: largeModel.Model.Model(),
569+ Provider: largeModel.Model.Provider(),
559570 IsSummaryMessage: true,
560571 })
561572 if err != nil {
@@ -570,8 +581,8 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
570581 ProviderOptions: opts,
571582 PrepareStep: func(callContext context.Context, options fantasy.PrepareStepFunctionOptions) (_ context.Context, prepared fantasy.PrepareStepResult, err error) {
572583 prepared.Messages = options.Messages
573− if a.systemPromptPrefix != "" {
574− prepared.Messages = append([]fantasy.Message{fantasy.NewSystemMessage(a.systemPromptPrefix)}, prepared.Messages...)
584+ if systemPromptPrefix != "" {
585+ prepared.Messages = append([]fantasy.Message{fantasy.NewSystemMessage(systemPromptPrefix)}, prepared.Messages...)
575586 }
576587 return callContext, prepared, nil
577588 },
@@ -622,7 +633,7 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
622633 }
623634 }
624635
625− a.updateSessionUsage(a.largeModel, &currentSession, resp.TotalUsage, openrouterCost)
636+ a.updateSessionUsage(largeModel, &currentSession, resp.TotalUsage, openrouterCost)
626637
627638 // Just in case, get just the last usage info.
628639 usage := resp.Response.Usage
@@ -730,9 +741,13 @@ func (a *sessionAgent) generateTitle(ctx context.Context, sessionID string, user
730741 return
731742 }
732743
744+ smallModel := a.smallModel.Get()
745+ largeModel := a.largeModel.Get()
746+ systemPromptPrefix := a.systemPromptPrefix.Get()
747+
733748 var maxOutputTokens int64 = 40
734− if a.smallModel.CatwalkCfg.CanReason {
735− maxOutputTokens = a.smallModel.CatwalkCfg.DefaultMaxTokens
749+ if smallModel.CatwalkCfg.CanReason {
750+ maxOutputTokens = smallModel.CatwalkCfg.DefaultMaxTokens
736751 }
737752
738753 newAgent := func(m fantasy.LanguageModel, p []byte, tok int64) fantasy.Agent {
@@ -746,9 +761,9 @@ func (a *sessionAgent) generateTitle(ctx context.Context, sessionID string, user
746761 Prompt: fmt.Sprintf("Generate a concise title for the following content:\n\n%s\n <think>\n\n</think>", userPrompt),
747762 PrepareStep: func(callCtx context.Context, opts fantasy.PrepareStepFunctionOptions) (_ context.Context, prepared fantasy.PrepareStepResult, err error) {
748763 prepared.Messages = opts.Messages
749− if a.systemPromptPrefix != "" {
764+ if systemPromptPrefix != "" {
750765 prepared.Messages = append([]fantasy.Message{
751− fantasy.NewSystemMessage(a.systemPromptPrefix),
766+ fantasy.NewSystemMessage(systemPromptPrefix),
752767 }, prepared.Messages...)
753768 }
754769 return callCtx, prepared, nil
@@ -756,7 +771,7 @@ func (a *sessionAgent) generateTitle(ctx context.Context, sessionID string, user
756771 }
757772
758773 // Use the small model to generate the title.
759− model := &a.smallModel
774+ model := smallModel
760775 agent := newAgent(model.Model, titlePrompt, maxOutputTokens)
761776 resp, err := agent.Stream(ctx, streamCall)
762777 if err == nil {
@@ -765,7 +780,7 @@ func (a *sessionAgent) generateTitle(ctx context.Context, sessionID string, user
765780 } else {
766781 // It didn't work. Let's try with the big model.
767782 slog.Error("error generating title with small model; trying big model", "err", err)
768− model = &a.largeModel
783+ model = largeModel
769784 agent = newAgent(model.Model, titlePrompt, maxOutputTokens)
770785 resp, err = agent.Stream(ctx, streamCall)
771786 if err == nil {
@@ -960,24 +975,20 @@ func (a *sessionAgent) QueuedPromptsList(sessionID string) []string {
960975}
961976
962977func (a *sessionAgent) SetModels(large Model, small Model) {
963− a.largeModel = large
964− a.smallModel = small
978+ a.largeModel.Set(large)
979+ a.smallModel.Set(small)
965980}
966981
967982func (a *sessionAgent) SetTools(tools []fantasy.AgentTool) {
968− a.tools = tools
983+ a.tools.SetSlice(tools)
969984}
970985
971986func (a *sessionAgent) SetSystemPrompt(systemPrompt string) {
972− a.systemPrompt = systemPrompt
987+ a.systemPrompt.Set(systemPrompt)
973988}
974989
975990func (a *sessionAgent) Model() Model {
976− return a.largeModel
977−}
978−
979−func (a *sessionAgent) promptPrefix() string {
980− return a.systemPromptPrefix
991+ return a.largeModel.Get()
981992}
982993
983994// convertToToolResult converts a fantasy tool result to a message tool result.
@@ -1034,9 +1045,9 @@ func (a *sessionAgent) convertToToolResult(result fantasy.ToolResultContent) mes
10341045//
10351046// BEFORE: [tool result: image data]
10361047// AFTER: [tool result: "Image loaded - see attached"], [user: image attachment]
1037−func (a *sessionAgent) workaroundProviderMediaLimitations(messages []fantasy.Message) []fantasy.Message {
1038− providerSupportsMedia := a.largeModel.ModelCfg.Provider == string(catwalk.InferenceProviderAnthropic) ||
1039− a.largeModel.ModelCfg.Provider == string(catwalk.InferenceProviderBedrock)
1048+func (a *sessionAgent) workaroundProviderMediaLimitations(messages []fantasy.Message, largeModel Model) []fantasy.Message {
1049+ providerSupportsMedia := largeModel.ModelCfg.Provider == string(catwalk.InferenceProviderAnthropic) ||
1050+ largeModel.ModelCfg.Provider == string(catwalk.InferenceProviderBedrock)
10401051
10411052 if providerSupportsMedia {
10421053 return messages
internal/agent/event.go+6−6
@@ -7,23 +7,23 @@ import (
77 "github.com/charmbracelet/crush/internal/event"
88)
99
10−func (a sessionAgent) eventPromptSent(sessionID string) {
10+func (a *sessionAgent) eventPromptSent(sessionID string) {
1111 event.PromptSent(
12− a.eventCommon(sessionID, a.largeModel)...,
12+ a.eventCommon(sessionID, a.largeModel.Get())...,
1313 )
1414}
1515
16−func (a sessionAgent) eventPromptResponded(sessionID string, duration time.Duration) {
16+func (a *sessionAgent) eventPromptResponded(sessionID string, duration time.Duration) {
1717 event.PromptResponded(
1818 append(
19− a.eventCommon(sessionID, a.largeModel),
19+ a.eventCommon(sessionID, a.largeModel.Get()),
2020 "prompt duration pretty", duration.String(),
2121 "prompt duration in seconds", int64(duration.Seconds()),
2222 )...,
2323 )
2424}
2525
26−func (a sessionAgent) eventTokensUsed(sessionID string, model Model, usage fantasy.Usage, cost float64) {
26+func (a *sessionAgent) eventTokensUsed(sessionID string, model Model, usage fantasy.Usage, cost float64) {
2727 event.TokensUsed(
2828 append(
2929 a.eventCommon(sessionID, model),
@@ -37,7 +37,7 @@ func (a sessionAgent) eventTokensUsed(sessionID string, model Model, usage fanta
3737 )
3838}
3939
40−func (a sessionAgent) eventCommon(sessionID string, model Model) []any {
40+func (a *sessionAgent) eventCommon(sessionID string, model Model) []any {
4141 m := model.ModelCfg
4242
4343 return []any{
internal/csync/slices.go+10−34
@@ -2,7 +2,6 @@ package csync
22
33import (
44 "iter"
5− "slices"
65 "sync"
76)
87
@@ -63,24 +62,6 @@ func (s *Slice[T]) Append(items ...T) {
6362 s.inner = append(s.inner, items...)
6463}
6564
66−// Prepend adds an element to the beginning of the slice.
67−func (s *Slice[T]) Prepend(item T) {
68− s.mu.Lock()
69− defer s.mu.Unlock()
70− s.inner = append([]T{item}, s.inner...)
71−}
72−
73−// Delete removes the element at the specified index.
74−func (s *Slice[T]) Delete(index int) bool {
75− s.mu.Lock()
76− defer s.mu.Unlock()
77− if index < 0 || index >= len(s.inner) {
78− return false
79− }
80− s.inner = slices.Delete(s.inner, index, index+1)
81− return true
82−}
83−
8465// Get returns the element at the specified index.
8566func (s *Slice[T]) Get(index int) (T, bool) {
8667 s.mu.RLock()
@@ -92,17 +73,6 @@ func (s *Slice[T]) Get(index int) (T, bool) {
9273 return s.inner[index], true
9374}
9475
95−// Set updates the element at the specified index.
96−func (s *Slice[T]) Set(index int, item T) bool {
97− s.mu.Lock()
98− defer s.mu.Unlock()
99− if index < 0 || index >= len(s.inner) {
100− return false
101− }
102− s.inner[index] = item
103− return true
104−}
105−
10676// Len returns the number of elements in the slice.
10777func (s *Slice[T]) Len() int {
10878 s.mu.RLock()
@@ -131,10 +101,7 @@ func (s *Slice[T]) Seq() iter.Seq[T] {
131101
132102// Seq2 returns an iterator that yields index-value pairs from the slice.
133103func (s *Slice[T]) Seq2() iter.Seq2[int, T] {
134− s.mu.RLock()
135− items := make([]T, len(s.inner))
136− copy(items, s.inner)
137− s.mu.RUnlock()
104+ items := s.Copy()
138105 return func(yield func(int, T) bool) {
139106 for i, v := range items {
140107 if !yield(i, v) {
@@ -143,3 +110,12 @@ func (s *Slice[T]) Seq2() iter.Seq2[int, T] {
143110 }
144111 }
145112}
113+
114+// Copy returns a copy of the inner slice.
115+func (s *Slice[T]) Copy() []T {
116+ s.mu.RLock()
117+ defer s.mu.RUnlock()
118+ items := make([]T, len(s.inner))
119+ copy(items, s.inner)
120+ return items
121+}
internal/csync/slices_test.go−57
@@ -109,44 +109,6 @@ func TestSlice(t *testing.T) {
109109 require.Equal(t, "world", val)
110110 })
111111
112− t.Run("Prepend", func(t *testing.T) {
113− s := NewSlice[string]()
114− s.Append("world")
115− s.Prepend("hello")
116−
117− require.Equal(t, 2, s.Len())
118− val, ok := s.Get(0)
119− require.True(t, ok)
120− require.Equal(t, "hello", val)
121−
122− val, ok = s.Get(1)
123− require.True(t, ok)
124− require.Equal(t, "world", val)
125− })
126−
127− t.Run("Delete", func(t *testing.T) {
128− s := NewSliceFrom([]int{1, 2, 3, 4, 5})
129−
130− // Delete middle element
131− ok := s.Delete(2)
132− require.True(t, ok)
133− require.Equal(t, 4, s.Len())
134−
135− expected := []int{1, 2, 4, 5}
136− actual := slices.Collect(s.Seq())
137− require.Equal(t, expected, actual)
138−
139− // Delete out of bounds
140− ok = s.Delete(10)
141− require.False(t, ok)
142− require.Equal(t, 4, s.Len())
143−
144− // Delete negative index
145− ok = s.Delete(-1)
146− require.False(t, ok)
147− require.Equal(t, 4, s.Len())
148− })
149−
150112 t.Run("Get", func(t *testing.T) {
151113 s := NewSliceFrom([]string{"a", "b", "c"})
152114
@@ -163,25 +125,6 @@ func TestSlice(t *testing.T) {
163125 require.False(t, ok)
164126 })
165127
166− t.Run("Set", func(t *testing.T) {
167− s := NewSliceFrom([]string{"a", "b", "c"})
168−
169− ok := s.Set(1, "modified")
170− require.True(t, ok)
171−
172− val, ok := s.Get(1)
173− require.True(t, ok)
174− require.Equal(t, "modified", val)
175−
176− // Out of bounds
177− ok = s.Set(10, "invalid")
178− require.False(t, ok)
179−
180− // Negative index
181− ok = s.Set(-1, "invalid")
182− require.False(t, ok)
183− })
184−
185128 t.Run("SetSlice", func(t *testing.T) {
186129 s := NewSlice[int]()
187130 s.Append(1)
internal/csync/value.gonew file+44
@@ -0,0 +1,44 @@
1+package csync
2+
3+import (
4+ "reflect"
5+ "sync"
6+)
7+
8+// Value is a generic thread-safe wrapper for any value type.
9+//
10+// For slices, use [Slice]. For maps, use [Map]. Pointers are not supported.
11+type Value[T any] struct {
12+ v T
13+ mu sync.RWMutex
14+}
15+
16+// NewValue creates a new Value with the given initial value.
17+//
18+// Panics if t is a pointer, slice, or map. Use the dedicated types for those.
19+func NewValue[T any](t T) *Value[T] {
20+ v := reflect.ValueOf(t)
21+ switch v.Kind() {
22+ case reflect.Pointer:
23+ panic("csync.Value does not support pointer types")
24+ case reflect.Slice:
25+ panic("csync.Value does not support slice types; use csync.Slice")
26+ case reflect.Map:
27+ panic("csync.Value does not support map types; use csync.Map")
28+ }
29+ return &Value[T]{v: t}
30+}
31+
32+// Get returns the current value.
33+func (v *Value[T]) Get() T {
34+ v.mu.RLock()
35+ defer v.mu.RUnlock()
36+ return v.v
37+}
38+
39+// Set updates the value.
40+func (v *Value[T]) Set(t T) {
41+ v.mu.Lock()
42+ defer v.mu.Unlock()
43+ v.v = t
44+}
internal/csync/value_test.gonew file+99
@@ -0,0 +1,99 @@
1+package csync
2+
3+import (
4+ "sync"
5+ "testing"
6+
7+ "github.com/stretchr/testify/require"
8+)
9+
10+func TestValue_GetSet(t *testing.T) {
11+ t.Parallel()
12+
13+ v := NewValue(42)
14+ require.Equal(t, 42, v.Get())
15+
16+ v.Set(100)
17+ require.Equal(t, 100, v.Get())
18+}
19+
20+func TestValue_ZeroValue(t *testing.T) {
21+ t.Parallel()
22+
23+ v := NewValue("")
24+ require.Equal(t, "", v.Get())
25+
26+ v.Set("hello")
27+ require.Equal(t, "hello", v.Get())
28+}
29+
30+func TestValue_Struct(t *testing.T) {
31+ t.Parallel()
32+
33+ type config struct {
34+ Name string
35+ Count int
36+ }
37+
38+ v := NewValue(config{Name: "test", Count: 1})
39+ require.Equal(t, config{Name: "test", Count: 1}, v.Get())
40+
41+ v.Set(config{Name: "updated", Count: 2})
42+ require.Equal(t, config{Name: "updated", Count: 2}, v.Get())
43+}
44+
45+func TestValue_PointerPanics(t *testing.T) {
46+ t.Parallel()
47+
48+ require.Panics(t, func() {
49+ NewValue(&struct{}{})
50+ })
51+}
52+
53+func TestValue_SlicePanics(t *testing.T) {
54+ t.Parallel()
55+
56+ require.Panics(t, func() {
57+ NewValue([]string{"a", "b"})
58+ })
59+}
60+
61+func TestValue_MapPanics(t *testing.T) {
62+ t.Parallel()
63+
64+ require.Panics(t, func() {
65+ NewValue(map[string]int{"a": 1})
66+ })
67+}
68+
69+func TestValue_ConcurrentAccess(t *testing.T) {
70+ t.Parallel()
71+
72+ v := NewValue(0)
73+ var wg sync.WaitGroup
74+
75+ // Concurrent writers.
76+ for i := range 100 {
77+ wg.Add(1)
78+ go func(val int) {
79+ defer wg.Done()
80+ v.Set(val)
81+ }(i)
82+ }
83+
84+ // Concurrent readers.
85+ for range 100 {
86+ wg.Add(1)
87+ go func() {
88+ defer wg.Done()
89+ _ = v.Get()
90+ }()
91+ }
92+
93+ wg.Wait()
94+
95+ // Value should be one of the set values (0-99).
96+ got := v.Get()
97+ require.GreaterOrEqual(t, got, 0)
98+ require.Less(t, got, 100)
99+}