4498afc55b
Mutators are registered into a global variable, and then into the context when it is created. Only call registerMutators once so that the same mutator is not listed multiple times in the global variable when multiple contexts are created. Test: go test -v android/soong/android Change-Id: Ie9e3ed09a89b848462b898476cdfb81a90c64bd3
83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
// Copyright 2015 Google Inc. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package android
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/google/blueprint"
|
|
)
|
|
|
|
type moduleType struct {
|
|
name string
|
|
factory blueprint.ModuleFactory
|
|
}
|
|
|
|
var moduleTypes []moduleType
|
|
|
|
type singleton struct {
|
|
name string
|
|
factory blueprint.SingletonFactory
|
|
}
|
|
|
|
var singletons []singleton
|
|
|
|
type mutator struct {
|
|
name string
|
|
bottomUpMutator blueprint.BottomUpMutator
|
|
topDownMutator blueprint.TopDownMutator
|
|
parallel bool
|
|
}
|
|
|
|
var mutators []*mutator
|
|
|
|
func RegisterModuleType(name string, factory blueprint.ModuleFactory) {
|
|
moduleTypes = append(moduleTypes, moduleType{name, factory})
|
|
}
|
|
|
|
func RegisterSingletonType(name string, factory blueprint.SingletonFactory) {
|
|
singletons = append(singletons, singleton{name, factory})
|
|
}
|
|
|
|
var registerMutatorsOnce sync.Once
|
|
|
|
func NewContext() *blueprint.Context {
|
|
ctx := blueprint.NewContext()
|
|
|
|
for _, t := range moduleTypes {
|
|
ctx.RegisterModuleType(t.name, t.factory)
|
|
}
|
|
|
|
for _, t := range singletons {
|
|
ctx.RegisterSingletonType(t.name, t.factory)
|
|
}
|
|
|
|
registerMutatorsOnce.Do(registerMutators)
|
|
|
|
for _, t := range mutators {
|
|
var handle blueprint.MutatorHandle
|
|
if t.bottomUpMutator != nil {
|
|
handle = ctx.RegisterBottomUpMutator(t.name, t.bottomUpMutator)
|
|
} else if t.topDownMutator != nil {
|
|
handle = ctx.RegisterTopDownMutator(t.name, t.topDownMutator)
|
|
}
|
|
if t.parallel {
|
|
handle.Parallel()
|
|
}
|
|
}
|
|
|
|
return ctx
|
|
}
|