Function exportBlocks

cmd/export_cmd.go:50–147

(ctx *cli.Context)

}

func exportBlocks(ctx *cli.Context) error {
	SetRpcPort(ctx)
	exportFile := ctx.String(utils.GetFlagName(utils.ExportFileFlag))
	if exportFile == "" {
		PrintErrorMsg("Missing %s argument.", utils.ExportFileFlag.Name)
		cli.ShowSubcommandHelp(ctx)
		return nil
	}

	startHeight := ctx.Uint(utils.GetFlagName(utils.ExportStartHeightFlag))
	endHeight := ctx.Uint(utils.GetFlagName(utils.ExportEndHeightFlag))
	if endHeight > 0 && startHeight > endHeight {
		return fmt.Errorf("export error: start height should smaller than end height")
	}
	blockCount, err := utils.GetBlockCount()
	if err != nil {
		return fmt.Errorf("GetBlockCount error:%s", err)
	}
	currentBlockHeight := uint(blockCount - 1)
	if startHeight > currentBlockHeight {
		PrintWarnMsg("StartBlockHeight:%d larger than CurrentBlockHeight:%d, No blocks to export.", startHeight, currentBlockHeight)
		return nil
	}
	if endHeight == 0 || endHeight > currentBlockHeight {
		endHeight = currentBlockHeight
	}

	speed := ctx.String(utils.GetFlagName(utils.ExportSpeedFlag))
	var sleepTime time.Duration
	switch speed {
	case "h":
		sleepTime = 0
	case "m":
		sleepTime = time.Millisecond * 2
	default:
		sleepTime = time.Millisecond * 5
	}

	exportFile = utils.GenExportBlocksFileName(exportFile, uint32(startHeight), uint32(endHeight))
	ef, err := os.OpenFile(exportFile, os.O_RDWR|os.O_CREATE, 0664)
	if err != nil {
		return fmt.Errorf("open file:%s error:%s", exportFile, err)
	}
	defer ef.Close()
	fWriter := bufio.NewWriter(ef)

	metadata := utils.NewExportBlockMetadata()
	metadata.StartBlockHeight = uint32(startHeight)
	metadata.EndBlockHeight = uint32(endHeight)
	err = metadata.Serialize(fWriter)
	if err != nil {
		return fmt.Errorf("write export metadata error:%s", err)
	}

	//progress bar
	uiprogress.Start()
	bar := uiprogress.AddBar(int(endHeight - startHeight + 1)).
		AppendCompleted().