change: update example llk

examples/game/llk/cmd/main.go

@@ -1,6 +1,10 @@
 package main
 
 import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
 	"time"
 
 	hrp "github.com/httprunner/httprunner/v5"
@@ -18,8 +22,36 @@ func main() {
 	}
 	defer bot.Close()
 
-	// err = bot.EnterGame(context.Background())
-	// require.NoError(t, err, "Failed to enter game")
+	err = bot.EnterGame(context.Background())
+	if err != nil {
+		log.Fatal().Err(err).Msg("Failed to enter game")
+	}
+	// Handle graceful shutdown and report generation
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// Create channel to handle OS signals
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+
+	// Start goroutine to handle signals
+	go func() {
+		<-sigChan
+		log.Info().Msg("Received shutdown signal, generating report...")
+		if err := bot.GenerateReport(); err != nil {
+			log.Error().Err(err).Msg("Failed to generate report")
+		}
+		cancel()
+	}()
+
+	// Start goroutine to handle context cancellation
+	go func() {
+		<-ctx.Done()
+		log.Info().Msg("Context cancelled, generating report...")
+		if err := bot.GenerateReport(); err != nil {
+			log.Error().Err(err).Msg("Failed to generate report")
+		}
+	}()
 
 	for {
 		err = bot.Play()

examples/game/llk/main.go

@@ -223,12 +223,17 @@ func (bot *LLKGameBot) Play() error {
 		)
 		if err != nil && !errors.Is(err, code.MaxRetryError) {
 			log.Error().Err(err).Msg("Failed to click game interface")
+			return err
 		}
 	}
 
 	return nil
 }
 
+func (bot *LLKGameBot) GenerateReport() error {
+	return bot.Session.GenerateReport()
+}
+
 // Close cleans up resources
 func (bot *LLKGameBot) Close() error {
 	if bot.DriverExt != nil {