提交origin2.0版本

node/node.go

@@ -0,0 +1,115 @@
+package node
+
+import (
+	"fmt"
+	"github.com/duanhf2012/originnet/service"
+	"github.com/duanhf2012/originnet/cluster"
+	"io/ioutil"
+	"os"
+	"os/signal"
+	"strconv"
+	"syscall"
+	"time"
+)
+
+var closeSig chan bool
+var sigs chan os.Signal
+
+var preSetupService []service.IService //预安装
+
+func init() {
+	closeSig = make(chan bool,1)
+	sigs = make(chan os.Signal, 3)
+	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM,syscall.Signal(10))
+}
+
+
+func  getRunProcessPid() (int,error) {
+	f, err := os.OpenFile(os.Args[0]+".pid", os.O_RDONLY, 0600)
+	defer f.Close()
+	if err!= nil {
+		return 0,err
+	}
+
+	pidbyte,errs := ioutil.ReadAll(f)
+	if errs!=nil {
+		return 0,errs
+	}
+
+	return strconv.Atoi(string(pidbyte))
+}
+
+func writeProcessPid() {
+	//pid
+	f, err := os.OpenFile(os.Args[0]+".pid", os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0600)
+	defer f.Close()
+	if err != nil {
+		fmt.Println(err.Error())
+		os.Exit(-1)
+	} else {
+		_,err=f.Write([]byte(fmt.Sprintf("%d",os.Getpid())))
+		if err != nil {
+			fmt.Println(err.Error())
+			os.Exit(-1)
+		}
+	}
+}
+
+func GetNodeId() int {
+	return 1
+}
+
+func Init(){
+	//1.初始化集群
+	err := cluster.GetCluster().Init(GetNodeId())
+	if err != nil {
+		panic(err)
+	}
+
+	//2.service模块初始化
+	service.Init(closeSig)
+
+	//3.初始化预安装的服务
+	for _,s := range preSetupService {
+		pServiceCfg := cluster.GetCluster().GetServiceCfg(s.GetName())
+		s.Init(s,cluster.GetRpcClient,cluster.GetRpcServer,pServiceCfg)
+		//是否配置的service
+		if cluster.GetCluster().IsConfigService(s.GetName()) == false {
+			continue
+		}
+		service.Setup(s)
+	}
+}
+
+func Start() {
+	cluster.GetCluster().Start()
+	service.Start()
+	writeProcessPid()
+	for {
+		select {
+		case <-sigs:
+			fmt.Printf("Recv stop sig")
+			break
+		default:
+			time.Sleep(time.Second)
+		}
+	}
+
+	close(closeSig)
+	service.WaitStop()
+}
+
+
+func Setup(s ...service.IService)  {
+	for _,sv := range s {
+		preSetupService = append(preSetupService,sv)
+	}
+}
+
+func GetService(servicename string) service.IService {
+	return service.GetService(servicename)
+}
+
+func SetConfigDir(configdir string){
+	cluster.SetConfigDir(configdir)
+}