-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.go
140 lines (116 loc) · 3.09 KB
/
run.go
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package main
import (
"flag"
"log"
"os"
"os/signal"
"syscall"
futil "github.com/contribsys/faktory/util"
"github.com/contribsys/sparq"
"github.com/contribsys/sparq/core"
"github.com/contribsys/sparq/db"
"github.com/contribsys/sparq/util"
"github.com/pressly/goose/v3"
)
func runExec(args []string) {
logPreamble()
opts := parseArguments(args)
util.InitLogger(opts.LogLevel)
futil.InitLogger(opts.LogLevel)
util.Debugf("Options: %v", opts)
_ = goose.SetDialect("sqlite3")
db.InstanceHostname = opts.Hostname
dbx, stopper, err := db.InitDB(opts.Hostname, false)
if err != nil {
util.Error("Unable to start Sparq", err)
return
}
defer stopper()
var ver string
err = dbx.QueryRow("select sqlite_version()").Scan(&ver)
if err != nil {
util.Error("Unable to query Sparq database", err)
return
}
util.Infof("Running sqlite %s", ver)
s, err := core.NewService(dbx, opts)
if err != nil {
util.Error("Unable to start Sparq", err)
return
}
go handleSignals(s)
// does not return until shutdown
err = s.Run()
if err != nil {
util.Error("Error running Sparq", err)
return
}
}
func parseArguments(args []string) core.Options {
flags := flag.NewFlagSet("run", flag.ExitOnError)
host := os.Getenv("SPARQ_HOSTNAME")
if host == "" {
host = "localhost.dev"
}
defaults := core.Options{
Binding: "localhost:9494",
Hostname: host,
LogLevel: "info",
ConfigDirectory: ".",
StorageDirectory: ".",
}
flags.Usage = runHelp
flags.StringVar(&defaults.Hostname, "h", "localhost.dev", "Instance hostname")
flags.StringVar(&defaults.Binding, "b", "localhost:9494", "Network binding")
flags.StringVar(&defaults.LogLevel, "l", "info", "Logging level (error, warn, info, debug)")
// undocumented on purpose, we don't want people changing these if possible
flags.StringVar(&defaults.StorageDirectory, "d", ".", "Storage directory")
flags.StringVar(&defaults.ConfigDirectory, "c", ".", "Config directory")
err := flags.Parse(args)
if err != nil {
log.Println(err)
os.Exit(0)
}
return defaults
}
func runHelp() {
log.Println(`
The "run" command starts the Sparq server.
Usage:
sparq run -h localhost.dev -b localhost:9494
sparq run -h example.social -b :8080 -l debug
Arguments:
-h [hostname]\tInstance hostname, default: localhost.dev
-b [binding]\tNetwork binding (use :9494 to listen on all interfaces), default: localhost:9494
-l [level]\tSet logging level (error, warn, info, debug), default: info`)
}
var (
Term os.Signal = syscall.SIGTERM
Hup os.Signal = syscall.SIGHUP
Info os.Signal = syscall.SIGTTIN
SignalHandlers = map[os.Signal]func(*core.Service){
Term: exit,
os.Interrupt: exit,
// Hup: reload,
Info: threadDump,
}
)
func handleSignals(s *core.Service) {
signals := make(chan os.Signal, 1)
for k := range SignalHandlers {
signal.Notify(signals, k)
}
for {
sig := <-signals
util.Debugf("Received signal: %v", sig)
funk := SignalHandlers[sig]
funk(s)
}
}
func exit(s *core.Service) {
util.Infof("%s shutting down", sparq.Name)
s.Close()
}
func threadDump(s *core.Service) {
util.DumpProcessTrace()
}