sequence.c 21.2 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * sequence.c
4
 *	  PostgreSQL sequences support code.
5
 *
6
 * Portions Copyright (c) 1996-2001, PostgreSQL Global Development Group
7 8 9 10
 * Portions Copyright (c) 1994, Regents of the University of California
 *
 *
 * IDENTIFICATION
11
 *	  $Header: /cvsroot/pgsql/src/backend/commands/sequence.c,v 1.62 2001/08/10 18:57:34 tgl Exp $
12
 *
13 14
 *-------------------------------------------------------------------------
 */
15
#include "postgres.h"
16

17 18
#include <ctype.h>

19 20 21
#include "access/heapam.h"
#include "commands/creatinh.h"
#include "commands/sequence.h"
B
Bruce Momjian 已提交
22
#include "miscadmin.h"
23
#include "utils/acl.h"
B
Bruce Momjian 已提交
24
#include "utils/builtins.h"
25 26 27 28
#ifdef MULTIBYTE
#include "mb/pg_wchar.h"
#endif

29

30
#define SEQ_MAGIC	  0x1717
31 32 33 34

#define SEQ_MAXVALUE	((int4)0x7FFFFFFF)
#define SEQ_MINVALUE	-(SEQ_MAXVALUE)

V
Vadim B. Mikheev 已提交
35 36 37 38 39
/*
 * We don't want to log each fetching values from sequences,
 * so we pre-log a few fetches in advance. In the event of
 * crash we can lose as much as we pre-logged.
 */
B
Bruce Momjian 已提交
40
#define SEQ_LOG_VALS	32
41 42 43

typedef struct sequence_magic
{
44
	uint32		magic;
45
} sequence_magic;
46

47 48
typedef struct SeqTableData
{
49 50 51 52 53 54
	char	   *name;
	Oid			relid;
	Relation	rel;
	int4		cached;
	int4		last;
	int4		increment;
55
	struct SeqTableData *next;
56
} SeqTableData;
57 58 59 60 61

typedef SeqTableData *SeqTable;

static SeqTable seqtab = NULL;

62
static char *get_seq_name(text *seqin);
63
static SeqTable init_sequence(char *caller, char *name);
64 65
static Form_pg_sequence read_info(char *caller, SeqTable elm, Buffer *buf);
static void init_params(CreateSeqStmt *seq, Form_pg_sequence new);
66
static int	get_param(DefElem *def);
67
static void do_setval(char *seqname, int32 next, bool iscalled);
68 69

/*
B
Bruce Momjian 已提交
70
 * DefineSequence
71
 *				Creates a new sequence relation
72 73
 */
void
74
DefineSequence(CreateSeqStmt *seq)
75
{
76
	FormData_pg_sequence new;
77 78 79 80 81 82
	CreateStmt *stmt = makeNode(CreateStmt);
	ColumnDef  *coldef;
	TypeName   *typnam;
	Relation	rel;
	Buffer		buf;
	PageHeader	page;
83
	sequence_magic *sm;
84 85 86 87 88
	HeapTuple	tuple;
	TupleDesc	tupDesc;
	Datum		value[SEQ_COL_LASTCOL];
	char		null[SEQ_COL_LASTCOL];
	int			i;
89
	NameData	name;
90 91 92 93 94

	/* Check and set values */
	init_params(seq, &new);

	/*
95
	 * Create relation (and fill *null & *value)
96 97 98
	 */
	stmt->tableElts = NIL;
	for (i = SEQ_COL_FIRSTCOL; i <= SEQ_COL_LASTCOL; i++)
99
	{
100 101 102
		typnam = makeNode(TypeName);
		typnam->setof = FALSE;
		typnam->arrayBounds = NULL;
B
Bruce Momjian 已提交
103
		typnam->typmod = -1;
104 105
		coldef = makeNode(ColumnDef);
		coldef->typename = typnam;
106 107
		coldef->raw_default = NULL;
		coldef->cooked_default = NULL;
108 109 110 111 112
		coldef->is_not_null = false;
		null[i - 1] = ' ';

		switch (i)
		{
113 114 115
			case SEQ_COL_NAME:
				typnam->name = "name";
				coldef->colname = "sequence_name";
116 117
				namestrcpy(&name, seq->seqname);
				value[i - 1] = NameGetDatum(&name);
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
				break;
			case SEQ_COL_LASTVAL:
				typnam->name = "int4";
				coldef->colname = "last_value";
				value[i - 1] = Int32GetDatum(new.last_value);
				break;
			case SEQ_COL_INCBY:
				typnam->name = "int4";
				coldef->colname = "increment_by";
				value[i - 1] = Int32GetDatum(new.increment_by);
				break;
			case SEQ_COL_MAXVALUE:
				typnam->name = "int4";
				coldef->colname = "max_value";
				value[i - 1] = Int32GetDatum(new.max_value);
				break;
			case SEQ_COL_MINVALUE:
				typnam->name = "int4";
				coldef->colname = "min_value";
				value[i - 1] = Int32GetDatum(new.min_value);
				break;
			case SEQ_COL_CACHE:
				typnam->name = "int4";
				coldef->colname = "cache_value";
				value[i - 1] = Int32GetDatum(new.cache_value);
				break;
V
Vadim B. Mikheev 已提交
144 145 146
			case SEQ_COL_LOG:
				typnam->name = "int4";
				coldef->colname = "log_cnt";
B
Bruce Momjian 已提交
147
				value[i - 1] = Int32GetDatum((int32) 1);
V
Vadim B. Mikheev 已提交
148
				break;
149 150 151 152 153 154 155 156 157 158
			case SEQ_COL_CYCLE:
				typnam->name = "char";
				coldef->colname = "is_cycled";
				value[i - 1] = CharGetDatum(new.is_cycled);
				break;
			case SEQ_COL_CALLED:
				typnam->name = "char";
				coldef->colname = "is_called";
				value[i - 1] = CharGetDatum('f');
				break;
159 160 161 162 163 164 165
		}
		stmt->tableElts = lappend(stmt->tableElts, coldef);
	}

	stmt->relname = seq->seqname;
	stmt->inhRelnames = NIL;
	stmt->constraints = NIL;
166 167
	stmt->istemp = seq->istemp;
	stmt->hasoids = false;
168

169
	DefineRelation(stmt, RELKIND_SEQUENCE);
170

171
	rel = heap_openr(seq->seqname, AccessExclusiveLock);
172
	tupDesc = RelationGetDescr(rel);
173

174 175
	/* Initialize first page of relation with special magic number */

176 177 178
	buf = ReadBuffer(rel, P_NEW);

	if (!BufferIsValid(buf))
179
		elog(ERROR, "DefineSequence: ReadBuffer failed");
180

181 182
	Assert(BufferGetBlockNumber(buf) == 0);

183 184 185 186 187 188
	page = (PageHeader) BufferGetPage(buf);

	PageInit((Page) page, BufferGetPageSize(buf), sizeof(sequence_magic));
	sm = (sequence_magic *) PageGetSpecialPointer(page);
	sm->magic = SEQ_MAGIC;

189 190 191
	/* hack: ensure heap_insert will insert on the just-created page */
	rel->rd_targblock = 0;

192 193 194 195
	/* Now - form & insert sequence tuple */
	tuple = heap_formtuple(tupDesc, value, null);
	heap_insert(rel, tuple);

196 197 198 199 200
	/*
	 * After crash REDO of heap_insert above would re-init page and
	 * our magic number would be lost. We have to log sequence creation.
	 * This means two log records instead of one -:(
	 */
201
	LockBuffer(buf, BUFFER_LOCK_EXCLUSIVE);
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229
	START_CRIT_SECTION();
	{
		xl_seq_rec			xlrec;
		XLogRecPtr			recptr;
		XLogRecData 		rdata[2];
		Form_pg_sequence	newseq = (Form_pg_sequence) GETSTRUCT(tuple);

		/* We do not log first nextval call, so "advance" sequence here */
		newseq->is_called = 't';
		newseq->log_cnt = 0;

		xlrec.node = rel->rd_node;
		rdata[0].buffer = InvalidBuffer;
		rdata[0].data = (char *) &xlrec;
		rdata[0].len = sizeof(xl_seq_rec);
		rdata[0].next = &(rdata[1]);

		rdata[1].buffer = InvalidBuffer;
		rdata[1].data = (char*) tuple->t_data;
		rdata[1].len = tuple->t_len;
		rdata[1].next = NULL;

		recptr = XLogInsert(RM_SEQ_ID, XLOG_SEQ_LOG | XLOG_NO_TRAN, rdata);

		PageSetLSN(page, recptr);
		PageSetSUI(page, ThisStartUpID);
	}
	END_CRIT_SECTION();
230

231 232
	LockBuffer(buf, BUFFER_LOCK_UNLOCK);
	ReleaseBuffer(buf);
233
	heap_close(rel, AccessExclusiveLock);
234 235 236
}


237 238
Datum
nextval(PG_FUNCTION_ARGS)
239
{
240 241
	text	   *seqin = PG_GETARG_TEXT_P(0);
	char	   *seqname = get_seq_name(seqin);
242 243
	SeqTable	elm;
	Buffer		buf;
244
	Form_pg_sequence seq;
245
	int32		incby,
246 247
				maxv,
				minv,
V
Vadim B. Mikheev 已提交
248 249 250 251
				cache,
				log,
				fetch,
				last;
252
	int32		result,
253 254
				next,
				rescnt = 0;
V
Vadim B. Mikheev 已提交
255
	bool		logit = false;
256

257
	if (pg_aclcheck(seqname, GetUserId(), ACL_UPDATE) != ACLCHECK_OK)
258 259 260
		elog(ERROR, "%s.nextval: you don't have permissions to set sequence %s",
			 seqname, seqname);

V
Vadim B. Mikheev 已提交
261
	/* open and AccessShareLock sequence */
262
	elm = init_sequence("nextval", seqname);
263

264 265 266 267 268
	pfree(seqname);

	if (elm->last != elm->cached)		/* some numbers were cached */
	{
		elm->last += elm->increment;
269
		PG_RETURN_INT32(elm->last);
270
	}
271

B
Bruce Momjian 已提交
272 273
	seq = read_info("nextval", elm, &buf);		/* lock page' buffer and
												 * read tuple */
274

V
Vadim B. Mikheev 已提交
275
	last = next = result = seq->last_value;
276 277 278
	incby = seq->increment_by;
	maxv = seq->max_value;
	minv = seq->min_value;
V
Vadim B. Mikheev 已提交
279 280
	fetch = cache = seq->cache_value;
	log = seq->log_cnt;
281 282

	if (seq->is_called != 't')
V
Vadim B. Mikheev 已提交
283
	{
284
		rescnt++;				/* last_value if not called */
V
Vadim B. Mikheev 已提交
285 286 287
		fetch--;
		log--;
	}
288

V
Vadim B. Mikheev 已提交
289 290 291 292 293 294
	if (log < fetch)
	{
		fetch = log = fetch - log + SEQ_LOG_VALS;
		logit = true;
	}

B
Bruce Momjian 已提交
295
	while (fetch)				/* try to fetch cache [+ log ] numbers */
296
	{
297 298 299 300 301

		/*
		 * Check MAXVALUE for ascending sequences and MINVALUE for
		 * descending sequences
		 */
302
		if (incby > 0)
303
		{
304
			/* ascending sequence */
305 306 307 308
			if ((maxv >= 0 && next > maxv - incby) ||
				(maxv < 0 && next + incby > maxv))
			{
				if (rescnt > 0)
V
Vadim B. Mikheev 已提交
309
					break;		/* stop fetching */
310
				if (seq->is_cycled != 't')
311
					elog(ERROR, "%s.nextval: reached MAXVALUE (%d)",
312 313 314 315 316 317 318 319
						 elm->name, maxv);
				next = minv;
			}
			else
				next += incby;
		}
		else
		{
320
			/* descending sequence */
321 322 323 324
			if ((minv < 0 && next < minv - incby) ||
				(minv >= 0 && next + incby < minv))
			{
				if (rescnt > 0)
V
Vadim B. Mikheev 已提交
325
					break;		/* stop fetching */
326
				if (seq->is_cycled != 't')
327
					elog(ERROR, "%s.nextval: reached MINVALUE (%d)",
328 329 330 331 332 333
						 elm->name, minv);
				next = maxv;
			}
			else
				next += incby;
		}
V
Vadim B. Mikheev 已提交
334 335 336 337 338 339
		fetch--;
		if (rescnt < cache)
		{
			log--;
			rescnt++;
			last = next;
B
Bruce Momjian 已提交
340 341
			if (rescnt == 1)	/* if it's first result - */
				result = next;	/* it's what to return */
V
Vadim B. Mikheev 已提交
342
		}
343 344 345 346
	}

	/* save info in local cache */
	elm->last = result;			/* last returned number */
V
Vadim B. Mikheev 已提交
347 348
	elm->cached = last;			/* last fetched number */

349
	START_CRIT_SECTION();
V
Vadim B. Mikheev 已提交
350 351 352 353
	if (logit)
	{
		xl_seq_rec	xlrec;
		XLogRecPtr	recptr;
B
Bruce Momjian 已提交
354
		XLogRecData rdata[2];
355
		Page		page = BufferGetPage(buf);
V
Vadim B. Mikheev 已提交
356 357

		xlrec.node = elm->rel->rd_node;
358
		rdata[0].buffer = InvalidBuffer;
B
Bruce Momjian 已提交
359
		rdata[0].data = (char *) &xlrec;
360 361 362 363 364 365 366
		rdata[0].len = sizeof(xl_seq_rec);
		rdata[0].next = &(rdata[1]);

		seq->last_value = next;
		seq->is_called = 't';
		seq->log_cnt = 0;
		rdata[1].buffer = InvalidBuffer;
B
Bruce Momjian 已提交
367 368 369
		rdata[1].data = (char *) page + ((PageHeader) page)->pd_upper;
		rdata[1].len = ((PageHeader) page)->pd_special -
			((PageHeader) page)->pd_upper;
370 371
		rdata[1].next = NULL;

B
Bruce Momjian 已提交
372
		recptr = XLogInsert(RM_SEQ_ID, XLOG_SEQ_LOG | XLOG_NO_TRAN, rdata);
V
Vadim B. Mikheev 已提交
373

374 375
		PageSetLSN(page, recptr);
		PageSetSUI(page, ThisStartUpID);
V
Vadim B. Mikheev 已提交
376

B
Bruce Momjian 已提交
377
		if (fetch)				/* not all numbers were fetched */
378
			log -= fetch;
V
Vadim B. Mikheev 已提交
379
	}
380

381
	/* update on-disk data */
V
Vadim B. Mikheev 已提交
382
	seq->last_value = last;		/* last fetched number */
383
	seq->is_called = 't';
V
Vadim B. Mikheev 已提交
384 385
	Assert(log >= 0);
	seq->log_cnt = log;			/* how much is logged */
386
	END_CRIT_SECTION();
387

V
Vadim B. Mikheev 已提交
388 389
	LockBuffer(buf, BUFFER_LOCK_UNLOCK);

390
	if (WriteBuffer(buf) == STATUS_ERROR)
391
		elog(ERROR, "%s.nextval: WriteBuffer failed", elm->name);
392

393
	PG_RETURN_INT32(result);
394 395
}

396 397
Datum
currval(PG_FUNCTION_ARGS)
398
{
399 400
	text	   *seqin = PG_GETARG_TEXT_P(0);
	char	   *seqname = get_seq_name(seqin);
401
	SeqTable	elm;
402 403
	int32		result;

404
	if (pg_aclcheck(seqname, GetUserId(), ACL_SELECT) != ACLCHECK_OK)
405 406
		elog(ERROR, "%s.currval: you don't have permissions to read sequence %s",
			 seqname, seqname);
407

V
Vadim B. Mikheev 已提交
408
	/* open and AccessShareLock sequence */
409 410 411
	elm = init_sequence("currval", seqname);

	if (elm->increment == 0)	/* nextval/read_info were not called */
412 413
		elog(ERROR, "%s.currval is not yet defined in this session",
			 seqname);
414 415 416

	result = elm->last;

417
	pfree(seqname);
418

419
	PG_RETURN_INT32(result);
420 421
}

B
Bruce Momjian 已提交
422
/*
423 424 425 426
 * Main internal procedure that handles 2 & 3 arg forms of SETVAL.
 *
 * Note that the 3 arg version (which sets the is_called flag) is
 * only for use in pg_dump, and setting the is_called flag may not
B
Bruce Momjian 已提交
427
 * work if multiple users are attached to the database and referencing
428 429
 * the sequence (unlikely if pg_dump is restoring it).
 *
B
Bruce Momjian 已提交
430
 * It is necessary to have the 3 arg version so that pg_dump can
431 432 433 434
 * restore the state of a sequence exactly during data-only restores -
 * it is the only way to clear the is_called flag in an existing
 * sequence.
 */
B
Bruce Momjian 已提交
435
static void
436
do_setval(char *seqname, int32 next, bool iscalled)
M
 
Marc G. Fournier 已提交
437 438
{
	SeqTable	elm;
439
	Buffer		buf;
440
	Form_pg_sequence seq;
M
 
Marc G. Fournier 已提交
441

442
	if (pg_aclcheck(seqname, GetUserId(), ACL_UPDATE) != ACLCHECK_OK)
M
 
Marc G. Fournier 已提交
443 444 445
		elog(ERROR, "%s.setval: you don't have permissions to set sequence %s",
			 seqname, seqname);

V
Vadim B. Mikheev 已提交
446
	/* open and AccessShareLock sequence */
447
	elm = init_sequence("setval", seqname);
B
Bruce Momjian 已提交
448 449
	seq = read_info("setval", elm, &buf);		/* lock page' buffer and
												 * read tuple */
M
 
Marc G. Fournier 已提交
450

451
	if ((next < seq->min_value) || (next > seq->max_value))
452
		elog(ERROR, "%s.setval: value %d is out of bounds (%d,%d)",
453
			 seqname, next, seq->min_value, seq->max_value);
M
 
Marc G. Fournier 已提交
454 455 456

	/* save info in local cache */
	elm->last = next;			/* last returned number */
B
Bruce Momjian 已提交
457 458
	elm->cached = next;			/* last cached number (forget cached
								 * values) */
M
 
Marc G. Fournier 已提交
459

460
	START_CRIT_SECTION();
V
Vadim B. Mikheev 已提交
461 462 463
	{
		xl_seq_rec	xlrec;
		XLogRecPtr	recptr;
B
Bruce Momjian 已提交
464
		XLogRecData rdata[2];
465
		Page		page = BufferGetPage(buf);
V
Vadim B. Mikheev 已提交
466 467

		xlrec.node = elm->rel->rd_node;
468
		rdata[0].buffer = InvalidBuffer;
B
Bruce Momjian 已提交
469
		rdata[0].data = (char *) &xlrec;
470 471 472 473 474 475 476
		rdata[0].len = sizeof(xl_seq_rec);
		rdata[0].next = &(rdata[1]);

		seq->last_value = next;
		seq->is_called = 't';
		seq->log_cnt = 0;
		rdata[1].buffer = InvalidBuffer;
B
Bruce Momjian 已提交
477 478 479
		rdata[1].data = (char *) page + ((PageHeader) page)->pd_upper;
		rdata[1].len = ((PageHeader) page)->pd_special -
			((PageHeader) page)->pd_upper;
480 481
		rdata[1].next = NULL;

B
Bruce Momjian 已提交
482
		recptr = XLogInsert(RM_SEQ_ID, XLOG_SEQ_LOG | XLOG_NO_TRAN, rdata);
483 484 485

		PageSetLSN(page, recptr);
		PageSetSUI(page, ThisStartUpID);
V
Vadim B. Mikheev 已提交
486
	}
487 488 489 490
	/* save info in sequence relation */
	seq->last_value = next;		/* last fetched number */
	seq->is_called = iscalled ? 't' : 'f';
	seq->log_cnt = (iscalled) ? 0 : 1;
491
	END_CRIT_SECTION();
M
 
Marc G. Fournier 已提交
492

V
Vadim B. Mikheev 已提交
493 494
	LockBuffer(buf, BUFFER_LOCK_UNLOCK);

495
	if (WriteBuffer(buf) == STATUS_ERROR)
496 497 498 499
		elog(ERROR, "%s.setval: WriteBuffer failed", seqname);

	pfree(seqname);

500 501
}

502 503 504 505
/*
 * Implement the 2 arg setval procedure.
 * See do_setval for discussion.
 */
506 507 508 509 510 511 512 513 514 515 516 517
Datum
setval(PG_FUNCTION_ARGS)
{
	text	   *seqin = PG_GETARG_TEXT_P(0);
	int32		next = PG_GETARG_INT32(1);
	char	   *seqname = get_seq_name(seqin);

	do_setval(seqname, next, true);

	PG_RETURN_INT32(next);
}

518 519 520 521
/*
 * Implement the 3 arg setval procedure.
 * See do_setval for discussion.
 */
522 523 524 525 526 527 528 529 530 531
Datum
setval_and_iscalled(PG_FUNCTION_ARGS)
{
	text	   *seqin = PG_GETARG_TEXT_P(0);
	int32		next = PG_GETARG_INT32(1);
	bool		iscalled = PG_GETARG_BOOL(2);
	char	   *seqname = get_seq_name(seqin);

	do_setval(seqname, next, iscalled);

532 533 534 535 536
	PG_RETURN_INT32(next);
}

/*
 * Given a 'text' parameter to a sequence function, extract the actual
537 538
 * sequence name.  We downcase the name if it's not double-quoted,
 * and truncate it if it's too long.
539 540 541 542 543 544
 *
 * This is a kluge, really --- should be able to write nextval(seqrel).
 */
static char *
get_seq_name(text *seqin)
{
545
	char	   *rawname = DatumGetCString(DirectFunctionCall1(textout,
B
Bruce Momjian 已提交
546
												PointerGetDatum(seqin)));
547 548
	int			rawlen = strlen(rawname);
	char	   *seqname;
M
 
Marc G. Fournier 已提交
549

550 551 552 553 554 555 556 557 558 559 560
	if (rawlen >= 2 &&
		rawname[0] == '\"' && rawname[rawlen - 1] == '\"')
	{
		/* strip off quotes, keep case */
		rawname[rawlen - 1] = '\0';
		seqname = pstrdup(rawname + 1);
		pfree(rawname);
	}
	else
	{
		seqname = rawname;
B
Bruce Momjian 已提交
561

562 563 564 565 566 567
		/*
		 * It's important that this match the identifier downcasing code
		 * used by backend/parser/scan.l.
		 */
		for (; *rawname; rawname++)
		{
568 569
			if (isupper((unsigned char) *rawname))
				*rawname = tolower((unsigned char) *rawname);
570 571
		}
	}
572 573 574 575 576 577 578

	/* Truncate name if it's overlength; again, should match scan.l */
	if (strlen(seqname) >= NAMEDATALEN)
	{
#ifdef MULTIBYTE
		int len;

579
		len = pg_mbcliplen(seqname, rawlen, NAMEDATALEN-1);
580 581 582 583 584 585
		seqname[len] = '\0';
#else
		seqname[NAMEDATALEN-1] = '\0';
#endif
	}

586
	return seqname;
M
 
Marc G. Fournier 已提交
587 588
}

589
static Form_pg_sequence
B
Bruce Momjian 已提交
590
read_info(char *caller, SeqTable elm, Buffer *buf)
591
{
B
Bruce Momjian 已提交
592 593 594
	PageHeader	page;
	ItemId		lp;
	HeapTupleData tuple;
595
	sequence_magic *sm;
B
Bruce Momjian 已提交
596
	Form_pg_sequence seq;
597

598
	if (elm->rel->rd_nblocks > 1)
599
		elog(ERROR, "%s.%s: invalid number of blocks in sequence",
600 601 602 603
			 elm->name, caller);

	*buf = ReadBuffer(elm->rel, 0);
	if (!BufferIsValid(*buf))
604
		elog(ERROR, "%s.%s: ReadBuffer failed", elm->name, caller);
605

V
Vadim B. Mikheev 已提交
606 607
	LockBuffer(*buf, BUFFER_LOCK_EXCLUSIVE);

608 609 610 611
	page = (PageHeader) BufferGetPage(*buf);
	sm = (sequence_magic *) PageGetSpecialPointer(page);

	if (sm->magic != SEQ_MAGIC)
612
		elog(ERROR, "%s.%s: bad magic (%08X)", elm->name, caller, sm->magic);
613 614 615

	lp = PageGetItemId(page, FirstOffsetNumber);
	Assert(ItemIdIsUsed(lp));
616
	tuple.t_data = (HeapTupleHeader) PageGetItem((Page) page, lp);
617

618
	seq = (Form_pg_sequence) GETSTRUCT(&tuple);
619 620 621

	elm->increment = seq->increment_by;

622
	return seq;
623 624 625 626

}


627
static SeqTable
628
init_sequence(char *caller, char *name)
629
{
630
	SeqTable	elm,
631 632
				prev = (SeqTable) NULL;
	Relation	seqrel;
633

634 635
	/* Look to see if we already have a seqtable entry for name */
	for (elm = seqtab; elm != (SeqTable) NULL; elm = elm->next)
636 637 638
	{
		if (strcmp(elm->name, name) == 0)
			break;
639
		prev = elm;
640 641
	}

642 643 644
	/* If so, and if it's already been opened in this xact, just return it */
	if (elm != (SeqTable) NULL && elm->rel != (Relation) NULL)
		return elm;
645

646 647 648
	/* Else open and check it */
	seqrel = heap_openr(name, AccessShareLock);
	if (seqrel->rd_rel->relkind != RELKIND_SEQUENCE)
649
		elog(ERROR, "%s.%s: %s is not a sequence", name, caller, name);
650

651 652
	if (elm != (SeqTable) NULL)
	{
653 654 655

		/*
		 * We are using a seqtable entry left over from a previous xact;
656 657 658 659
		 * must check for relid change.
		 */
		elm->rel = seqrel;
		if (RelationGetRelid(seqrel) != elm->relid)
660 661
		{
			elog(NOTICE, "%s.%s: sequence was re-created",
662
				 name, caller);
663
			elm->relid = RelationGetRelid(seqrel);
664 665 666 667 668
			elm->cached = elm->last = elm->increment = 0;
		}
	}
	else
	{
669 670 671

		/*
		 * Time to make a new seqtable entry.  These entries live as long
672 673 674
		 * as the backend does, so we use plain malloc for them.
		 */
		elm = (SeqTable) malloc(sizeof(SeqTableData));
T
Tom Lane 已提交
675 676 677 678 679
		if (elm == NULL)
			elog(ERROR, "Memory exhausted in init_sequence");
		elm->name = strdup(name);
		if (elm->name == NULL)
			elog(ERROR, "Memory exhausted in init_sequence");
680 681 682 683 684
		elm->rel = seqrel;
		elm->relid = RelationGetRelid(seqrel);
		elm->cached = elm->last = elm->increment = 0;
		elm->next = (SeqTable) NULL;

685 686 687
		if (seqtab == (SeqTable) NULL)
			seqtab = elm;
		else
688
			prev->next = elm;
689 690
	}

691
	return elm;
692 693 694 695
}


/*
B
Bruce Momjian 已提交
696
 * CloseSequences
697
 *				is calling by xact mgr at commit/abort.
698 699
 */
void
700
CloseSequences(void)
701
{
702 703
	SeqTable	elm;
	Relation	rel;
704

705
	for (elm = seqtab; elm != (SeqTable) NULL; elm = elm->next)
706
	{
707 708 709 710
		if (elm->rel != (Relation) NULL)		/* opened in current xact */
		{
			rel = elm->rel;
			elm->rel = (Relation) NULL;
711
			heap_close(rel, AccessShareLock);
712 713
		}
	}
714 715 716
}


717
static void
718
init_params(CreateSeqStmt *seq, Form_pg_sequence new)
719
{
720 721 722 723 724 725
	DefElem    *last_value = NULL;
	DefElem    *increment_by = NULL;
	DefElem    *max_value = NULL;
	DefElem    *min_value = NULL;
	DefElem    *cache_value = NULL;
	List	   *option;
726 727 728 729

	new->is_cycled = 'f';
	foreach(option, seq->options)
	{
730
		DefElem    *defel = (DefElem *) lfirst(option);
731 732 733 734 735 736 737 738 739 740 741 742 743 744

		if (!strcasecmp(defel->defname, "increment"))
			increment_by = defel;
		else if (!strcasecmp(defel->defname, "start"))
			last_value = defel;
		else if (!strcasecmp(defel->defname, "maxvalue"))
			max_value = defel;
		else if (!strcasecmp(defel->defname, "minvalue"))
			min_value = defel;
		else if (!strcasecmp(defel->defname, "cache"))
			cache_value = defel;
		else if (!strcasecmp(defel->defname, "cycle"))
		{
			if (defel->arg != (Node *) NULL)
745
				elog(ERROR, "DefineSequence: CYCLE ??");
746 747 748
			new->is_cycled = 't';
		}
		else
749
			elog(ERROR, "DefineSequence: option \"%s\" not recognized",
750 751 752 753 754 755
				 defel->defname);
	}

	if (increment_by == (DefElem *) NULL)		/* INCREMENT BY */
		new->increment_by = 1;
	else if ((new->increment_by = get_param(increment_by)) == 0)
756
		elog(ERROR, "DefineSequence: can't INCREMENT by 0");
757 758

	if (max_value == (DefElem *) NULL)	/* MAXVALUE */
759
	{
760 761 762 763
		if (new->increment_by > 0)
			new->max_value = SEQ_MAXVALUE;		/* ascending seq */
		else
			new->max_value = -1;/* descending seq */
764
	}
765
	else
766
		new->max_value = get_param(max_value);
767

768
	if (min_value == (DefElem *) NULL)	/* MINVALUE */
769
	{
770 771 772 773
		if (new->increment_by > 0)
			new->min_value = 1; /* ascending seq */
		else
			new->min_value = SEQ_MINVALUE;		/* descending seq */
774
	}
775
	else
776 777 778
		new->min_value = get_param(min_value);

	if (new->min_value >= new->max_value)
779
		elog(ERROR, "DefineSequence: MINVALUE (%d) can't be >= MAXVALUE (%d)",
780 781 782
			 new->min_value, new->max_value);

	if (last_value == (DefElem *) NULL) /* START WITH */
783
	{
784 785 786 787
		if (new->increment_by > 0)
			new->last_value = new->min_value;	/* ascending seq */
		else
			new->last_value = new->max_value;	/* descending seq */
788
	}
789
	else
790 791 792
		new->last_value = get_param(last_value);

	if (new->last_value < new->min_value)
793
		elog(ERROR, "DefineSequence: START value (%d) can't be < MINVALUE (%d)",
794 795
			 new->last_value, new->min_value);
	if (new->last_value > new->max_value)
796
		elog(ERROR, "DefineSequence: START value (%d) can't be > MAXVALUE (%d)",
797 798 799 800 801
			 new->last_value, new->max_value);

	if (cache_value == (DefElem *) NULL)		/* CACHE */
		new->cache_value = 1;
	else if ((new->cache_value = get_param(cache_value)) <= 0)
802
		elog(ERROR, "DefineSequence: CACHE (%d) can't be <= 0",
803
			 new->cache_value);
804 805 806 807

}

static int
808
get_param(DefElem *def)
809
{
810
	if (def->arg == (Node *) NULL)
811
		elog(ERROR, "DefineSequence: \"%s\" value unspecified", def->defname);
812 813

	if (nodeTag(def->arg) == T_Integer)
814
		return intVal(def->arg);
815

816
	elog(ERROR, "DefineSequence: \"%s\" value must be integer", def->defname);
817
	return -1;
818
}
V
Vadim B. Mikheev 已提交
819

B
Bruce Momjian 已提交
820 821
void
seq_redo(XLogRecPtr lsn, XLogRecord *record)
V
Vadim B. Mikheev 已提交
822
{
B
Bruce Momjian 已提交
823 824 825 826 827 828 829
	uint8		info = record->xl_info & ~XLR_INFO_MASK;
	Relation	reln;
	Buffer		buffer;
	Page		page;
	char	   *item;
	Size		itemsz;
	xl_seq_rec *xlrec = (xl_seq_rec *) XLogRecGetData(record);
830
	sequence_magic *sm;
V
Vadim B. Mikheev 已提交
831

832 833
	if (info != XLOG_SEQ_LOG)
		elog(STOP, "seq_redo: unknown op code %u", info);
V
Vadim B. Mikheev 已提交
834 835 836 837 838

	reln = XLogOpenRelation(true, RM_SEQ_ID, xlrec->node);
	if (!RelationIsValid(reln))
		return;

839
	buffer = XLogReadBuffer(true, reln, 0);
V
Vadim B. Mikheev 已提交
840
	if (!BufferIsValid(buffer))
B
Bruce Momjian 已提交
841 842
		elog(STOP, "seq_redo: can't read block of %u/%u",
			 xlrec->node.tblNode, xlrec->node.relNode);
V
Vadim B. Mikheev 已提交
843 844 845

	page = (Page) BufferGetPage(buffer);

846 847 848
	PageInit((Page) page, BufferGetPageSize(buffer), sizeof(sequence_magic));
	sm = (sequence_magic *) PageGetSpecialPointer(page);
	sm->magic = SEQ_MAGIC;
V
Vadim B. Mikheev 已提交
849

B
Bruce Momjian 已提交
850
	item = (char *) xlrec + sizeof(xl_seq_rec);
851 852
	itemsz = record->xl_len - sizeof(xl_seq_rec);
	itemsz = MAXALIGN(itemsz);
B
Bruce Momjian 已提交
853
	if (PageAddItem(page, (Item) item, itemsz,
854 855
					FirstOffsetNumber, LP_USED) == InvalidOffsetNumber)
		elog(STOP, "seq_redo: failed to add item to page");
V
Vadim B. Mikheev 已提交
856 857 858 859 860 861 862 863

	PageSetLSN(page, lsn);
	PageSetSUI(page, ThisStartUpID);
	UnlockAndWriteBuffer(buffer);

	return;
}

B
Bruce Momjian 已提交
864 865
void
seq_undo(XLogRecPtr lsn, XLogRecord *record)
V
Vadim B. Mikheev 已提交
866 867 868
{
}

B
Bruce Momjian 已提交
869 870
void
seq_desc(char *buf, uint8 xl_info, char *rec)
V
Vadim B. Mikheev 已提交
871
{
B
Bruce Momjian 已提交
872 873
	uint8		info = xl_info & ~XLR_INFO_MASK;
	xl_seq_rec *xlrec = (xl_seq_rec *) rec;
V
Vadim B. Mikheev 已提交
874 875 876 877 878 879 880 881 882

	if (info == XLOG_SEQ_LOG)
		strcat(buf, "log: ");
	else
	{
		strcat(buf, "UNKNOWN");
		return;
	}

883
	sprintf(buf + strlen(buf), "node %u/%u",
B
Bruce Momjian 已提交
884
			xlrec->node.tblNode, xlrec->node.relNode);
V
Vadim B. Mikheev 已提交
885
}