| /* |
| * 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. |
| */ |
| |
| #include "SqLiteImpl.hpp" |
| |
| namespace { |
| std::string g_default_persistence_directory = "GeodeRegionData"; |
| } // namespace |
| |
| using namespace apache::geode::client; |
| |
| void SqLiteImpl::init(const RegionPtr& region, PropertiesPtr& diskProperties) { |
| // Set the default values |
| |
| int maxPageCount = 0; |
| int pageSize = 0; |
| m_persistanceDir = g_default_persistence_directory; |
| std::string regionName = region->getName(); |
| if (diskProperties != NULLPTR) { |
| CacheableStringPtr maxPageCountPtr = diskProperties->find(MAX_PAGE_COUNT); |
| CacheableStringPtr pageSizePtr = diskProperties->find(PAGE_SIZE); |
| CacheableStringPtr persDir = diskProperties->find(PERSISTENCE_DIR); |
| |
| if (maxPageCountPtr != NULLPTR) { |
| maxPageCount = atoi(maxPageCountPtr->asChar()); |
| } |
| |
| if (pageSizePtr != NULLPTR) pageSize = atoi(pageSizePtr->asChar()); |
| |
| if (persDir != NULLPTR) m_persistanceDir = persDir->asChar(); |
| } |
| |
| #ifndef _WIN32 |
| char currWDPath[512]; |
| ::getcwd(currWDPath, 512); |
| |
| if (m_persistanceDir.at(0) != '/') { |
| if (0 == ::strlen(currWDPath)) { |
| throw InitFailedException( |
| "Failed to get absolute path for persistence directory."); |
| } |
| m_persistanceDir = std::string(currWDPath) + "/" + m_persistanceDir; |
| } |
| |
| // Create persistence directory |
| LOGFINE("SqLiteImpl::init creating persistence directory: %s", |
| m_persistanceDir.c_str()); |
| ::mkdir(m_persistanceDir.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH); |
| |
| // Create region directory |
| std::string regionDirectory = m_persistanceDir + "/" + regionName; |
| ::mkdir(regionDirectory.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH); |
| m_regionDBFile = regionDirectory + "/" + regionName + ".db"; |
| LOGFINE("SqLiteImpl::init creating persistence region file: %s", |
| m_regionDBFile.c_str()); |
| |
| #else |
| char currWDPath[512]; |
| GetCurrentDirectory(512, currWDPath); |
| |
| if (m_persistanceDir.find(":", 0) == std::string::npos) { |
| if (currWDPath == NULL) |
| throw InitFailedException( |
| "Failed to get absolute path for persistence directory."); |
| m_persistanceDir = std::string(currWDPath) + "/" + m_persistanceDir; |
| } |
| |
| LOGINFO("SqLiteImpl::init absolute path for persistence directory: %s", |
| m_persistanceDir.c_str()); |
| |
| // Create persistence directory |
| LOGFINE("SqLiteImpl::init creating persistence directory: %s", |
| m_persistanceDir.c_str()); |
| CreateDirectory(m_persistanceDir.c_str(), NULL); |
| |
| // Create region directory |
| std::string regionDirectory = m_persistanceDir + "/" + regionName; |
| CreateDirectory(regionDirectory.c_str(), NULL); |
| m_regionDBFile = regionDirectory + "/" + regionName + ".db"; |
| |
| LOGFINE("SqLiteImpl::init creating persistence region file: %s", |
| m_regionDBFile.c_str()); |
| #endif |
| |
| if (m_sqliteHelper->initDB(region->getName(), maxPageCount, pageSize, |
| m_regionDBFile.c_str()) != 0) { |
| throw IllegalStateException("Failed to initialize database in SQLITE."); |
| } |
| } |
| |
| void SqLiteImpl::write(const CacheableKeyPtr& key, const CacheablePtr& value, |
| void*& dbHandle) { |
| // Serialize key and value. |
| DataOutput keyDataBuffer, valueDataBuffer; |
| uint32_t keyBufferSize, valueBufferSize; |
| |
| keyDataBuffer.writeObject(key); |
| valueDataBuffer.writeObject(value); |
| void* keyData = const_cast<uint8_t*>(keyDataBuffer.getBuffer(&keyBufferSize)); |
| void* valueData = |
| const_cast<uint8_t*>(valueDataBuffer.getBuffer(&valueBufferSize)); |
| |
| if (m_sqliteHelper->insertKeyValue(keyData, keyBufferSize, valueData, |
| valueBufferSize) != 0) { |
| throw IllegalStateException("Failed to write key value in SQLITE."); |
| } |
| } |
| |
| bool SqLiteImpl::writeAll() { return true; } |
| |
| CacheablePtr SqLiteImpl::read(const CacheableKeyPtr& key, void*& dbHandle) { |
| // Serialize key. |
| DataOutput keyDataBuffer; |
| uint32_t keyBufferSize; |
| keyDataBuffer.writeObject(key); |
| void* keyData = const_cast<uint8_t*>(keyDataBuffer.getBuffer(&keyBufferSize)); |
| void* valueData; |
| uint32_t valueBufferSize; |
| |
| if (m_sqliteHelper->getValue(keyData, keyBufferSize, valueData, |
| valueBufferSize) != 0) { |
| throw IllegalStateException("Failed to read the value from SQLITE."); |
| } |
| |
| // Deserialize object and return value. |
| DataInput valueDataBuffer(reinterpret_cast<uint8_t*>(valueData), |
| valueBufferSize); |
| CacheablePtr retValue; |
| valueDataBuffer.readObject(retValue); |
| |
| // Free memory for serialized form of Cacheable object. |
| free(valueData); |
| return retValue; |
| } |
| |
| bool SqLiteImpl::readAll() { return true; } |
| |
| void SqLiteImpl::destroyRegion() { |
| if (m_sqliteHelper->closeDB() != 0) { |
| throw IllegalStateException("Failed to destroy region from SQLITE."); |
| } |
| |
| #ifndef _WIN32 |
| ::unlink(m_regionDBFile.c_str()); |
| ::rmdir(m_regionDir.c_str()); |
| ::rmdir(m_persistanceDir.c_str()); |
| #else |
| DeleteFile(m_regionDBFile.c_str()); |
| RemoveDirectory(m_regionDir.c_str()); |
| RemoveDirectory(m_persistanceDir.c_str()); |
| #endif |
| } |
| |
| void SqLiteImpl::destroy(const CacheableKeyPtr& key, void*& dbHandle) { |
| // Serialize key and value. |
| DataOutput keyDataBuffer; |
| uint32_t keyBufferSize; |
| keyDataBuffer.writeObject(key); |
| void* keyData = const_cast<uint8_t*>(keyDataBuffer.getBuffer(&keyBufferSize)); |
| if (m_sqliteHelper->removeKey(keyData, keyBufferSize) != 0) { |
| throw IllegalStateException("Failed to destroy the key from SQLITE."); |
| } |
| } |
| |
| SqLiteImpl::SqLiteImpl() { m_sqliteHelper = new SqLiteHelper(); } |
| |
| void SqLiteImpl::close() { |
| int lastError ATTR_UNUSED = 0; |
| m_sqliteHelper->closeDB(); |
| |
| #ifndef _WIN32 |
| ::unlink(m_regionDBFile.c_str()); |
| ::rmdir(m_regionDir.c_str()); |
| ::rmdir(m_persistanceDir.c_str()); |
| #else |
| DeleteFile(m_regionDBFile.c_str()); |
| RemoveDirectory(m_regionDir.c_str()); |
| RemoveDirectory(m_persistanceDir.c_str()); |
| #endif |
| } |
| |
| extern "C" { |
| |
| LIBEXP PersistenceManager* createSqLiteInstance() { return new SqLiteImpl; } |
| } |