21 lines
534 B
Go
21 lines
534 B
Go
|
package extractor
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/moond4rk/hackbrowserdata/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
var extractorRegistry = make(map[types.DataType]func() Extractor)
|
||
|
|
|
||
|
|
// RegisterExtractor is used to register the data source
|
||
|
|
func RegisterExtractor(dataType types.DataType, factoryFunc func() Extractor) {
|
||
|
|
extractorRegistry[dataType] = factoryFunc
|
||
|
|
}
|
||
|
|
|
||
|
|
// CreateExtractor is used to create the data source
|
||
|
|
func CreateExtractor(dataType types.DataType) Extractor {
|
||
|
|
if factoryFunc, ok := extractorRegistry[dataType]; ok {
|
||
|
|
return factoryFunc()
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|