mtsatellite/main.go
Sascha L. Teichmann 749c22e83a Added license: MIT
2014-08-03 15:59:56 +02:00

84 lines
1.6 KiB
Go

// 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.
package main
import (
"flag"
"fmt"
"log"
"net"
"os"
"os/signal"
)
func main() {
var port int
var host string
var driver string
var cacheSize int
flag.IntVar(&port, "port", 6379, "port to bind")
flag.StringVar(&driver, "driver", "leveldb", "type of database (leveldb, sqlite)")
flag.StringVar(&host, "host", "", "host to bind")
flag.IntVar(&cacheSize, "cache", 32, "cache size in MB")
flag.Parse()
args := flag.Args()
if len(args) < 1 {
log.Fatal("Missing path to world")
}
var err error
var backend Backend
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()
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())
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())
connChan <- conn
}
}()
for {
select {
case conn := <-connChan:
go NewConnection(conn, backend).Run()
case <-sigChan:
log.Println("Shutting down")
return
}
}
}