/* * Copyright 2020 Intel Corporation, Inc * * 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 module import ( "encoding/json" "github.com/onap/multicloud-k8s/src/orchestrator/pkg/infra/db" pkgerrors "github.com/pkg/errors" ) // Project contains the metaData for Projects type Project struct { MetaData ProjectMetaData `json:"metadata"` } // ProjectMetaData contains the parameters for creating a project type ProjectMetaData struct { Name string `json:"name"` Description string `json:"description"` UserData1 string `userData1:"userData1"` UserData2 string `userData2:"userData2"` } // ProjectKey is the key structure that is used in the database type ProjectKey struct { ProjectName string `json:"project"` } // We will use json marshalling to convert to string to // preserve the underlying structure. func (pk ProjectKey) String() string { out, err := json.Marshal(pk) if err != nil { return "" } return string(out) } // ProjectManager is an interface exposes the Project functionality type ProjectManager interface { CreateProject(pr Project) (Project, error) GetProject(name string) (Project, error) DeleteProject(name string) error } // ProjectClient implements the ProjectManager // It will also be used to maintain some localized state type ProjectClient struct { storeName string tagMeta, tagContent string } // NewProjectClient returns an instance of the ProjectClient // which implements the ProjectManager func NewProjectClient() *ProjectClient { return &ProjectClient{ storeName: "orchestrator", tagMeta: "projectmetadata", } } // CreateProject a new collection based on the project func (v *ProjectClient) CreateProject(p Project) (Project, error) { //Construct the composite key to select the entry key := ProjectKey{ ProjectName: p.MetaData.Name, } //Check if this Project already exists _, err := v.GetProject(p.MetaData.Name) if err == nil { return Project{}, pkgerrors.New("Project already exists") } err = db.DBconn.Create(v.storeName, key, v.tagMeta, p) if err != nil { return Project{}, pkgerrors.Wrap(err, "Creating DB Entry") } return p, nil } // GetProject returns the Project for corresponding name func (v *ProjectClient) GetProject(name string) (Project, error) { //Construct the composite key to select the entry key := ProjectKey{ ProjectName: name, } value, err := db.DBconn.Read(v.storeName, key, v.tagMeta) if err != nil { return Project{}, pkgerrors.Wrap(err, "Get Project") } //value is a byte array if value != nil { proj := Project{} err = db.DBconn.Unmarshal(value, &proj) if err != nil { return Project{}, pkgerrors.Wrap(err, "Unmarshaling Value") } return proj, nil } return Project{}, pkgerrors.New("Error getting Project") } // DeleteProject the Project from database func (v *ProjectClient) DeleteProject(name string) error { //Construct the composite key to select the entry key := ProjectKey{ ProjectName: name, } err := db.DBconn.Delete(v.storeName, key, v.tagMeta) if err != nil { return pkgerrors.Wrap(err, "Delete Project Entry;") } //TODO: Delete the collection when the project is deleted return nil }