1 /*
2  * Copyright 2009-2017 Alibaba Cloud All rights reserved.
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 
18 #include <alibabacloud/oss/model/PutObjectRequest.h>
19 #include <alibabacloud/oss/http/HttpType.h>
20 #include "utils/Utils.h"
21 #include "ModelError.h"
22 #include <sstream>
23 using namespace AlibabaCloud::OSS;
24 
PutObjectRequest(const std::string & bucket,const std::string & key,const std::shared_ptr<std::iostream> & content)25 PutObjectRequest::PutObjectRequest(const std::string &bucket, const std::string &key,
26     const std::shared_ptr<std::iostream> &content) :
27     OssObjectRequest(bucket, key),
28     content_(content)
29 {
30     setFlags(Flags() | REQUEST_FLAG_CHECK_CRC64);
31 }
32 
PutObjectRequest(const std::string & bucket,const std::string & key,const std::shared_ptr<std::iostream> & content,const ObjectMetaData & metaData)33 PutObjectRequest::PutObjectRequest(const std::string &bucket, const std::string &key,
34     const std::shared_ptr<std::iostream> &content, const ObjectMetaData &metaData) :
35     OssObjectRequest(bucket, key),
36     content_(content),
37     metaData_(metaData)
38 {
39     setFlags(Flags() | REQUEST_FLAG_CHECK_CRC64);
40 }
41 
setCacheControl(const std::string & value)42 void PutObjectRequest::setCacheControl(const std::string &value)
43 {
44     metaData_.addHeader(Http::CACHE_CONTROL, value);
45 }
46 
setContentDisposition(const std::string & value)47 void PutObjectRequest::setContentDisposition(const std::string &value)
48 {
49     metaData_.addHeader(Http::CONTENT_DISPOSITION, value);
50 }
51 
setContentEncoding(const std::string & value)52 void PutObjectRequest::setContentEncoding(const std::string &value)
53 {
54     metaData_.addHeader(Http::CONTENT_ENCODING, value);
55 }
56 
setContentMd5(const std::string & value)57 void PutObjectRequest::setContentMd5(const std::string &value)
58 {
59     metaData_.addHeader(Http::CONTENT_MD5, value);
60 }
61 
setExpires(const std::string & value)62 void PutObjectRequest::setExpires(const std::string &value)
63 {
64     metaData_.addHeader(Http::EXPIRES, value);
65 }
66 
setCallback(const std::string & callback,const std::string & callbackVar)67 void PutObjectRequest::setCallback(const std::string& callback, const std::string& callbackVar)
68 {
69     metaData_.removeHeader("x-oss-callback");
70     metaData_.removeHeader("x-oss-callback-var");
71 
72     if (!callback.empty()) {
73         metaData_.addHeader("x-oss-callback", callback);
74     }
75 
76     if (!callbackVar.empty()) {
77         metaData_.addHeader("x-oss-callback-var", callbackVar);
78     }
79 }
80 
setTagging(const std::string & value)81 void PutObjectRequest::setTagging(const std::string& value)
82 {
83     metaData_.addHeader("x-oss-tagging", value);
84 }
85 
setTrafficLimit(uint64_t value)86 void PutObjectRequest::setTrafficLimit(uint64_t value)
87 {
88     metaData_.addHeader("x-oss-traffic-limit", std::to_string(value));
89 }
90 
MetaData()91 ObjectMetaData &PutObjectRequest::MetaData()
92 {
93     return metaData_;
94 }
95 
Body() const96 std::shared_ptr<std::iostream> PutObjectRequest::Body() const
97 {
98     return content_;
99 }
100 
specialHeaders() const101 HeaderCollection PutObjectRequest::specialHeaders() const
102 {
103     auto headers = metaData_.toHeaderCollection();
104 
105     if (headers.find(Http::CONTENT_TYPE) == headers.end()) {
106         headers[Http::CONTENT_TYPE] = LookupMimeType(Key());
107     }
108 
109     auto baseHeaders = OssObjectRequest::specialHeaders();
110     headers.insert(baseHeaders.begin(), baseHeaders.end());
111 
112     return headers;
113 }
114 
validate() const115 int PutObjectRequest::validate() const
116 {
117     int ret = OssObjectRequest::validate();
118     if (ret != 0) {
119         return ret;
120     }
121 
122     if (content_ == nullptr) {
123         return ARG_ERROR_REQUEST_BODY_NULLPTR;
124     }
125 
126     if (content_->bad()) {
127         return ARG_ERROR_REQUEST_BODY_BAD_STATE;
128     }
129 
130     if (content_->fail()) {
131         return ARG_ERROR_REQUEST_BODY_FAIL_STATE;
132     }
133 
134     return 0;
135 }
136 
137