JCS-pub/main.go

package main

import (
	"fmt"
	"os"

	_ "google.golang.org/grpc/balancer/grpclb"

	distlocksvc "gitlink.org.cn/cloudream/common/pkg/distlock/service"
	log "gitlink.org.cn/cloudream/common/pkg/logger"
	"gitlink.org.cn/cloudream/common/utils/ipfs"
	"gitlink.org.cn/cloudream/storage-client/internal/cmdline"
	"gitlink.org.cn/cloudream/storage-client/internal/config"
	"gitlink.org.cn/cloudream/storage-client/internal/services"
	"gitlink.org.cn/cloudream/storage-client/internal/task"
	coorcli "gitlink.org.cn/cloudream/storage-common/pkgs/mq/client/coordinator"
	sccli "gitlink.org.cn/cloudream/storage-common/pkgs/mq/client/scanner"
)

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

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

	coorClient, err := coorcli.NewClient(&config.Cfg().RabbitMQ)
	if err != nil {
		log.Warnf("new coordinator client failed, err: %s", err.Error())
		os.Exit(1)
	}

	scanner, err := sccli.NewClient(&config.Cfg().RabbitMQ)
	if err != nil {
		log.Warnf("new scanner client failed, err: %s", err.Error())
		os.Exit(1)
	}

	var ipfsCli *ipfs.IPFS
	if config.Cfg().IPFS != nil {
		log.Infof("IPFS config is not empty, so create a ipfs client")

		ipfsCli, err = ipfs.NewIPFS(config.Cfg().IPFS)
		if err != nil {
			log.Warnf("new ipfs client failed, err: %s", err.Error())
			os.Exit(1)
		}
	}

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

	taskMgr := task.NewManager(ipfsCli, distlockSvc, coorClient)

	svc, err := services.NewService(coorClient, ipfsCli, scanner, distlockSvc, &taskMgr)
	if err != nil {
		log.Warnf("new services failed, err: %s", err.Error())
		os.Exit(1)
	}

	cmds, err := cmdline.NewCommandline(svc, distlockSvc, ipfsCli)
	if err != nil {
		log.Warnf("new command line failed, err: %s", err.Error())
		os.Exit(1)
	}

	cmds.DispatchCommand(os.Args[1:])
	/*
		TO DO future:
		1. ls命令，显示用户指定桶下的所有对象，及相关的元数据
		2. rm命令，用户指定bucket和object名，执行删除操作
		3. update命令，用户发起对象更新命令，查询元数据，判断对象的冗余方式，删除旧对象（unpin所有的副本或编码块），写入新对象
		4. ipfsStat命令，查看本地有无ipfsdaemon，ipfs目录的使用率
		5. ipfsFlush命令，unpin本地ipfs目录中的所有cid(block)
		6. 改为交互式client，输入用户名及秘钥后进入交互界面
		7. 支持纯缓存类型的IPFS节点，数据一律存在后端存储服务中
	*/
}

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

	err := svc.Serve()

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

	log.Info("distlock stopped")
}