Method setExtractCreator

dsort/manager.go:449–494

setExtractCreator sets what type of file extraction and creation is used based on the RequestSpec.

()

// setExtractCreator sets what type of file extraction and creation is used based on the RequestSpec.
func (m *Manager) setExtractCreator() (err error) {
	var keyExtractor extract.KeyExtractor

	switch m.rs.Algorithm.Kind {
	case SortKindContent:
		keyExtractor, err = extract.NewContentKeyExtractor(m.rs.Algorithm.FormatType, m.rs.Algorithm.Extension)
	case SortKindMD5:
		keyExtractor, err = extract.NewMD5KeyExtractor()
	default:
		keyExtractor, err = extract.NewNameKeyExtractor()
	}

	if err != nil {
		return errors.WithStack(err)
	}

	onDuplicatedRecords := func(msg string) error {
		return m.react(m.rs.DuplicatedRecords, msg)
	}

	var extractCreator extract.Creator
	switch m.rs.Extension {
	case cos.ExtTar:
		extractCreator = extract.NewTarExtractCreator(m.ctx.t)
	case cos.ExtTarTgz, cos.ExtTgz:
		extractCreator = extract.NewTargzExtractCreator(m.ctx.t)
	case cos.ExtZip:
		extractCreator = extract.NewZipExtractCreator(m.ctx.t)
	default:
		cos.Assertf(false, "unknown extension %s", m.rs.Extension)
	}

	if !m.rs.DryRun {
		m.extractCreator = extractCreator
	} else {
		m.extractCreator = extract.NopExtractCreator(extractCreator)
	}

	m.recManager = extract.NewRecordManager(
		m.ctx.t, m.rs.Bck,
		m.rs.Extension, m.extractCreator,
		keyExtractor, onDuplicatedRecords,
	)

	return nil
}

// updateFinishedAck marks daemonID as finished. If all daemons ack then the
// finalCleanup is dispatched in separate goroutine.