| // Licensed to the Apache Software Foundation (ASF) under one or more |
| // contributor license agreements. See the NOTICE file distributed with |
| // this work for additional information regarding copyright ownership. |
| // The ASF licenses this file to You 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 bufmodule |
| |
| import ( |
| "context" |
| |
| "github.com/apache/dubbo-kubernetes/pkg/bufman/bufpkg/bufmodule/bufmoduleref" |
| "github.com/apache/dubbo-kubernetes/pkg/bufman/pkg/storage" |
| ) |
| |
| type workspace struct { |
| // bufmoduleref.ModuleIdentity -> bufmodule.Module |
| namedModules map[string]Module |
| allModules []Module |
| } |
| |
| func newWorkspace( |
| ctx context.Context, |
| namedModules map[string]Module, |
| allModules []Module, |
| ) (*workspace, error) { |
| pathToExternalPaths := make(map[string][]string) |
| for _, module := range allModules { |
| fileInfos, err := module.SourceFileInfos(ctx) |
| if err != nil { |
| return nil, err |
| } |
| for _, fileInfo := range fileInfos { |
| pathToExternalPaths[fileInfo.Path()] = append(pathToExternalPaths[fileInfo.Path()], fileInfo.ExternalPath()) |
| } |
| } |
| for path, externalPaths := range pathToExternalPaths { |
| // Will be >1 even if the externalPaths are equal, we mostly care ab0out the count |
| if len(externalPaths) > 1 { |
| return nil, storage.NewErrExistsMultipleLocations(path, externalPaths...) |
| } |
| } |
| return &workspace{ |
| namedModules: namedModules, |
| allModules: allModules, |
| }, nil |
| } |
| |
| func (w *workspace) GetModule(moduleIdentity bufmoduleref.ModuleIdentity) (Module, bool) { |
| module, ok := w.namedModules[moduleIdentity.IdentityString()] |
| return module, ok |
| } |
| |
| func (w *workspace) GetModules() []Module { |
| return w.allModules |
| } |