173 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			173 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * This file is distributed as part of the MariaDB Corporation MaxScale.  It is free
 | |
|  * software: you can redistribute it and/or modify it under the terms of the
 | |
|  * GNU General Public License as published by the Free Software Foundation,
 | |
|  * version 2.
 | |
|  *
 | |
|  * This program is distributed in the hope that it will be useful, but WITHOUT
 | |
|  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 | |
|  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 | |
|  * details.
 | |
|  *
 | |
|  * You should have received a copy of the GNU General Public License along with
 | |
|  * this program; if not, write to the Free Software Foundation, Inc., 51
 | |
|  * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 | |
|  *
 | |
|  * Copyright MariaDB Corporation Ab 2013-2014
 | |
|  */
 | |
| 
 | |
| /**
 | |
|  * @file spinlock.c  -  Spinlock operations for the MariaDB Corporation MaxScale
 | |
|  *
 | |
|  * @verbatim
 | |
|  * Revision History
 | |
|  *
 | |
|  * Date		Who		Description
 | |
|  * 10/06/13	Mark Riddoch	Initial implementation
 | |
|  *
 | |
|  * @endverbatim
 | |
|  */
 | |
| 
 | |
| #include <spinlock.h>
 | |
| #include <atomic.h>
 | |
| #include <time.h>
 | |
| 
 | |
| /**
 | |
|  * Initialise a spinlock.
 | |
|  *
 | |
|  * @param lock The spinlock to initialise.
 | |
|  */
 | |
| void
 | |
| spinlock_init(SPINLOCK *lock)
 | |
| {
 | |
|     lock->lock = 0;
 | |
| #if SPINLOCK_PROFILE
 | |
|     lock->spins = 0;
 | |
|     lock->acquired = 0;
 | |
|     lock->waiting = 0;
 | |
|     lock->max_waiting = 0;
 | |
|     lock->contended = 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Acquire a spinlock.
 | |
|  *
 | |
|  * @param lock The spinlock to acquire
 | |
|  */
 | |
| void
 | |
| spinlock_acquire(SPINLOCK *lock)
 | |
| {
 | |
| #if SPINLOCK_PROFILE
 | |
| int	spins = 0;
 | |
| 
 | |
| 	atomic_add(&(lock->waiting), 1);
 | |
| #endif
 | |
|         
 | |
| #ifdef __GNUC__
 | |
|     while (__sync_lock_test_and_set(&(lock->lock), 1))
 | |
|         while (lock->lock) {
 | |
| #else    
 | |
|     while (atomic_add(&(lock->lock), 1) != 0)
 | |
|     {
 | |
| 	atomic_add(&(lock->lock), -1);
 | |
| #endif
 | |
| #if SPINLOCK_PROFILE
 | |
| 		atomic_add(&(lock->spins), 1);
 | |
| 		spins++;
 | |
| #endif
 | |
| 	}
 | |
| #if SPINLOCK_PROFILE
 | |
|     if (spins)
 | |
|     {
 | |
|         lock->contended++;
 | |
|         if (lock->maxspins < spins)
 | |
|             lock->maxspins = spins;
 | |
|     }
 | |
|     lock->acquired++;
 | |
|     lock->owner = THREAD_SHELF();
 | |
|     atomic_add(&(lock->waiting), -1);
 | |
| #endif
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Acquire a spinlock if it is not already locked.
 | |
|  *
 | |
|  * @param lock The spinlock to acquire
 | |
|  * @return True if the spinlock was acquired, otherwise false
 | |
|  */
 | |
| int
 | |
| spinlock_acquire_nowait(SPINLOCK *lock)
 | |
| {
 | |
| #ifdef __GNUC__
 | |
|     if (__sync_lock_test_and_set(&(lock->lock), 1)) return FALSE;
 | |
| #else
 | |
|     if (atomic_add(&(lock->lock), 1) != 0)
 | |
|     {
 | |
|         atomic_add(&(lock->lock), -1);
 | |
|         return FALSE;
 | |
|     }
 | |
| #endif
 | |
| #if SPINLOCK_PROFILE
 | |
|     lock->acquired++;
 | |
|     lock->owner = THREAD_SHELF();
 | |
| #endif
 | |
|     return TRUE;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Release a spinlock.
 | |
|  *
 | |
|  * @param lock The spinlock to release
 | |
|  */
 | |
| void
 | |
| spinlock_release(SPINLOCK *lock)
 | |
| {
 | |
|  #if SPINLOCK_PROFILE
 | |
|     if (lock->waiting > lock->max_waiting)
 | |
|         lock->max_waiting = lock->waiting;
 | |
| #endif
 | |
| #ifdef __GNUC__
 | |
|     __sync_synchronize(); /* Memory barrier. */
 | |
|     lock->lock = 0;
 | |
| #else
 | |
|     atomic_add(&(lock->lock), -1);
 | |
| #endif
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Report statistics on a spinlock. This only has an effect if the
 | |
|  * spinlock code has been compiled with the SPINLOCK_PROFILE option set.
 | |
|  *
 | |
|  * NB A callback function is used to return the data rather than
 | |
|  * merely printing to a DCB in order to avoid a dependency on the DCB
 | |
|  * form the spinlock code and also to facilitate other uses of the
 | |
|  * statistics reporting.
 | |
|  *
 | |
|  * @param lock		The spinlock to report on
 | |
|  * @param reporter	The callback function to pass the statistics to
 | |
|  * @param hdl		A handle that is passed to the reporter function
 | |
|  */
 | |
| void
 | |
| spinlock_stats(SPINLOCK *lock, void (*reporter)(void *, char *, int), void *hdl)
 | |
| {
 | |
| #if SPINLOCK_PROFILE
 | |
| 	reporter(hdl, "Spinlock acquired", lock->acquired);
 | |
| 	if (lock->acquired)
 | |
| 	{
 | |
| 		reporter(hdl, "Total no. of spins", lock->spins);
 | |
| 		reporter(hdl, "Average no. of spins (overall)",
 | |
| 					lock->spins / lock->acquired);
 | |
| 		if (lock->contended)
 | |
| 			reporter(hdl, "Average no. of spins (when contended)",
 | |
| 					lock->spins / lock->contended);
 | |
| 		reporter(hdl, "Maximum no. of spins", lock->maxspins);
 | |
| 		reporter(hdl, "Maximim no. of blocked threads",
 | |
| 				lock->max_waiting);
 | |
| 		reporter(hdl, "Contended locks", lock->contended);
 | |
| 		reporter(hdl, "Contention percentage",
 | |
| 				(lock->contended * 100) / lock->acquired);
 | |
| 	}
 | |
| #endif
 | |
| }
 | 
