Java tutorial
/** * personium.io * Copyright 2014 FUJITSU LIMITED * * Licensed 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. */ package com.fujitsu.dc.test.jersey.concurrent; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertTrue; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.http.HttpHeaders; import org.apache.http.HttpStatus; import org.json.simple.JSONObject; import org.junit.Ignore; import org.junit.Test; import org.junit.experimental.categories.Category; import org.junit.runner.RunWith; import com.fujitsu.dc.common.es.response.DcSearchResponse; import com.fujitsu.dc.core.model.impl.es.EsModel; import com.fujitsu.dc.core.model.impl.es.accessor.EntitySetAccessor; import com.fujitsu.dc.core.model.lock.Lock; import com.fujitsu.dc.core.model.lock.LockManager; import com.fujitsu.dc.test.categories.Integration; import com.fujitsu.dc.test.categories.Regression; import com.fujitsu.dc.test.categories.Unit; import com.fujitsu.dc.test.jersey.AbstractCase; import com.fujitsu.dc.test.jersey.DcRunner; import com.fujitsu.dc.test.unit.core.UrlUtils; import com.fujitsu.dc.test.utils.Http; import com.fujitsu.dc.test.utils.TResponse; import com.sun.jersey.test.framework.JerseyTest; /** * ODataAPI???. */ @RunWith(DcRunner.class) @Category({ Unit.class, Integration.class, Regression.class }) public class ConcurrentODataRequestTest extends JerseyTest { private static final String CELL_NAME = "testcell1"; private static final String BOX_NAME = "box1"; private static final String NEW_ROLE_NAME = "roleForConcurrentTest"; private static final String NEW_ROLE_NAME_2 = "roleForConcurrentTest2"; private static final String NEW_BOX_NAME = "boxForConcurrentTest"; private static final String NEW_BOX_SCHEMA = "http://example.com/schema1/"; private static final String NEW_BOX_SCHEMA_2 = "https://example.net/schema2/"; private static final String ACCOUNT_NAME = "account2"; private static final String ROLE_NAME = "role1"; private static final int NUM_CONCURRENCY = 10; /** . */ static Log log = LogFactory.getLog(ConcurrentODataRequestTest.class);; /** * . */ public ConcurrentODataRequestTest() { super("com.fujitsu.dc.core.rs"); } /** * . */ static class Counters { int countSuccess = 0; int countFailure = 0; int countOverflow = 0; synchronized void incSuccess() { this.countSuccess++; } synchronized void incFailure() { this.countFailure++; } synchronized void incOverflow() { this.countOverflow++; } void assertSuccessCount(int expectedSuccessCount) { assertEquals(expectedSuccessCount, this.countSuccess); } void assertFailureCount(int expectedFailureCount) { assertEquals(expectedFailureCount, this.countFailure); } void assertTotalCount(int expectedTotalCount) { assertEquals(expectedTotalCount, this.countSuccess + this.countFailure + this.countOverflow); } void assertOverflowNonZero() { assertTrue(this.countOverflow > 0); } void debugPrint() { log.debug(" Success: " + countSuccess); log.debug(" Failure: " + countFailure); log.debug("Overflow: " + countOverflow); } } /** * ???????????. * @throws InterruptedException InterruptedException */ @Test public final void ? ??????????()throws InterruptedException { // final Counters counters = new Counters(); // ?? List<Thread> listThread = new ArrayList<Thread>(); for (int i = 0; i < NUM_CONCURRENCY; i++) { final Http theReq = this.createRoleRequest(NEW_ROLE_NAME, BOX_NAME); Runnable runnable = new Runnable() { @Override public void run() { // TResponse resp = theReq.returns(); log.debug("Status Code = " + resp.getStatusCode()); // ? if (HttpStatus.SC_CREATED == resp.getStatusCode()) { counters.incSuccess(); } else if (HttpStatus.SC_SERVICE_UNAVAILABLE == resp.getStatusCode()) { counters.incOverflow(); } else { counters.incFailure(); } } }; Thread t = new Thread(runnable); listThread.add(t); } try { // ??? for (Thread t : listThread) { t.start(); } // ? for (Thread t : listThread) { t.join(); } // ?? counters.debugPrint(); // ???????? counters.assertTotalCount(NUM_CONCURRENCY); // ??????????? counters.assertSuccessCount(1); } finally { // ????Entity? this.deleteRoleRequest(NEW_ROLE_NAME, BOX_NAME).returns().statusCode(HttpStatus.SC_NO_CONTENT); // ???204??? } } /** * ???????????. * @throws InterruptedException InterruptedException */ @Test public final void ? ??????????()throws InterruptedException { // ???? this.createRoleRequest(NEW_ROLE_NAME, BOX_NAME).returns(); // final Counters counters = new Counters(); // ?? List<Thread> listThread = new ArrayList<Thread>(); for (int i = 0; i < NUM_CONCURRENCY; i++) { final Http theReq = this.deleteRoleRequest(NEW_ROLE_NAME, BOX_NAME); Runnable runnable = new Runnable() { @Override public void run() { // TResponse resp = theReq.returns(); log.debug("Status Code = " + resp.getStatusCode()); // ? if (HttpStatus.SC_NO_CONTENT == resp.getStatusCode()) { counters.incSuccess(); } else if (HttpStatus.SC_SERVICE_UNAVAILABLE == resp.getStatusCode()) { counters.incOverflow(); } else { counters.incFailure(); } } }; Thread t = new Thread(runnable); listThread.add(t); } // ?? try { // ??? for (Thread t : listThread) { t.start(); } // ? for (Thread t : listThread) { t.join(); } // ?? counters.debugPrint(); // ???????? counters.assertTotalCount(NUM_CONCURRENCY); // ??????????? counters.assertSuccessCount(1); } finally { // ? this.deleteRoleRequest(NEW_ROLE_NAME, BOX_NAME).returns().statusCode(HttpStatus.SC_NOT_FOUND); // ?(???)??404??? } } /** * ??ETag????????????. * @throws InterruptedException InterruptedException */ @Test public final void ? ?ETag????????????()throws InterruptedException { // ???? TResponse resp = this.createBoxRequest(NEW_BOX_NAME, NEW_BOX_SCHEMA).returns(); final String etag = resp.getHeader(HttpHeaders.ETAG); // final Counters counters = new Counters(); // ?? List<Thread> listThread = new ArrayList<Thread>(); for (int i = 0; i < NUM_CONCURRENCY; i++) { final Http theReq = this.updateBoxRequest(NEW_BOX_NAME, NEW_BOX_NAME, NEW_BOX_SCHEMA_2, etag); Runnable runnable = new Runnable() { @Override public void run() { // TResponse resp = theReq.returns(); log.debug("Status Code = " + resp.getStatusCode()); // ? if (HttpStatus.SC_NO_CONTENT == resp.getStatusCode()) { counters.incSuccess(); } else if (HttpStatus.SC_SERVICE_UNAVAILABLE == resp.getStatusCode()) { counters.incOverflow(); } else { counters.incFailure(); } } }; Thread t = new Thread(runnable); listThread.add(t); } // ?? try { // ??? for (Thread t : listThread) { t.start(); } // ? for (Thread t : listThread) { t.join(); } // ?? counters.debugPrint(); // ???????? counters.assertTotalCount(NUM_CONCURRENCY); // ????????? counters.assertSuccessCount(1); } finally { // ? this.deleteBoxRequest(NEW_BOX_NAME).returns().statusCode(HttpStatus.SC_NO_CONTENT); // ???204??? } } /** * ??ETag???????????. * @throws InterruptedException InterruptedException */ @Test public final void ? ?ETag?? ?????????() throws InterruptedException { // ???? this.createRoleRequest(NEW_ROLE_NAME, BOX_NAME).returns(); // final Counters counters = new Counters(); // ?? List<Thread> listThread = new ArrayList<Thread>(); for (int i = 0; i < NUM_CONCURRENCY; i++) { final Http theReq = this.updateRoleRequest(NEW_ROLE_NAME, BOX_NAME, NEW_ROLE_NAME_2); Runnable runnable = new Runnable() { @Override public void run() { // TResponse resp = theReq.returns(); log.debug("Status Code = " + resp.getStatusCode()); // ? if (HttpStatus.SC_NO_CONTENT == resp.getStatusCode()) { counters.incSuccess(); } else if (HttpStatus.SC_SERVICE_UNAVAILABLE == resp.getStatusCode()) { counters.incOverflow(); } else { counters.incFailure(); } } }; Thread t = new Thread(runnable); listThread.add(t); } // ?? try { // ??? for (Thread t : listThread) { t.start(); } // ? for (Thread t : listThread) { t.join(); } // ?? counters.debugPrint(); // ???????? counters.assertTotalCount(NUM_CONCURRENCY); // ??????????? counters.assertSuccessCount(1); } finally { // ? this.deleteRoleRequest(NEW_ROLE_NAME_2, BOX_NAME).returns().statusCode(HttpStatus.SC_NO_CONTENT); // ???204??? } } /** * ??????????. * @throws InterruptedException InterruptedException */ @Test public final void ??????????()throws InterruptedException { // final Counters counters = new Counters(); // ?? List<Thread> listThread = new ArrayList<Thread>(); for (int i = 0; i < NUM_CONCURRENCY; i++) { final Http theReq = this.linkAccountRoleRequest(ACCOUNT_NAME, ROLE_NAME, null); Runnable runnable = new Runnable() { @Override public void run() { // TResponse resp = theReq.returns(); log.debug("Status Code = " + resp.getStatusCode()); // ? if (HttpStatus.SC_NO_CONTENT == resp.getStatusCode()) { counters.incSuccess(); } else if (HttpStatus.SC_SERVICE_UNAVAILABLE == resp.getStatusCode()) { counters.incOverflow(); } else { counters.incFailure(); } } }; Thread t = new Thread(runnable); listThread.add(t); } // ?? try { // ??? for (Thread t : listThread) { t.start(); } // ? for (Thread t : listThread) { t.join(); } // ?? counters.debugPrint(); // ???????? counters.assertTotalCount(NUM_CONCURRENCY); // ??????????? counters.assertSuccessCount(1); } finally { // ? this.unlinkAccountRoleRequest(ACCOUNT_NAME, ROLE_NAME, null).returns() .statusCode(HttpStatus.SC_NO_CONTENT); // ???204??? } } /** * ??????????. * @throws InterruptedException InterruptedException */ @Test public final void ??????????()throws InterruptedException { this.linkAccountRoleRequest(ACCOUNT_NAME, ROLE_NAME, null).returns(); // final Counters counters = new Counters(); // ?? List<Thread> listThread = new ArrayList<Thread>(); for (int i = 0; i < NUM_CONCURRENCY; i++) { final Http theReq = this.unlinkAccountRoleRequest(ACCOUNT_NAME, ROLE_NAME, null); Runnable runnable = new Runnable() { @Override public void run() { // TResponse resp = theReq.returns(); log.debug("Status Code = " + resp.getStatusCode()); // ? if (HttpStatus.SC_NO_CONTENT == resp.getStatusCode()) { counters.incSuccess(); } else if (HttpStatus.SC_SERVICE_UNAVAILABLE == resp.getStatusCode()) { counters.incOverflow(); } else { counters.incFailure(); } } }; Thread t = new Thread(runnable); listThread.add(t); } // ?? try { // ??? for (Thread t : listThread) { t.start(); } // ? for (Thread t : listThread) { t.join(); } // ?? counters.debugPrint(); // ???????? counters.assertTotalCount(NUM_CONCURRENCY); // ??????????? counters.assertSuccessCount(1); } finally { // ? this.unlinkAccountRoleRequest(ACCOUNT_NAME, ROLE_NAME, null).returns() // ???????????? .statusCode(HttpStatus.SC_NOT_FOUND); // ?????????????? } } /** * ??????503??. */ @Test @Ignore("?IT???????????????") public void ??????503 ??() { String accessTargetCellId = getTestingCellId(); Lock lock = getLockForTimeoutTest(accessTargetCellId); try { int httpStatusCode = createRoleAndGetStatus(); assertEquals(HttpStatus.SC_SERVICE_UNAVAILABLE, httpStatusCode); } finally { releaseLockForTimeoutTest(lock); } } private Lock getLockForTimeoutTest(String cellId) { return LockManager.getLock("odata", cellId, null, null); } private void releaseLockForTimeoutTest(Lock lock) { lock.release(); } private int createRoleAndGetStatus() { Http request = this.createRoleRequest(NEW_ROLE_NAME, BOX_NAME); TResponse response = request.returns(); return response.getStatusCode(); } private String getTestingCellId() { Map<String, Object> query = buildQueryForRetrieveCellByName(); return retrieveCellId(query); } private Map<String, Object> buildQueryForRetrieveCellByName() { Map<String, Object> termQuery = buildBasicQueryForRetrieveByName(); Map<String, Object> query = buildQueryMap(termQuery); Map<String, Object> filter = buildFilterMap(query); return filter; } private Map<String, Object> buildBasicQueryForRetrieveByName() { Map<String, Object> query = new HashMap<String, Object>(); query.put("s.Name", CELL_NAME); return query; } private Map<String, Object> buildQueryMap(Map<String, Object> subQuery) { Map<String, Object> query = new HashMap<String, Object>(); query.put("term", subQuery); return query; } private Map<String, Object> buildFilterMap(Map<String, Object> subQuery) { Map<String, Object> filter = new HashMap<String, Object>(); filter.put("filter", subQuery); return filter; } private String retrieveCellId(Map<String, Object> query) { EntitySetAccessor ecCells = EsModel.cell(); DcSearchResponse response = ecCells.search(query); if (isRetriveFailed(response)) { return null; } return response.getHits().getAt(0).getId(); } private boolean isRetriveFailed(DcSearchResponse response) { return response == null || response.getHits().getCount() == 0; } /** * Role?. * @param roleName ?? * @param boxName Box?? * @return */ Http deleteRoleRequest(String roleName, String boxName) { return Http.request("role-delete.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("rolename", roleName) .with("boxname", "'" + boxName + "'"); } /** * Role??. * @param roleName ?? * @param boxName Box?? * @return */ @SuppressWarnings("unchecked") Http createRoleRequest(String roleName, String boxName) { JSONObject body = new JSONObject(); body.put("Name", roleName); body.put("_Box.Name", boxName); return Http.request("role-create.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("body", body.toString()); } @SuppressWarnings("unchecked") Http updateRoleRequest(String roleName, String boxName, String newRoleName) { JSONObject body = new JSONObject(); body.put("Name", newRoleName); body.put("_Box.Name", boxName); return Http.request("cell/role-update.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("rolename", roleName) .with("boxname", "'" + boxName + "'").with("body", body.toString()); } /** * Box?. * @param name Box?? * @return */ Http deleteBoxRequest(String name) { return Http.request("cell/box-delete.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("boxPath", name); } /** * Box??. * @param name Box?? * @param schema schema url * @return */ Http createBoxRequest(String name, String schema) { return Http.request("cell/box-create-with-scheme.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("boxPath", name).with("schema", schema); } Http updateBoxRequest(String name, String newName, String schema, String etag) { return Http.request("cell/box-update.txt").with("cellPath", CELL_NAME).with("boxPath", name) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("etag", etag).with("newBoxPath", newName) .with("schema", schema); } /** * Account-Role?Link??. * @param roleName ?? * @param boxName Box?? * @return */ Http linkAccountRoleRequest(String accountName, String roleName, String boxName) { String roleKeyPredicate = "Name='" + roleName + "'"; if (boxName != null) { roleKeyPredicate += ",_Box.Name='" + boxName + "'"; } String roleUrl = UrlUtils.cellCtl(CELL_NAME, "Role"); roleUrl += "(" + roleKeyPredicate + ")"; return Http.request("cell/link-account-role.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("username", accountName) .with("roleUrl", roleUrl); } /** * Account-Role?Link?. * @param accountName Account?? * @param roleName ?? * @param boxName Box?? * @return */ Http unlinkAccountRoleRequest(String accountName, String roleName, String boxName) { String roleKeyPredicate = "Name='" + roleName + "'"; if (boxName != null) { roleKeyPredicate += ",_Box.Name='" + boxName + "'"; } return Http.request("cell/unlink-account-role.txt").with("cellPath", CELL_NAME) .with("token", AbstractCase.MASTER_TOKEN_NAME).with("username", accountName) .with("keyPredicate", roleKeyPredicate); } }