Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitbf219ed

Browse files
committed
create_range_partitions() for array of boundaries
1 parent3999694 commitbf219ed

File tree

2 files changed

+144
-4
lines changed

2 files changed

+144
-4
lines changed

‎range.sql

Lines changed: 71 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -317,10 +317,6 @@ BEGIN
317317
attribute :=lower(attribute);
318318
PERFORM @extschema@.common_relation_checks(parent_relid, attribute);
319319

320-
IF p_interval<=0 THEN
321-
RAISE EXCEPTION'interval must be positive';
322-
END IF;
323-
324320
/* Check boundaries*/
325321
PERFORM @extschema@.check_boundaries(parent_relid,
326322
attribute,
@@ -435,6 +431,77 @@ BEGIN
435431
END
436432
$$ LANGUAGE plpgsql;
437433

434+
435+
CREATEOR REPLACE FUNCTION @extschema@.create_range_partitions2(
436+
parent_relidREGCLASS,
437+
attributeTEXT,
438+
boundsANYARRAY)
439+
RETURNSINTEGERAS
440+
$$
441+
DECLARE
442+
part_countINTEGER;
443+
BEGIN
444+
IF array_ndims(bounds)>1 THEN
445+
RAISE EXCEPTION'Bounds array must be a one dimensional array';
446+
END IF;
447+
448+
IF array_length(bounds,1)<2 THEN
449+
RAISE EXCEPTION'Bounds array must have at least two values';
450+
END IF;
451+
452+
PERFORM @extschema@.validate_relname(parent_relid);
453+
454+
IF partition_data= true THEN
455+
/* Acquire data modification lock*/
456+
PERFORM @extschema@.prevent_relation_modification(parent_relid);
457+
ELSE
458+
/* Acquire lock on parent*/
459+
PERFORM @extschema@.lock_partitioned_relation(parent_relid);
460+
END IF;
461+
462+
attribute :=lower(attribute);
463+
PERFORM @extschema@.common_relation_checks(parent_relid, attribute);
464+
465+
/* Check boundaries*/
466+
PERFORM @extschema@.check_boundaries(parent_relid,
467+
attribute,
468+
bounds[0],
469+
bounds[array_length(bounds,1)-1]);
470+
471+
INSERT INTO @extschema@.pathman_config (partrel, attname, parttype, range_interval)
472+
VALUES (parent_relid, attribute,2,NULL);
473+
474+
/* Create sequence for child partitions names*/
475+
PERFORM @extschema@.create_or_replace_sequence(parent_relid)
476+
FROM @extschema@.get_plain_schema_and_relname(parent_relid);
477+
478+
/* Create partitions*/
479+
part_count := @extschema@.create_range_partitions_internal(parent_relid, bounds);
480+
481+
/* Notify backend about changes*/
482+
PERFORM @extschema@.on_create_partitions(parent_relid);
483+
484+
/* Relocate data if asked to*/
485+
IF partition_data= true THEN
486+
PERFORM @extschema@.set_enable_parent(parent_relid, false);
487+
PERFORM @extschema@.partition_data(parent_relid);
488+
ELSE
489+
PERFORM @extschema@.set_enable_parent(parent_relid, true);
490+
END IF;
491+
492+
RETURN0;
493+
END
494+
$$
495+
LANGUAGE plpgsql;
496+
497+
498+
CREATEOR REPLACE FUNCTION @extschema@.create_range_partitions_internal(
499+
parent_relidREGCLASS,
500+
valueANYARRAY)
501+
RETURNS REGCLASSAS'pg_pathman','create_range_partitions_internal'
502+
LANGUAGE C;
503+
504+
438505
/*
439506
* Split RANGE partition
440507
*/

‎src/pl_range_funcs.c

Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,8 @@ PG_FUNCTION_INFO_V1( merge_range_partitions );
6969
PG_FUNCTION_INFO_V1(drop_range_partition_expand_next );
7070
PG_FUNCTION_INFO_V1(validate_interval_value );
7171

72+
PG_FUNCTION_INFO_V1(create_range_partitions_internal );
73+
7274

7375
/*
7476
* -----------------------------
@@ -1007,3 +1009,74 @@ drop_table_by_oid(Oid relid)
10071009

10081010
RemoveRelations(n);
10091011
}
1012+
1013+
1014+
Datum
1015+
create_range_partitions_internal(PG_FUNCTION_ARGS)
1016+
{
1017+
Oidrelid=PG_GETARG_OID(0);
1018+
// char *attname = TextDatumGetCString(PG_GETARG_TEXT_P(1));
1019+
int16typlen;
1020+
booltypbyval;
1021+
chartypalign;
1022+
FmgrInfocmp_func;
1023+
1024+
/* bounds */
1025+
ArrayType*arr=PG_GETARG_ARRAYTYPE_P(1);
1026+
Oidelemtype=ARR_ELEMTYPE(arr);
1027+
Datum*datums;
1028+
bool*nulls;
1029+
intndatums;
1030+
inti;
1031+
1032+
/* Extract bounds */
1033+
get_typlenbyvalalign(elemtype,&typlen,&typbyval,&typalign);
1034+
deconstruct_array(arr,elemtype,
1035+
typlen,typbyval,typalign,
1036+
&datums,&nulls,&ndatums);
1037+
1038+
/* Check if bounds array is ascending */
1039+
fill_type_cmp_fmgr_info(&cmp_func,
1040+
getBaseType(elemtype),
1041+
getBaseType(elemtype));
1042+
for (i=0;i<ndatums-1;i++)
1043+
{
1044+
/*
1045+
* Only first bound can be NULL
1046+
*
1047+
* XXX Probably the last one too...
1048+
*/
1049+
if (nulls[i])
1050+
{
1051+
if (i==0)
1052+
continue;
1053+
else
1054+
elog(ERROR,
1055+
"Only first bound can be NULL");
1056+
}
1057+
1058+
if (DatumGetInt32(FunctionCall2(&cmp_func,datums[i],datums[i+1])) >=0)
1059+
elog(ERROR,
1060+
"Bounds array must be ascending");
1061+
}
1062+
1063+
/* Create partitions */
1064+
for (i=0;i<ndatums-1;i++)
1065+
{
1066+
Boundstart=nulls[i] ?
1067+
MakeBoundInf(MINUS_INFINITY) :
1068+
MakeBound(datums[i]);
1069+
Boundend=nulls[i+1] ?
1070+
MakeBoundInf(PLUS_INFINITY) :
1071+
MakeBound(datums[i+1]);
1072+
1073+
(void)create_single_range_partition_internal(relid,
1074+
&start,
1075+
&end,
1076+
elemtype,
1077+
NULL,
1078+
NULL);
1079+
}
1080+
1081+
PG_RETURN_VOID();
1082+
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp