50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
|
// options
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"path/filepath"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Options struct {
|
||
|
|
RedisTTL time.Duration
|
||
|
|
CurrentPath string
|
||
|
|
Exe string
|
||
|
|
LogFile string
|
||
|
|
Timeout bool
|
||
|
|
Debug bool
|
||
|
|
Version bool
|
||
|
|
BufferSize int
|
||
|
|
}
|
||
|
|
|
||
|
|
func usage() {
|
||
|
|
fmt.Println("Usage: llmongo -m <mongo uri> -r <redis uri> -t <ttl> -l <logfile> -b <buffer size> -T -D -v\n")
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
var err error
|
||
|
|
opts.CurrentPath, err = filepath.Abs(filepath.Dir(os.Args[0]))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
opts.LogFile = path.Join(opts.CurrentPath, opts.LogFile)
|
||
|
|
pid.PIDFile = path.Join(opts.CurrentPath, "run", path.Base(os.Args[0])+".pid")
|
||
|
|
opts.Exe = path.Base(os.Args[0])
|
||
|
|
|
||
|
|
flag.StringVar(&dbs.MongoUri, "m", dbs.MongoUri, "Mongodb")
|
||
|
|
flag.StringVar(&dbs.RedisUri, "r", dbs.RedisUri, "Redis")
|
||
|
|
flag.StringVar(&opts.LogFile, "l", opts.LogFile, "Logs filename")
|
||
|
|
flag.DurationVar(&opts.RedisTTL, "t", opts.RedisTTL, "Redis TTL")
|
||
|
|
flag.BoolVar(&opts.Version, "v", false, "Version")
|
||
|
|
flag.BoolVar(&opts.Timeout, "T", false, "Timeout")
|
||
|
|
flag.BoolVar(&opts.Debug, "D", false, "Debug")
|
||
|
|
flag.IntVar(&opts.BufferSize, "b", 1, "Buffer size")
|
||
|
|
}
|