main.go

// Copyright (c) 2022 Institute of Software, Chinese Academy of Sciences (ISCAS)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

/*
 * @Author: wenchao
 * @Date: 2021-10-16 10:23:56
 * @LastEditors: jeffery
 * @LastEditTime: 2022-06-06 11:28:46
 * @Description:
 */
package main

import (
	"agent/biz/alivechecker"
	"agent/biz/disk_space_monitor/log_dir_monitor"
	"agent/biz/docker"
	"agent/config"
	"strings"

	"agent/biz/model/clientinfo"
	"agent/biz/model/device"
	"agent/biz/model/did/leveldb"
	"agent/biz/service/platform"
	"agent/biz/service/upgrade"
	"agent/biz/web"
	"agent/utils/logger"
	"os"
	"os/signal"
	"syscall"

	serviceswithplatform "agent/biz/service/switch-platform"
	// _ "net/http/pprof"
)

func main() {

	logger.SetLogPath(config.Config.Log.Path)
	logger.SetLogConfig(int(config.Config.Log.RotationSize),
		int(config.Config.Log.RotationCount),
		int(config.Config.Log.MaxAge), false)
	logger.PrecreateAllLoggers()

	config.Version = Version
	config.VersionNumber = VersionNumber

	logger.AppLogger().Infof("================[%v Started] [system-agent version:%v]================",
		os.Args[0], config.Version+"-"+config.VersionNumber)
	logger.AppLogger().Infof("startup config: AoLogDirBase=%v, singleDockerModeEnv=%v, platformEnabled=%v, dockerManage=%v",
		config.Config.Log.AoLogDirBase,
		os.Getenv(config.Config.Box.RunInDocker.AoSpaceSingleDockerModeEnv),
		config.Config.PlatformEnabled,
		config.Config.EnableDockerManage)

	if err := AgentCmd.Execute(); err != nil {
		logger.AppLogger().Errorf("agent command execute failed: %v", err)
		os.Exit(1)
	}
	logger.AppLogger().Infof("initializing device identity and keys")
	device.InitDeviceInfo()
	device.InitDeviceKey()
	clientinfo.InitClientInfo()
	logger.AppLogger().Infof("device identity initialized")

	if !strings.EqualFold(os.Getenv(config.Config.Box.RunInDocker.AoSpaceSingleDockerModeEnv), "true") && config.Config.PlatformEnabled {
		go platform.InitPlatformAbility()
		serviceswithplatform.RetryUnfinishedStatus()
		upgrade.CronForUpgrade()
	}

	// 启动 web/http api 服务
	logger.AppLogger().Infof("starting web api services")
	web.Start()
	logger.AppLogger().Infof("web api services started")

	// 启动 docker 微服务创建或启动
	if config.Config.EnableDockerManage {
		logger.AppLogger().Infof("starting docker services management")
		if strings.EqualFold(os.Getenv(config.Config.Box.RunInDocker.AoSpaceSingleDockerModeEnv), "true") {
			docker.MigrateFileStorageData()
		} else {
			docker.Start()
		}
		logger.AppLogger().Infof("docker services management initialized")
	} else {
		logger.AppLogger().Infof("Docker management disabled by config")
	}
	alivechecker.Start()
	logger.AppLogger().Infof("alivechecker started")

	// 检测是否需要发送升级推送
	if !strings.EqualFold(os.Getenv(config.Config.Box.RunInDocker.AoSpaceSingleDockerModeEnv), "true") && config.Config.PlatformEnabled {
		go upgrade.CheckUpgradeSucc()
	}

	// 日志目录监控
	log_dir_monitor.Start()
	logger.AppLogger().Infof("log directory monitor started")

	if err := leveldb.OpenDB(); err != nil {
		logger.LevelDBLogger().Errorf("failed leveldb.OpenDB: %v", err)
		os.Exit(2)
	}
	logger.LevelDBLogger().Infof("leveldb opened successfully")

	quitChan := make(chan os.Signal)
	signal.Notify(quitChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM,
		syscall.SIGQUIT, syscall.SIGUSR1, syscall.SIGUSR2)

	for s := range quitChan {
		switch s {
		case syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT:
			logger.AppLogger().Infof("receive signal: %v", s)
			GracefullExit()
		case syscall.SIGUSR1:
			// fmt.Println("usr1 signal", s)
		case syscall.SIGUSR2:
			// fmt.Println("usr2 signal", s)
		default:
			// fmt.Println("other signal", s)
		}
	}

}

func GracefullExit() {
	os.Exit(0)
}