crm/.svn/pristine/a8/a894dcb704e41b5e347879639a0ba431186c19fa.svn-base

package main

import (
	"./comm"
	"./mdns"
	"./models"
	"./pkg/logging"
	"./pkg/setting"
	"./pkg/util"
	"./routers"
	"fmt"
	"github.com/kardianos/service"
	"log"
	"net/http"
	"os"
	"sync"

	"strings"
	//"unsafe"
)

// @title 科湃腾Go语言后台API
// @version 1.1.2
// @description  kptyun_service

// @contact.name 朱红宾
// @contact.url https://www.dairyinfo.com.cn
// @contact.email kptzhu@163.com

// @host
// @BasePath /
func main() {

	setting.Setup("")

	models.Setup()
	logging.Setup()

	svcConfig := &service.Config{
		Name:        setting.ServerSetting.ServerName, //服务显示名称
		DisplayName:  setting.ServerSetting.DisplayName, //服务名称
		Description:  setting.ServerSetting.Description, //服务描述
	}
	prg := &program{}
	s, err := service.New(prg, svcConfig)
	if err != nil {
		fmt.Println(err, "service.New() err",err)
	}

	if len(os.Args) > 1 {
		if os.Args[1] == "install" {
			err = s.Install()
			if err != nil {
				fmt.Println("install err", err)
			} else {
				fmt.Println("install success")
			}
			return
		}
		if os.Args[1] == "remove" {
			err = s.Uninstall()
			if err != nil {
				fmt.Println("Uninstall err", err)
			} else {
				fmt.Println("Uninstall success")
			}
			return
		}
	}
	_ = s.Run()


}


type program struct{}
func (p *program) Start(s service.Service) error {
	go p.run()
	return nil
}

func (p *program) run() {
	// 代码写在这儿
	if setting.ServerSetting.MailStart == "1" {
		util.CronMail()
	}
	if setting.ServerSetting.WXworkStart == "1" {
		util.CronWXwork()
	}
	//util.InitWebS("田欣宇","xdmy@1234","http://60.174.151.11:6888")
	//println(setting.ServerSetting.EasName,setting.ServerSetting.EasPassword,setting.ServerSetting.EasUrl)


	//println(sessionid)
	util.InitWebS(setting.ServerSetting.EasName,setting.ServerSetting.EasPassword,setting.ServerSetting.EasUrl)
	//s,err :=util.EASWebServiceLogin()
	//logging.Info("eas账号",s,err)

	//sessionid,_ :=  util.EASWebServiceLogin()
	////err = util.EASWebServiceQureyParts(sessionid,"04.04.08.01.001227")
	//err = util.EASWebServiceQureyBill(sessionid,"CK08210604016-RKq","006")
	//if err != nil {
	//	fmt.Println("==========",err)
	//}
	//pu := util.BillLaid{}
	//pu.TransactionType.Number = "wrrwtwwwwwwww"
	//pujson,_ := json.Marshal(pu)
	//err = util.EASWebServicInsertPartsuselaid(s,string(pujson),"006")
	//if err != nil {
	//	println(err.Error())
	//}
	//util.EASWebServiceLogout()
	mu :=sync.Mutex{}
	mu.Lock()
	routersInit := routers.InitRouter()
	readTimeout := setting.ServerSetting.ReadTimeout
	writeTimeout := setting.ServerSetting.WriteTimeout
	endPoint := fmt.Sprintf(":%d", setting.ServerSetting.HttpPort)
	maxHeaderBytes := 1 << 20
	server := &http.Server{
		Addr:           endPoint,
		Handler:        routersInit,
		ReadTimeout:    readTimeout,
		WriteTimeout:   writeTimeout,
		MaxHeaderBytes: maxHeaderBytes,
	}

	if setting.CommSetting.PortName != "" {
		go comm.OpenComm()
	}

	if strings.Trim(setting.ServerSetting.Mdns_servicename, " ") != "" {
		go mdns.Mdns(strings.Trim(setting.ServerSetting.Mdns_servicename, " "), setting.ServerSetting.Mdns_serviceport)
	}

	httpsPort := fmt.Sprintf(":%d", setting.ServerSetting.HttpsPort)
	if setting.ServerSetting.Https == 1 {
		log.Printf("[info] start https server listening port %s", httpsPort)
		_ = routersInit.RunTLS(httpsPort, setting.ServerSetting.CrtPath, setting.ServerSetting.KeyPath)
	} else if setting.ServerSetting.Https == 2 {
		log.Printf("[info] start https & http  server listening port https %s, http %s", httpsPort, endPoint)
		go routersInit.RunTLS(httpsPort, setting.ServerSetting.CrtPath, setting.ServerSetting.KeyPath)
		_ = server.ListenAndServe()
	} else {
		log.Printf("[info] start http server listening port %s", endPoint)
		_ = server.ListenAndServe()
	}

}


func (p *program) Stop(s service.Service) error {
	fmt.Println("程序被关闭")
	return nil
}