POC - Changes to use persistent recursive watch for PRS enabled collections

go.mod

@@ -3,3 +3,5 @@ module github.com/fullstorydev/gosolr
 go 1.13
 
 require github.com/fullstorydev/zk v1.0.3-0.20200828191825-edfcb5d63fdd
+
+replace github.com/fullstorydev/zk => github.com/patsonluk/zk v1.0.3-0.20230620222602-ecd88884e248

go.sum

@@ -1,2 +1,2 @@
-github.com/fullstorydev/zk v1.0.3-0.20200828191825-edfcb5d63fdd h1:0lubUEXk7T4Z4fJtam94F9IVGK/k/wFw7jjTMpX0Ke8=
-github.com/fullstorydev/zk v1.0.3-0.20200828191825-edfcb5d63fdd/go.mod h1:8fIOiu6/LYedHuzFUvyuLg0K2kFGSWGrEnsBytyZ9N4=
+github.com/patsonluk/zk v1.0.3-0.20230620222602-ecd88884e248 h1:m4pApDjJN6FgotGGoDIbPIPaMl4JYVmNm6S9cSXXtZ8=
+github.com/patsonluk/zk v1.0.3-0.20230620222602-ecd88884e248/go.mod h1:ssbnU3H1lBdE6eTeQ4pyY53X+kHUNqwongLRP8AYygo=

solrmonitor/main/solrmonitor/solrmonitor.go

@@ -237,3 +237,24 @@ func (s *sexPantherZkCli) State() zk.State {
 func (s *sexPantherZkCli) Close() {
 	s.delegate.Close()
 }
+
+func (s *sexPantherZkCli) Children(path string) ([]string, *zk.Stat, error) {
+	if s.isFlaky() && s.rnd.Float32() > flakeChance {
+		return nil, nil, errors.New("flaky error")
+	}
+	return s.delegate.Children(path)
+}
+
+func (s *sexPantherZkCli) AddPersistentWatch(path string, mode zk.AddWatchMode) (ch zk.EventQueue, err error) {
+	if s.isFlaky() && s.rnd.Float32() > flakeChance {
+		return nil, errors.New("flaky error")
+	}
+	return s.delegate.AddPersistentWatch(path, mode)
+}
+
+func (s *sexPantherZkCli) RemoveAllPersistentWatches(path string) (err error) {
+	if s.isFlaky() && s.rnd.Float32() > flakeChance {
+		return errors.New("flaky error")
+	}
+	return s.delegate.RemoveAllPersistentWatches(path)
+}

solrmonitor/mock.go

@@ -23,6 +23,19 @@ func newMockZkClient() ZkCli {
 type mockZkClient struct {
 }
 
+func (m mockZkClient) Children(path string) ([]string, *zk.Stat, error) {
+	//TODO implement me
+	panic("implement me")
+}
+
+func (m mockZkClient) AddPersistentWatch(path string, mode zk.AddWatchMode) (ch zk.EventQueue, err error) {
+	panic("implement me")
+}
+
+func (m mockZkClient) RemoveAllPersistentWatches(path string) (err error) {
+	panic("implement me")
+}
+
 func (m mockZkClient) ChildrenW(path string) ([]string, *zk.Stat, <-chan zk.Event, error) {
 	panic("Not implemented")
 }