|
| 1 | +/*------------------------------------------------------------------------- |
| 2 | + * |
| 3 | + * pglogical_relid_map.c |
| 4 | + * Logical Replication map of local Oids to to remote |
| 5 | + * |
| 6 | + * Copyright (c) 2012-2015, PostgreSQL Global Development Group |
| 7 | + * |
| 8 | + * IDENTIFICATION |
| 9 | + * pglogical_relid_map.c |
| 10 | + * |
| 11 | + *------------------------------------------------------------------------- |
| 12 | + */ |
| 13 | +#include"postgres.h" |
| 14 | +#include"utils/hsearch.h" |
| 15 | +#include"pglogical_relid_map.h" |
| 16 | + |
| 17 | +staticHTAB*relid_map; |
| 18 | + |
| 19 | +staticvoid |
| 20 | +pglogical_relid_map_init(void) |
| 21 | +{ |
| 22 | +HASHCTLctl; |
| 23 | +inthash_flags=HASH_ELEM; |
| 24 | + |
| 25 | +Assert(relid_map==NULL); |
| 26 | + |
| 27 | +MemSet(&ctl,0,sizeof(ctl)); |
| 28 | +ctl.keysize=sizeof(Oid); |
| 29 | +ctl.entrysize=sizeof(PGLRelidMapEntry); |
| 30 | + |
| 31 | +#ifPG_VERSION_NUM >=90500 |
| 32 | +hash_flags |=HASH_BLOBS; |
| 33 | +#else |
| 34 | +ctl.hash=tag_hash; |
| 35 | +hash_flags |=HASH_FUNCTION; |
| 36 | +#endif |
| 37 | + |
| 38 | +relid_map=hash_create("pglogical_relid_map",PGL_INIT_RELID_MAP_SIZE,&ctl,hash_flags); |
| 39 | + |
| 40 | +Assert(relid_map!=NULL); |
| 41 | +} |
| 42 | + |
| 43 | +Oidpglogical_relid_map_get(Oidrelid) |
| 44 | +{ |
| 45 | +if (relid_map!=NULL) { |
| 46 | +PGLRelidMapEntry*entry= (PGLRelidMapEntry*)hash_search(relid_map,&relid,HASH_FIND,NULL); |
| 47 | +returnentry ?entry->local_relid :InvalidOid; |
| 48 | +} |
| 49 | +returnInvalidOid; |
| 50 | +} |
| 51 | + |
| 52 | +boolpglogical_relid_map_put(Oidremote_relid,Oidlocal_relid) |
| 53 | +{ |
| 54 | +boolfound; |
| 55 | +PGLRelidMapEntry*entry; |
| 56 | +if (relid_map==NULL) { |
| 57 | +pglogical_relid_map_init(); |
| 58 | + } |
| 59 | +entry=hash_search(relid_map,&remote_relid,HASH_ENTER,&found); |
| 60 | +if (found) { |
| 61 | +Assert(entry->local_relid==local_relid); |
| 62 | +return false; |
| 63 | + } |
| 64 | +entry->local_relid=local_relid; |
| 65 | +return true; |
| 66 | +} |