server/test/com/cloud/async/TestAsyncJobManager.java

// 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.
package com.cloud.async;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import javax.inject.Inject;

import junit.framework.Assert;
import junit.framework.TestCase;

import org.apache.log4j.Logger;

import com.cloud.domain.DomainVO;
import com.cloud.domain.dao.DomainDao;
import com.cloud.domain.dao.DomainDaoImpl;
import com.cloud.exception.PermissionDeniedException;
import com.cloud.host.HostVO;
import com.cloud.host.dao.HostDao;
import com.cloud.host.dao.HostDaoImpl;
import com.cloud.utils.db.Transaction;

public class TestAsyncJobManager extends TestCase {
    public static final Logger s_logger = Logger.getLogger(TestAsyncJobManager.class.getName());

    volatile long s_count = 0;

    @Inject AsyncJobManager asyncMgr;

    public void asyncCall() {
//		long jobId = mgr.rebootVirtualMachineAsync(1, 1);
        long jobId = 0L;
        s_logger.info("Async-call job id: " + jobId);

        while(true) {
            AsyncJobResult result;
            try {
                result = asyncMgr.queryAsyncJobResult(jobId);

                if(result.getJobStatus() != AsyncJobResult.STATUS_IN_PROGRESS) {
                    s_logger.info("Async-call completed, result: " + result.toString());
                    break;
                }
                s_logger.info("Async-call is in progress, progress: " + result.toString());

            } catch (PermissionDeniedException e1) {
            }

            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
            }
        }
    }

    public void sequence() {
        final HostDao hostDao = new HostDaoImpl();
        long seq = hostDao.getNextSequence(1);
        s_logger.info("******* seq : " + seq + " ********");

        HashMap<Long, Long> hashMap = new HashMap<Long, Long>();
        final Map<Long, Long> map = Collections.synchronizedMap(hashMap);

        s_count = 0;
        final long maxCount = 1000000;	// test one million times

        Thread t1 = new Thread(new Runnable() {
            @Override
            public void run() {
                while(s_count < maxCount) {
                    s_count++;
                    long seq = hostDao.getNextSequence(1);
                    Assert.assertTrue(map.put(seq, seq) == null);
                }
            }
        });

        Thread t2 = new Thread(new Runnable() {
            @Override
            public void run() {
                while(s_count < maxCount) {
                    s_count++;
                    long seq = hostDao.getNextSequence(1);
                    Assert.assertTrue(map.put(seq, seq) == null);
                }
            }
        });

        t1.start();
        t2.start();

        try {
            t1.join();
            t2.join();
        } catch (InterruptedException e) {
        }
    }

    /*
	public void ipAssignment() {
		final IPAddressDao ipAddressDao = new IPAddressDaoImpl();

		final ConcurrentHashMap<String, IPAddressVO> map = new ConcurrentHashMap<String, IPAddressVO>();
		//final Map<String, String> map = Collections.synchronizedMap(hashMap);

		s_count = 0;
		final long maxCount = 1000000;	// test one million times

		Thread t1 = new Thread(new Runnable() {
			public void run() {
				while(s_count < maxCount) {
					s_count++;

					Transaction txn = Transaction.open("Alex1");
					try {
						IPAddressVO addr = ipAddressDao.assignIpAddress(1, 0, 1, false);
						IPAddressVO returnStr = map.put(addr.getAddress(), addr);
						if(returnStr != null) {
							System.out.println("addr : " + addr.getAddress());
						}
						Assert.assertTrue(returnStr == null);
					} finally {
						txn.close();
					}
				}
			}
		});

		Thread t2 = new Thread(new Runnable() {
			public void run() {
				while(s_count < maxCount) {
					s_count++;

					Transaction txn = Transaction.open("Alex2");
					try {
						IPAddressVO addr = ipAddressDao.assignIpAddress(1, 0, 1, false);
						Assert.assertTrue(map.put(addr.getAddress(), addr) == null);
					} finally {
						txn.close();
					}
				}
			}
		});

		t1.start();
		t2.start();

		try {
			t1.join();
			t2.join();
		} catch (InterruptedException e) {
		}
	}
     */

    private long getRandomLockId() {
        return 1L;

        /*
         *  will use in the future test cases
		int i = new Random().nextInt();
		if(i % 2 == 0)
			return 1L;
		return 2L;
         */
    }

    public void tstLocking() {

        int testThreads = 20;
        Thread[] threads = new Thread[testThreads];

        for(int i = 0; i < testThreads; i++) {
            final int current = i;
            threads[i] = new Thread(new Runnable() {
                @Override
                public void run() {

                    final HostDao hostDao = new HostDaoImpl();
                    while(true) {
                        Transaction txn = Transaction.currentTxn();
                        try {
                            HostVO host = hostDao.acquireInLockTable(getRandomLockId(), 10);
                            if(host != null) {
                                s_logger.info("Thread " + (current + 1) + " acquired lock");

                                try { Thread.sleep(1000); } catch (InterruptedException e) {}

                                s_logger.info("Thread " + (current + 1) + " released lock");
                                hostDao.releaseFromLockTable(host.getId());

                                try { Thread.sleep(1000); } catch (InterruptedException e) {}
                            } else {
                                s_logger.info("Thread " + (current + 1) + " is not able to acquire lock");
                            }
                        } finally {
                            txn.close();
                        }
                    }
                }
            });
            threads[i].start();
        }

        try {
            for(int i = 0; i < testThreads; i++)
                threads[i].join();
        } catch(InterruptedException e) {
        }
    }

    public void testDomain() {
        DomainDao domainDao = new DomainDaoImpl();

		DomainVO domain1 = new DomainVO("d1", 2L, 1L, null, 1);
        domainDao.create(domain1);

		DomainVO domain2 = new DomainVO("d2", 2L, 1L, null, 1);
        domainDao.create(domain2);

		DomainVO domain3 = new DomainVO("d3", 2L, 1L, null, 1);
        domainDao.create(domain3);

		DomainVO domain11 = new DomainVO("d11", 2L, domain1.getId(), null, 1);
        domainDao.create(domain11);

        domainDao.remove(domain11.getId());

		DomainVO domain12 = new DomainVO("d12", 2L, domain1.getId(), null, 1);
        domainDao.create(domain12);

        domainDao.remove(domain3.getId());
		DomainVO domain4 = new DomainVO("d4", 2L, 1L, null, 1);
        domainDao.create(domain4);
    }
}