benjamyn/any-sync
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

IterateComponents: Add test and comment

Browse Source
This commit is contained in:
Sergey 2023-06-21 17:54:38 +05:00
parent fa178d7c26
commit 4eb2245669
No known key found for this signature in database
GPG Key ID: 3B6BEF79160221C6
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
app/app.go
View File

@ -262,6 +262,7 @@ func (app *App) Start(ctx context.Context) (err error) {
return return
} }
// IterateComponents iterates over all registered components. It's safe for concurrent use.
func (app *App) IterateComponents(fn func(Component)) { func (app *App) IterateComponents(fn func(Component)) {
app.mu.RLock() app.mu.RLock()
defer app.mu.RUnlock() defer app.mu.RUnlock()

15
app/app_test.go
View File

@ -55,6 +55,21 @@ func TestAppServiceRegistry(t *testing.T) {
}) })
} }
func TestApp_IterateComponents(t *testing.T) {
app := new(App)
app.Register(newTestService(testTypeRunnable, "c1", nil, nil))
app.Register(newTestService(testTypeRunnable, "r1", nil, nil))
app.Register(newTestService(testTypeComponent, "s1", nil, nil))
var got []string
app.IterateComponents(func(s Component) {
got = append(got, s.Name())
})
assert.ElementsMatch(t, []string{"c1", "r1", "s1"}, got)
}
func TestAppStart(t *testing.T) { func TestAppStart(t *testing.T) {
t.Run("SuccessStartStop", func(t *testing.T) { t.Run("SuccessStartStop", func(t *testing.T) {
app := new(App) app := new(App)