main.go

package main

import (
	"encoding/json"
	"flag"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"runtime"
	"sync"
)

type SyncConfig struct { //json.Unmarshal struct must public var
	LocalDir  string //absolute path needed
	RemoteDir string

	SshHost     string
	SshPort     int
	SshUserName string
	SshPassword string

	IgnoreFiles []string
	IgnoreDirs  []string //relative path to LocalDir
	ReplaceRule map[string]string
}

var (
	g_WaitGroup   sync.WaitGroup
	g_SyncCfg     SyncConfig
	g_FileSyncer  *FileSyncer
	g_FileWatcher *FileWatcher
	g_ConfigFile  = "config.json"
)

func loadConfig() bool {
	flag.StringVar(&g_ConfigFile, "config", "config.json", "sync config file")
	flag.Parse()
	_, err := os.Stat(g_ConfigFile)
	if err != nil {
		log.Printf("Not Exist ConfigFile:%v\n", err)
		return false
	}
	configJson, err := ioutil.ReadFile(g_ConfigFile)
	if err != nil {
		log.Printf("ReadFile Error:%v\n", err)
		return false
	}
	err = json.Unmarshal(configJson, &g_SyncCfg)
	if err != nil {
		log.Printf("json.Unmarshal Error:%v\n", err)
		return false
	}

	if !filepath.IsAbs(g_SyncCfg.LocalDir) {
		log.Print("LocalDir must be Abs Path\n")
		return false
	}
	log.Printf("---load cfg: %v----\n", g_SyncCfg)

	return true
}

func init() {
	iCpuNum := runtime.NumCPU()
	runtime.GOMAXPROCS(iCpuNum)
}

func main() {
	ok := loadConfig()
	if !ok {
		os.Exit(-1)
	}

	g_FileSyncer = newFileSyncer()
	ok = g_FileSyncer.Connect()
	if !ok {
		os.Exit(-1)
	}
	g_FileSyncer.Disconnect()

	g_FileWatcher = newFileWatcher()
	ok = g_FileWatcher.Init()
	if !ok {
		os.Exit(-1)
	}

	g_WaitGroup.Add(3)
	go g_FileSyncer.Run()
	go g_FileWatcher.Run()
	go handleConsole()

	g_WaitGroup.Wait()

	log.Printf("Safe Exit!\n")
}