mtsatellite/main.go

122 lines
2.4 KiB
Go
Raw Normal View History

2014-08-03 15:59:56 +02:00
// Copyright 2014 by Sascha L. Teichmann
// Use of this source code is governed by the MIT license
// that can be found in the LICENSE file.
2014-08-03 00:01:14 +02:00
package main
import (
"flag"
"fmt"
"log"
"net"
"os"
"os/signal"
"runtime"
"time"
2014-08-03 00:01:14 +02:00
)
const (
2014-08-11 16:19:08 +02:00
Version = "0.3"
GCDuration = "24h"
)
2014-08-06 10:04:24 +02:00
2014-08-03 00:01:14 +02:00
func main() {
var (
port int
host string
driver string
cacheSize int
version bool
gcDuration string
)
2014-08-03 00:01:14 +02:00
flag.IntVar(&port, "port", 6379, "port to bind")
2014-08-03 14:52:24 +02:00
flag.StringVar(&driver, "driver", "leveldb", "type of database (leveldb, sqlite)")
2014-08-03 00:01:14 +02:00
flag.StringVar(&host, "host", "", "host to bind")
flag.IntVar(&cacheSize, "cache", 32, "cache size in MB")
2014-08-06 10:04:24 +02:00
flag.BoolVar(&version, "version", false, "Print version and exit.")
flag.StringVar(&gcDuration, "gc-duration", GCDuration, "Duration between forced GCs.")
2014-08-03 00:01:14 +02:00
flag.Parse()
2014-08-06 10:04:24 +02:00
if version {
fmt.Printf("Version: %s\n", Version)
os.Exit(0)
}
2014-08-03 00:01:14 +02:00
args := flag.Args()
if len(args) < 1 {
log.Fatal("Missing path to world")
}
var (
err error
backend Backend
gcDur time.Duration
)
if gcDur, err = time.ParseDuration(gcDuration); err != nil {
log.Fatal(err)
}
2014-08-03 00:01:14 +02:00
2014-08-03 15:42:12 +02:00
if driver == "sqlite" {
if backend, err = NewSqliteBackend(args[0]); err != nil {
log.Fatal(err)
}
} else {
if backend, err = NewLeveDBBackend(args[0], cacheSize); err != nil {
log.Fatal(err)
}
}
defer backend.Shutdown()
2014-08-03 00:01:14 +02:00
var listener net.Listener
listener, err = net.Listen("tcp", fmt.Sprintf("%s:%d", host, port))
if err != nil {
log.Fatal(err)
}
defer listener.Close()
log.Printf("Server started at %s", listener.Addr())
2014-08-03 00:01:14 +02:00
connChan := make(chan net.Conn)
defer close(connChan)
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, os.Interrupt, os.Kill)
go func() {
for {
conn, err := listener.Accept()
if err != nil {
log.Fatal(err)
}
log.Printf("Client accepted from: %s", conn.RemoteAddr())
2014-08-03 00:01:14 +02:00
connChan <- conn
}
}()
log.Printf("Doing garbage collection every: %s", gcDur)
gcChan := time.Tick(gcDur)
2014-08-03 00:01:14 +02:00
for {
select {
case conn := <-connChan:
var session Session
if session, err = backend.NewSession(); err != nil {
log.Printf("Cannot create session: %s", err)
conn.Close()
} else {
go NewConnection(conn, session).Run()
}
2014-08-03 00:01:14 +02:00
case <-sigChan:
log.Println("Shutting down")
return
case <-gcChan:
log.Println("Starting garbage collection.")
runtime.GC()
log.Println("Garbage collection done.")
2014-08-03 00:01:14 +02:00
}
}
}