File size: 3,467 Bytes
ca7217f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 | package engine
import (
"log"
"os"
"github.com/metatube-community/metatube-sdk-go/common/fetch"
mt "github.com/metatube-community/metatube-sdk-go/provider"
)
func (e *Engine) init() *Engine {
e.initLogger()
e.initFetcher()
e.initActorProviders()
e.initMovieProviders()
return e
}
func (e *Engine) initLogger() {
e.logger = log.New(os.Stdout, "[ENGINE]\u0020", log.LstdFlags|log.Llongfile)
}
func (e *Engine) initFetcher() {
e.fetcher = fetch.Default(&fetch.Config{Timeout: e.timeout})
}
// initActorProviders initializes actor providers.
func (e *Engine) initActorProviders() {
for name, factory := range mt.RangeActorFactory {
provider := factory()
if s, ok := provider.(mt.RequestTimeoutSetter); ok {
s.SetRequestTimeout(e.timeout /* global timeout */)
}
if config, hasConfig := e.actorProviderConfigs.Get(name); hasConfig {
e.applyProviderConfig("actor", provider, config)
}
if provider.Priority() <= 0 {
e.logger.Printf("Disable actor provider: %s", provider.Name())
continue
}
// Add actor provider by name.
e.actorProviders.Set(name, provider)
// Add actor provider by host.
host := provider.URL().Hostname()
e.actorHostProviders.Set(host,
append(e.actorHostProviders.
GetOrDefault(host, nil), provider))
}
}
// initMovieProviders initializes movie providers.
func (e *Engine) initMovieProviders() {
for name, factory := range mt.RangeMovieFactory {
provider := factory()
if s, ok := provider.(mt.RequestTimeoutSetter); ok {
s.SetRequestTimeout(e.timeout /* global timeout */)
}
if config, hasConfig := e.movieProviderConfigs.Get(name); hasConfig {
e.applyProviderConfig("movie", provider, config)
}
if provider.Priority() <= 0 {
e.logger.Printf("Disable movie provider: %s", provider.Name())
continue
}
// Add movie provider by name.
e.movieProviders.Set(name, provider)
// Add movie provider by host.
host := provider.URL().Hostname()
e.movieHostProviders.Set(host,
append(e.movieHostProviders.
GetOrDefault(host, nil), provider))
}
}
func (e *Engine) applyProviderConfig(providerType string, provider mt.Provider, config mt.Config) {
const (
proxyConfigKey = "proxy"
priorityConfigKey = "priority"
timeoutConfigKey = "timeout"
)
// Apply overridden priority.
if config.Has(priorityConfigKey) {
if v, err := config.GetFloat64(priorityConfigKey); err == nil {
e.logger.Printf("Override %s provider priority: %s=%.2f", providerType, provider.Name(), v)
provider.SetPriority(v)
}
}
// Apply overridden proxy.
if s, ok := provider.(mt.ProxySetter); ok && config.Has(proxyConfigKey) {
if v, err := config.GetString(proxyConfigKey); err == nil {
if err := s.SetProxy(v); err != nil {
e.logger.Fatalf("Set proxy for %s provider '%s' error: %v", providerType, provider.Name(), v)
}
e.logger.Printf("Override %s provider proxy: %s=%s", providerType, provider.Name(), v)
}
}
// Apply request timeout.
if s, ok := provider.(mt.RequestTimeoutSetter); ok && config.Has(timeoutConfigKey) {
if v, err := config.GetDuration(timeoutConfigKey); err == nil {
e.logger.Printf("Override %s provider request timeout: %s=%s", providerType, provider.Name(), v)
s.SetRequestTimeout(v)
}
}
// Apply full config.
if s, ok := provider.(mt.ConfigSetter); ok {
if err := s.SetConfig(config); err != nil {
e.logger.Fatalf("Set %s provider config for %s: %v", providerType, provider.Name(), err)
}
}
}
|