# include	<ingres.h>
# include	<aux.h>
# include	<tree.h>
# include	"globs.h"
# include	<lock.h>
# include	<sccs.h>

SCCSID(@(#)lockit.c	8.1	12/31/84)

/*
**  LOCKIT -- sets relation locks for integrity locking
**
**	Parameters:
**		root- the root of a query tree;
**		resvar- index of variable to be updated.
**
*/
lockit(root, resvar)
QTREE	*root;
int	resvar;
{
	register QTREE	*r;
	register int	i, j;
	long		vlist[MAXRANGE];
	int		bmap, cv;
	char		mode;
	int		skvr;
	int		redo;
	DESC		*d, *openr1();
	long		restid;
	int		k;

	r = root;
	bmap = r->sym.value.sym_root.lvarm | r->sym.value.sym_root.rvarm;
	if (resvar >= 0)
		bmap |= 01 << resvar;
	else
		restid = -1;
	i = 0;
	/* put relids of relations to be locked into vlist
	   check for and remove duplicates */
	for (j = 0; j < MAXRANGE; j++)
		if (bmap & (01 << j))
		{
			d = openr1(j);
			if (j == resvar)
				restid = d->reltid.ltid;
			for (k = 0; k < i; k++)
				if (vlist[k] == d->reltid.ltid)
					break;
			if (k == i)
				vlist[i++] = d->reltid.ltid;
		}
	cv = i;
/*
 *	set the locks: set the first lock with the sleep option
 *			set other locks checking for failure;
 *			if failure, release all locks, sleep on blocking
 *			lock.
 */
	skvr = -1;
	do
	{
		/* skvr is the index of the relation already locked
		   try to lock the remaining relations */
		redo = FALSE;
		for (i = 0; i < cv; i++)
			if (i != skvr)
			{
				if (restid == vlist[i])
					mode = M_EXCL;
				else
					mode = M_SHARE;
				if (setrll(A_RTN, vlist[i], mode) < 0)
					/* a lock request failed */
				{
					unlall();	/* release all locks */
					setrll(A_SLP, vlist[i], mode);
							/* wait on problem lock*/
					skvr = i;
					redo = TRUE;
					break;	/* reset the other locks */
				}
			}
	}
	while (redo);
}