608f4aff0fc4f5a5bd6dd3a56afa0212fac3aa06
[sdc.git] /
1 /*
2  * Copyright © 2016-2017 European Support Limited
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.openecomp.core.model.types;
18
19 import com.datastax.driver.mapping.annotations.ClusteringColumn;
20 import com.datastax.driver.mapping.annotations.Column;
21 import com.datastax.driver.mapping.annotations.Frozen;
22 import com.datastax.driver.mapping.annotations.PartitionKey;
23 import com.datastax.driver.mapping.annotations.Table;
24 import com.google.common.io.ByteStreams;
25 import org.openecomp.sdc.common.errors.SdcRuntimeException;
26 import org.openecomp.sdc.datatypes.error.ErrorLevel;
27 import org.openecomp.sdc.logging.types.LoggerConstants;
28 import org.openecomp.sdc.logging.types.LoggerErrorCode;
29 import org.openecomp.sdc.logging.types.LoggerErrorDescription;
30 import org.openecomp.sdc.logging.types.LoggerTragetServiceName;
31 import org.openecomp.sdc.versioning.dao.types.Version;
32
33 import java.io.IOException;
34 import java.nio.ByteBuffer;
35
36 @Table(keyspace = "dox", name = "vsp_service_artifact")
37 public class ServiceArtifactEntity implements ServiceElementEntity {
38     private static final String ENTITY_TYPE;
39
40     static {
41         ENTITY_TYPE = "Vendor Software Product Service artifact";
42     }
43
44     @PartitionKey
45     @Column(name = "vsp_id")
46     public String id;
47
48     @PartitionKey(value = 1)
49     @Frozen
50     public Version version;
51
52     @ClusteringColumn
53     @Column(name = "name")
54     public String name;
55
56     @Column(name = "content_data")
57     public ByteBuffer contentData;
58
59     /**
60      * Every entity class must have a default constructor according to
61      * <a href="http://docs.datastax.com/en/developer/java-driver/2.1/manual/object_mapper/creating/">
62      * Definition of mapped classes</a>.
63      */
64     public ServiceArtifactEntity() {
65         // Don't delete! Default constructor is required by DataStax driver
66     }
67
68     /**
69      * Instantiates a new Service artifact entity.
70      *
71      * @param entity the entity
72      */
73     public ServiceArtifactEntity(ServiceArtifact entity) {
74         this.id = entity.getVspId();
75         this.version = entity.getVersion();
76         this.name = entity.getName();
77
78         try {
79             this.contentData = ByteBuffer.wrap(ByteStreams.toByteArray(entity.getContent()));
80         } catch (IOException ioException) {
81             throw new SdcRuntimeException(ioException);
82         }
83
84     }
85
86     @Override
87     public String getEntityType() {
88         return ENTITY_TYPE;
89     }
90
91     @Override
92     public String getFirstClassCitizenId() {
93         return getId();
94     }
95
96     @Override
97     public String getId() {
98         return id;
99     }
100
101     @Override
102     public void setId(String id) {
103         this.id = id;
104     }
105
106     @Override
107     public Version getVersion() {
108         return version;
109     }
110
111     @Override
112     public void setVersion(Version version) {
113         this.version = version;
114     }
115
116     @Override
117     public String getName() {
118         return name;
119     }
120
121     public void setName(String name) {
122         this.name = name;
123     }
124
125     @Override
126     public ByteBuffer getContentData() {
127         return contentData;
128     }
129
130     public void setContentData(ByteBuffer contentData) {
131         this.contentData = contentData;
132     }
133
134     public ServiceArtifact getServiceArtifact() {
135         ServiceArtifact serviceArtifact = new ServiceArtifact();
136         serviceArtifact.setName(this.getName());
137         serviceArtifact.setVersion(this.getVersion());
138         serviceArtifact.setContentData(this.getContentData().array());
139         serviceArtifact.setVspId(this.getId());
140         return serviceArtifact;
141     }
142
143 }