Line data    Source code

       1             : /**
       2             : Copyright (c) 2019-2022 Roman Katuntsev <sbkarr@stappler.org>
       3             : Copyright (c) 2023 Stappler LLC <admin@stappler.dev>
       4             : 
       5             : Permission is hereby granted, free of charge, to any person obtaining a copy
       6             : of this software and associated documentation files (the "Software"), to deal
       7             : in the Software without restriction, including without limitation the rights
       8             : to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
       9             : copies of the Software, and to permit persons to whom the Software is
      10             : furnished to do so, subject to the following conditions:
      11             : 
      12             : The above copyright notice and this permission notice shall be included in
      13             : all copies or substantial portions of the Software.
      14             : 
      15             : THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
      16             : IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
      17             : FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
      18             : AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
      19             : LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
      20             : OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
      21             : THE SOFTWARE.
      22             : **/
      23             : 
      24             : #include "SPMemUserData.h"
      25             : 
      26             : namespace STAPPLER_VERSIONIZED stappler::memory::pool {
      27             : 
      28             : struct Pool_StoreHandle : AllocPool {
      29             :         void *pointer;
      30             :         memory::function<void()> callback;
      31             : };
      32             : 
      33       15025 : static status_t sa_request_store_custom_cleanup(void *ptr) {
      34       15025 :         if (ptr) {
      35       15025 :                 auto ref = (Pool_StoreHandle *)ptr;
      36       15025 :                 if (ref->callback) {
      37         400 :                         memory::pool::push(ref->callback.get_allocator());
      38         400 :                         ref->callback();
      39         400 :                         memory::pool::pop();
      40             :                 }
      41             :         }
      42       15025 :         return SUCCESS;
      43             : }
      44             : 
      45       18570 : void store(pool_t *pool, void *ptr, const StringView &key, memory::function<void()> &&cb) {
      46       18570 :         memory::pool::push(pool);
      47             : 
      48       18570 :         void * ret = nullptr;
      49       18570 :         pool::userdata_get(&ret, key.data(), key.size(), pool);
      50       18574 :         if (ret) {
      51        3554 :                 auto h = (Pool_StoreHandle *)ret;
      52        3554 :                 h->pointer = ptr;
      53        3554 :                 if (cb) {
      54           0 :                         h->callback = std::move(cb);
      55             :                 } else {
      56        3554 :                         h->callback = nullptr;
      57             :                 }
      58             :         } else {
      59       15020 :                 auto h = new (pool) Pool_StoreHandle();
      60       15018 :                 h->pointer = ptr;
      61       15018 :                 if (cb) {
      62         454 :                         h->callback = std::move(cb);
      63             :                 }
      64             : 
      65       15018 :                 if (key.terminated()) {
      66       15014 :                         pool::userdata_set(h, key.data(), sa_request_store_custom_cleanup, pool);
      67             :                 } else {
      68           0 :                         char buf[key.size() + 1];
      69           0 :                         memcpy(buf, key.data(), key.size());
      70           0 :                         buf[key.size()] = 0;
      71           0 :                         pool::userdata_set(h, key.data(), sa_request_store_custom_cleanup, pool);
      72           0 :                 }
      73             :         }
      74       18566 :         memory::pool::pop();
      75       18567 : }
      76             : 
      77             : }