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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
/*
* 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
GetAllProjects() ([]Project, 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.Insert(v.storeName, key, nil, 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.Find(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[0], &proj)
if err != nil {
return Project{}, pkgerrors.Wrap(err, "Unmarshaling Value")
}
return proj, nil
}
return Project{}, pkgerrors.New("Error getting Project")
}
// GetAllProjects returns all the projects
func (v *ProjectClient) GetAllProjects() ([]Project, error) {
key := ProjectKey{
ProjectName: "",
}
var res []Project
values, err := db.DBconn.Find(v.storeName, key, v.tagMeta)
if err != nil {
}
for _, value := range values {
p := Project{}
err = db.DBconn.Unmarshal(value, &p)
if err != nil {
return []Project{}, pkgerrors.Wrap(err, "Unmarshaling Project")
}
res = append(res, p)
}
return res, nil
}
// 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.Remove(v.storeName, key)
if err != nil {
return pkgerrors.Wrap(err, "Delete Project Entry;")
}
//TODO: Delete the collection when the project is deleted
return nil
}
|
