aggressive initial ssgen load

traPtitech · Jun 6, 2023 · e781405 · e781405
1 parent d60200c
commit e781405
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pkg/usecase/ssgen/generator.go b/pkg/usecase/ssgen/generator.go
@@ -71,7 +71,15 @@ func (s *generatorService) Start(_ context.Context) error {
 	go retry.Do(ctx, func(ctx context.Context) error {
 		return s.client.ConnectSSGen(ctx, s.onRequest)
 	}, 1*time.Second, 60*time.Second)
-	go s.reload()
+	go func() {
+		for i := 0; i < 300; i++ {
+			s.reload()
+			if s.reloaded.Load() {
+				break
+			}
+			<-time.After(1 * time.Second)
+		}
+	}()
 
 	s.running.Store(true)
 	return nil