main.go

package main

import (
    "code.google.com/p/gcfg"
    "flag"
    "fmt"
    "log"
    "math/rand"
    "os"
    "os/signal"
    "runtime"
    "syscall"
    "time"
)

type RockdisConfig struct {
    Server struct {
        Bind       string
        Port       int
        MonitorLog bool
    }
    Database struct {
        DbDir           string
        MaxMemory       string
        BlockSize       string
        CreateIfMissing bool
        BloomFilter     int
        Compression     string
        CompactionStyle string
        MaxOpenFiles    int
        MaxMerge        int
    }
}

func main() {
    var confName string
    flag.StringVar(&confName, "conf", "rockdis.conf", "Rockdis Configuration file")
    flag.Parse()

    var config RockdisConfig
    err := gcfg.ReadFileInto(&config, confName)
    if err != nil {
        log.Fatal(err)
    }
    globalStat.configFile = confName
    globalStat.config = config
    globalStat.startTime = time.Now()
    globalStat.qpsCommands = 0
    globalStat.qpsStart = AtomicInt(time.Now().Unix())
    go func(qpsInterval int) {
        tc := time.Tick(time.Minute * time.Duration(qpsInterval))
        for _ = range tc {
            globalStat.qpsCommands.Set(0)
            globalStat.qpsStart.Set(time.Now().Unix())
        }
    }(15)

    rock := NewRocksDBHandler(config)
    server := NewServer(config)
    defer func() {
        rock.Close()
        server.Close()
    }()

    signalChan := make(chan os.Signal, 1)
    signal.Notify(signalChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT, syscall.SIGABRT)
    go func() {
        s := <-signalChan
        log.Printf("[Main] Captured the signal %v", s)
        rock.Close()
        server.Close()
        os.Exit(0)
    }()

    if err := server.RegisterHandler(rock); err != nil {
        log.Fatalf("[Main] Register Handler error, %s", err)
    }
    if err := server.ListenAndServe(); err != nil {
        log.Fatalf("[Main] ListenAndServe error, %s", err)
    }
}

type Stat struct {
    version          string
    configFile       string
    config           RockdisConfig
    startTime        time.Time
    clients          AtomicInt
    totalConnections AtomicInt
    totalCommands    AtomicInt
    keyHits          AtomicInt
    keyMisses        AtomicInt
    qpsCommands      AtomicInt
    qpsStart         AtomicInt
}

var globalStat *Stat

func init() {
    globalStat = &Stat{version: "0.0.1"}
    log.SetFlags(log.LstdFlags | log.Lshortfile)
    runtime.GOMAXPROCS(runtime.NumCPU())
    rand.Seed(time.Now().UnixNano())
}

var _ = fmt.Println