blob: 594da1de6113791dcffec6194bcb8879c19f3dbb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
import re
import sys
with open('/home/kboz/eve_haul/backend-go/cmd/eve_haul_go/main.go', 'r') as f:
text = f.read()
old_server_logic = r'''log\.Println\("Listening on :8000"\)
if err := server\.ListenAndServe\(\); err != nil \{
log\.Fatalf\("Server failed: %v", err\)
\}'''
new_server_logic = '''go func() {
\tlog.Println("Listening on :8000")
\tif err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
\t\tlog.Fatalf("Server failed: %v", err)
\t}
}()
quit := make(chan os.Signal, 1)
signal.Notify(quit, os.Interrupt, syscall.SIGTERM)
<-quit
log.Println("\\nShutting down server gracefully...")
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := server.Shutdown(ctx); err != nil {
\tlog.Fatalf("Server forced to shutdown: %v", err)
}
log.Println("Server exited")'''
text = re.sub(old_server_logic, new_server_logic, text)
with open('/home/kboz/eve_haul/backend-go/cmd/eve_haul_go/main.go', 'w') as f:
f.write(text)
|
