testsuites/integration/integration-context-test/src/test/java/org/onap/policy/apex/testsuites/integration/context/lock/modifier/WriteLockAlbumModifier.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Ericsson. All rights reserved.
   4  *  Modifications Copyright (C) 2019 Nordix Foundation.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * SPDX-License-Identifier: Apache-2.0
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.policy.apex.testsuites.integration.context.lock.modifier;
  23
  24 import java.util.Random;
  25 import org.onap.policy.apex.context.ContextAlbum;
  26 import org.onap.policy.apex.context.ContextException;
  27 import org.onap.policy.apex.context.test.concepts.TestContextLongItem;
  28 import org.slf4j.ext.XLogger;
  29 import org.slf4j.ext.XLoggerFactory;
  30
  31 /**
  32  * The Class WriteLockAlbumModifier implements a write lock context album.
  33  */
  34 public class WriteLockAlbumModifier implements AlbumModifier {
  35     private static final XLogger LOGGER = XLoggerFactory.getXLogger(WriteLockAlbumModifier.class);
  36
  37     /**
  38      * {@inheritDoc}.
  39      */
  40     @Override
  41     public void modifyAlbum(final ContextAlbum contextAlbum, final int loopSize, final int arraySize) {
  42         final Random rand = new Random();
  43         for (int i = 0; i < loopSize; i++) {
  44             final String nextLongKey = Integer.toString(rand.nextInt(arraySize));
  45             try {
  46                 contextAlbum.lockForWriting(nextLongKey);
  47             } catch (final ContextException e) {
  48                 LOGGER.error("could not acquire write lock on context album, key=" + nextLongKey, e);
  49                 continue;
  50             }
  51
  52             final TestContextLongItem item = (TestContextLongItem) contextAlbum.get(nextLongKey);
  53             long value = item.getLongValue();
  54             if (LOGGER.isTraceEnabled()) {
  55                 LOGGER.trace("lock type={}, value={}", LockType.WRITE_LOCK, value);
  56             }
  57             item.setLongValue(++value);
  58             contextAlbum.put(nextLongKey, item);
  59
  60             try {
  61                 contextAlbum.unlockForWriting(nextLongKey);
  62             } catch (final ContextException e) {
  63                 LOGGER.error("could not release write lock on context album, key=" + nextLongKey, e);
  64             }
  65         }
  66
  67     }
  68
  69 }