JCS-pub/client/internal/cmdline/serve.go

package cmdline

import (
	"fmt"
	"os"
	"time"

	"github.com/spf13/cobra"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	"gitlink.org.cn/cloudream/storage2/client/internal/accessstat"
	"gitlink.org.cn/cloudream/storage2/client/internal/config"
	"gitlink.org.cn/cloudream/storage2/client/internal/db"
	"gitlink.org.cn/cloudream/storage2/client/internal/downloader"
	"gitlink.org.cn/cloudream/storage2/client/internal/downloader/strategy"
	"gitlink.org.cn/cloudream/storage2/client/internal/http"
	"gitlink.org.cn/cloudream/storage2/client/internal/metacache"
	"gitlink.org.cn/cloudream/storage2/client/internal/mount"
	"gitlink.org.cn/cloudream/storage2/client/internal/services"
	"gitlink.org.cn/cloudream/storage2/client/internal/uploader"
	stgglb "gitlink.org.cn/cloudream/storage2/common/globals"
	"gitlink.org.cn/cloudream/storage2/common/models/datamap"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/connectivity"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/distlock"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/storage/pool"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/sysevent"
)

// 初始化函数，将ServeHTTP命令注册到命令列表中。
func init() {
	var configPath string
	var opt serveHTTPOptions
	cmd := cobra.Command{
		Use:   "serve",
		Short: "start serving storage service",
		Run: func(cmd *cobra.Command, args []string) {
			serveHTTP(configPath, opt)
		},
	}
	cmd.Flags().StringVarP(&configPath, "config", "c", "", "config file path")
	cmd.Flags().BoolVarP(&opt.DisableHTTP, "no-http", "", false, "disable http server")
	cmd.Flags().StringVarP(&opt.HTTPListenAddr, "listen", "", "", "http listen address, will override config file")
	cmd.Flags().BoolVarP(&opt.DisableMount, "no-mount", "", false, "disable mount")
	cmd.Flags().StringVarP(&opt.MountPoint, "mount", "", "", "mount point, will override config file")
	RootCmd.AddCommand(&cmd)
}

type serveHTTPOptions struct {
	DisableHTTP    bool
	HTTPListenAddr string
	DisableMount   bool
	MountPoint     string
}

func serveHTTP(configPath string, opts serveHTTPOptions) {
	err := config.Init(configPath)
	if err != nil {
		fmt.Printf("init config failed, err: %s", err.Error())
		os.Exit(1)
	}

	err = logger.Init(&config.Cfg().Logger)
	if err != nil {
		fmt.Printf("init logger failed, err: %s", err.Error())
		os.Exit(1)
	}

	stgglb.InitLocal(config.Cfg().Local)
	stgglb.InitMQPool(config.Cfg().RabbitMQ)
	stgglb.InitHubRPCPool(&config.Cfg().HubGRPC)

	// 数据库
	db, err := db.NewDB(&config.Cfg().DB)
	if err != nil {
		logger.Fatalf("new db failed, err: %s", err.Error())
	}

	// 初始化系统事件发布器
	evtPub, err := sysevent.NewPublisher(sysevent.ConfigFromMQConfig(config.Cfg().RabbitMQ), &datamap.SourceClient{
		UserID: config.Cfg().Local.UserID,
	})
	if err != nil {
		logger.Errorf("new sysevent publisher: %v", err)
		os.Exit(1)
	}
	evtPubChan := evtPub.Start()
	defer evtPub.Stop()

	// 连接性信息收集
	conCol := connectivity.NewCollector(&config.Cfg().Connectivity, nil)
	conCol.CollectInPlace()

	// 元数据缓存
	metaCacheHost := metacache.NewHost(db)
	go metaCacheHost.Serve()
	stgMeta := metaCacheHost.AddStorageMeta()
	hubMeta := metaCacheHost.AddHubMeta()
	conMeta := metaCacheHost.AddConnectivity()

	// 分布式锁
	distlockSvc, err := distlock.NewService(&config.Cfg().DistLock)
	if err != nil {
		logger.Warnf("new distlock service failed, err: %s", err.Error())
		os.Exit(1)
	}
	go serveDistLock(distlockSvc)

	// 访问统计
	acStat := accessstat.NewAccessStat(accessstat.Config{
		// TODO 考虑放到配置里
		ReportInterval: time.Second * 10,
	}, db)
	acStatChan := acStat.Start()
	defer acStat.Stop()

	// 存储管理器
	stgPool := pool.NewPool()

	// 下载策略
	strgSel := strategy.NewSelector(config.Cfg().DownloadStrategy, stgMeta, hubMeta, conMeta)

	// 下载器
	dlder := downloader.NewDownloader(config.Cfg().Downloader, &conCol, stgPool, strgSel, db)

	// 上传器
	uploader := uploader.NewUploader(distlockSvc, &conCol, stgPool, stgMeta, db)

	svc := services.NewService(distlockSvc, dlder, acStat, uploader, strgSel, stgMeta, db, evtPub, nil)

	// HTTP接口
	httpCfg := config.Cfg().HTTP
	if !opts.DisableHTTP && httpCfg != nil && httpCfg.Enabled {
		if opts.HTTPListenAddr != "" {
			httpCfg.Listen = opts.HTTPListenAddr
		}
	} else {
		httpCfg = nil
	}
	httpSvr := http.NewServer(httpCfg, svc)
	httpChan := httpSvr.Start()
	defer httpSvr.Stop()

	// 挂载
	mntCfg := config.Cfg().Mount
	if !opts.DisableMount && mntCfg != nil && mntCfg.Enabled {
		if opts.MountPoint != "" {
			mntCfg.MountPoint = opts.MountPoint
		}
	} else {
		mntCfg = nil
	}
	mnt := mount.NewMount(mntCfg, db, uploader, dlder)
	mntChan := mnt.Start()
	defer mnt.Stop()

	/// 开始监听各个模块的事件

	evtPubEvt := evtPubChan.Receive()
	acStatEvt := acStatChan.Receive()
	httpEvt := httpChan.Receive()
	mntEvt := mntChan.Receive()

loop:
	for {
		select {
		case e := <-evtPubEvt.Chan():
			if e.Err != nil {
				logger.Errorf("receive publisher event: %v", err)
				break loop
			}

			switch val := e.Value.(type) {
			case sysevent.PublishError:
				logger.Errorf("publishing event: %v", val)

			case sysevent.PublisherExited:
				if val.Err != nil {
					logger.Errorf("publisher exited with error: %v", val.Err)
				} else {
					logger.Info("publisher exited")
				}
				break loop

			case sysevent.OtherError:
				logger.Errorf("sysevent: %v", val)
			}
			evtPubEvt = evtPubChan.Receive()

		case e := <-acStatEvt.Chan():
			if e.Err != nil {
				logger.Errorf("receive access stat event: %v", err)
				break loop
			}
			switch e := e.Value.(type) {
			case accessstat.ExitEvent:
				logger.Infof("access stat exited, err: %v", e.Err)
				break loop
			}
			acStatEvt = acStatChan.Receive()

		case e := <-httpEvt.Chan():
			if e.Err != nil {
				logger.Errorf("receive http event: %v", err)
				break loop
			}

			switch e := e.Value.(type) {
			case http.ExitEvent:
				logger.Infof("http server exited, err: %v", e.Err)
				break loop
			}
			httpEvt = httpChan.Receive()

		case e := <-mntEvt.Chan():
			if e.Err != nil {
				logger.Errorf("receive mount event: %v", e.Err)
				break loop
			}

			switch e := e.Value.(type) {
			case mount.ExitEvent:
				logger.Infof("mount exited, err: %v", e.Err)
				break loop
			}
			mntEvt = mntChan.Receive()
		}
	}
}

func serveDistLock(svc *distlock.Service) {
	logger.Info("start serving distlock")

	err := svc.Serve()

	if err != nil {
		logger.Errorf("distlock stopped with error: %s", err.Error())
	}

	logger.Info("distlock stopped")

	// TODO 仅简单结束了程序
	os.Exit(1)
}