forked from elves/elvish
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
76 lines (67 loc) · 1.52 KB
/
main.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
// elvishd is an agent for sharing variables and command-line history among
// multiple elvish processes.
package main
import (
"database/sql"
"log"
"net"
"os"
"os/signal"
"os/user"
"path"
"syscall"
"github.com/coopernurse/gorp"
_ "github.com/mattn/go-sqlite3"
"github.com/xiaq/elvish/service"
"github.com/xiaq/elvish/util"
)
const (
SignalBufferSize = 32
)
func main() {
laddr, err := util.SocketName()
if err != nil {
log.Fatalln("get socket name:", err)
}
// Listen to socket
listener, err := net.Listen("unix", laddr)
if err != nil {
log.Fatalln("listen to socket:", err)
}
// Construct database filename
u, err := user.Current()
if err != nil {
log.Fatalln("get current user:", err)
}
home := u.HomeDir
if home == "" {
log.Fatalln("current user does not have a home directory")
}
dbname := path.Join(home, ".elvishd.db")
// Open database and construct dbmap
db, err := sql.Open("sqlite3", dbname)
if err != nil {
log.Fatalln("open database:", err)
}
dbmap := &gorp.DbMap{Db: db, Dialect: gorp.SqliteDialect{}}
// Set up Unix signal handler
sigch := make(chan os.Signal, SignalBufferSize)
signal.Notify(sigch)
go func() {
for sig := range sigch {
switch sig {
case syscall.SIGINT, syscall.SIGTERM:
// TODO(xiaq): Notify current clients of termination
os.Remove(laddr)
db.Close() // Ignore possible errors
os.Exit(0)
default:
// Ignore all other signals
}
}
}()
err = service.Serve(listener, dbmap)
if err != nil {
log.Fatalln("start service:", err)
}
}