redisdog/src/cnphper.com/redisdog/main.go

package main

import (
	"database/sql"
	"encoding/json"
	"flag"
	"fmt"
	"html/template"
	"log"
	"net/http"
	"os"
	"os/signal"
	"strings"
	"syscall"

	_ "github.com/mattn/go-sqlite3"
)

type Config struct {
	Host         string
	Port         string
	Database     string
	RootPwd      string
	LogDir       string
	TmplDir      string
	ResourcesDir string
	SessionTTL   int64
}

var (
	Cfg         *Config
	Db          *sql.DB
	SessPoll    *SessionPoll
	TmplFuncMap template.FuncMap
	Srv         *http.Server
	Sender      *MailSender
	SysSetting  Setting
)

func loadConfig() *Config {
	var (
		cfgfile string
		cfg     *Config = &Config{}
	)
	flag.StringVar(&cfgfile, "c", "", "config file")
	flag.Parse()
	if cfgfile == "" {
		log.Fatalf("Usage: %s -c <config file>\n", os.Args[0])
	}

	file, err := os.OpenFile(cfgfile, os.O_RDONLY, 0644)
	if err != nil {
		log.Fatal(err)
	}
	decoder := json.NewDecoder(file)
	if err = decoder.Decode(cfg); err != nil {
		log.Fatal(err)
	}
	if cfg.Port == "" {
		log.Fatalf("Port is empty!\n")
	}
	if cfg.Database == "" {
		log.Fatalf("Database is empty!\n")
	}
	if cfg.RootPwd == "" {
		log.Fatalf("RootPwd is empty!\n")
	}
	if cfg.LogDir == "" {
		log.Fatalf("LogDir is empty!\n")
	}
	if cfg.TmplDir == "" {
		log.Fatalf("TmplDir is empty!\n")
	}
	if cfg.ResourcesDir == "" {
		log.Fatalf("ResourcesDir is empty!\n")
	}
	if cfg.SessionTTL == 0 {
		cfg.SessionTTL = 900
	}
	return cfg
}

func waitSignal() {
	ch := make(chan os.Signal, 1)
	signal.Notify(ch, syscall.SIGUSR1)
	signal.Notify(ch, syscall.SIGQUIT)
	for {
		sig := <-ch
		switch sig {
		//重启
		case syscall.SIGUSR1:
			close(ch)
			if Srv != nil {
				Srv.Close()
			}
			syscall.Exec(os.Args[0], os.Args[1:], nil)
		//退出
		case syscall.SIGQUIT:
			close(ch)
			if Srv != nil {
				Srv.Close()
			}
			os.Exit(0)
		}
	}
}

func openDatabase(dbfile string) *sql.DB {
	db, err := sql.Open("sqlite3", dbfile)
	if err != nil {
		log.Fatalf("sql.Open() Error: %s\n", err.Error())
	}

	db.Exec(Sql_Table_Syscfg)
	db.Exec(Sql_Table_Accounts)
	db.Exec(Sql_Table_Rediscfg)
	db.Exec(Sql_Table_Syslog)
	db.Exec(Sql_Index_Syslog)
	db.Exec(Sql_Table_Statuslog)
	db.Exec(Sql_Table_Warnlog)
	db.Exec(Sql_Table_Processlog)
	db.Exec(Sql_Values_Accounts)

	return db
}

func loadSystemSetting() map[string]string {

}

func start_http_server() *http.Server {
	handle := http.NewServeMux()
	//登录
	handle.HandleFunc("/login", login_index)
	//首页
	handle.HandleFunc("/", index_default)
	handle.HandleFunc("/index/stats", index_stats)
	handle.HandleFunc("/index/info", index_info)
	//用户操作
	handle.HandleFunc("/profile/logout", profile_logout)
	handle.HandleFunc("/profile/passwd", profile_passwd)
	//配置-redis
	handle.HandleFunc("/syscfg/redis", syscfg_redis)
	handle.HandleFunc("/syscfg/redis_list", syscfg_redis_list)
	handle.HandleFunc("/syscfg/redis_get", syscfg_redis_get)
	handle.HandleFunc("/syscfg/redis_set", syscfg_redis_set)
	handle.HandleFunc("/syscfg/redis_del", syscfg_redis_del)
	//配置-报警
	handle.HandleFunc("/syscfg/warn", syscfg_warn)
	handle.HandleFunc("/syscfg/warn_get", syscfg_warn_get)
	handle.HandleFunc("/syscfg/warn_set", syscfg_warn_set)
	//配置-账号
	handle.HandleFunc("/syscfg/account", syscfg_account)
	handle.HandleFunc("/syscfg/account_list", syscfg_account_list)
	handle.HandleFunc("/syscfg/account_get", syscfg_account_get)
	handle.HandleFunc("/syscfg/account_set", syscfg_account_set)
	handle.HandleFunc("/syscfg/account_del", syscfg_account_del)
	handle.HandleFunc("/syscfg/account_reset_pwd", syscfg_account_reset_pwd)
	//配置-杂项
	handle.HandleFunc("/syscfg/misc", syscfg_misc)
	handle.HandleFunc("/syscfg/misc_get", syscfg_misc_get)
	handle.HandleFunc("/syscfg/misc_set", syscfg_misc_set)
	//日志-报警
	handle.HandleFunc("/log/warn", log_warn)
	handle.HandleFunc("/log/warn_list", log_warn_list)
	//日志-扩容
	handle.HandleFunc("/log/autoprocess", log_autoprocess)
	handle.HandleFunc("/log/autoprocess_list", log_autoprocess_list)
	//日志-系统错误
	handle.HandleFunc("/log/syslog", log_syslog)
	handle.HandleFunc("/log/syslog_list", log_syslog_list)
	//日志-后台操作
	handle.HandleFunc("/log/account", log_account)
	handle.HandleFunc("/log/account_list", log_account_list)
	//资源文件
	handle.Handle("/resources/", http.StripPrefix("/resources/", http.FileServer(http.Dir(Cfg.ResourcesDir))))
	//启动HTTP服务
	srv := &http.Server{
		Addr:    fmt.Sprintf("%s:%s", Cfg.Host, Cfg.Port),
		Handler: handle,
	}
	go func(s *http.Server) {
		if err := s.ListenAndServe(); err != nil {
			log.Fatal(err)
		}
	}(srv)
	return srv
}

func main() {
	//加载基础配置
	Cfg = loadConfig()

	//连接DB
	Db = openDatabase(Cfg.Database)
	defer Db.Close()

	//创建HTTP会话池
	SessPoll = NewSessionPoll(Cfg.SessionTTL)

	//添加HTML模板函数
	TmplFuncMap = make(template.FuncMap)
	TmplFuncMap["has_prefix"] = strings.HasPrefix

	//从DB加载业务配置
	SysSetting = NewSetting()
	SysSetting.Load()

	//开始HTTP服务
	Srv = start_http_server()

	//开始监控Redis状态
	go monitor_loop()

	//开始处理邮件发送队列
	Sender = NewMailSender()
	go Sender.Loop()

	//监听信号
	waitSignal()
}