|
| 1 | +/* |
| 2 | + * |
| 3 | + */ |
| 4 | + |
| 5 | +#include"postgres.h" |
| 6 | + |
| 7 | +#include"storage/shmem.h" |
| 8 | + |
| 9 | +#include"aqo_shared.h" |
| 10 | + |
| 11 | +shmem_startup_hook_typeprev_shmem_startup_hook=NULL; |
| 12 | +staticAQOSharedState*aqo_state=NULL; |
| 13 | +unsigned longtemp_storage_size=1024*1024;/* Storage size, in bytes */ |
| 14 | +void*temp_storage=NULL; |
| 15 | + |
| 16 | +staticvoid |
| 17 | +attach_dsm_segment(void) |
| 18 | +{ |
| 19 | +dsm_segment*seg; |
| 20 | + |
| 21 | +LWLockAcquire(&aqo_state->lock,LW_EXCLUSIVE); |
| 22 | + |
| 23 | +if (aqo_state->dsm_handler!=DSM_HANDLE_INVALID) |
| 24 | +{ |
| 25 | +seg=dsm_attach(aqo_state->dsm_handler); |
| 26 | +} |
| 27 | +else |
| 28 | +{ |
| 29 | +seg=dsm_create(temp_storage_size,0); |
| 30 | +aqo_state->dsm_handler=dsm_segment_handle(seg); |
| 31 | +} |
| 32 | + |
| 33 | +temp_storage=dsm_segment_address(seg); |
| 34 | +LWLockRelease(&aqo_state->lock); |
| 35 | +} |
| 36 | + |
| 37 | +staticvoid |
| 38 | +aqo_detach_shmem(intcode,Datumarg) |
| 39 | +{ |
| 40 | +dsm_handlehandler=*(dsm_handle*)arg; |
| 41 | +dsm_detach(dsm_find_mapping(handler)); |
| 42 | +} |
| 43 | + |
| 44 | +void |
| 45 | +aqo_init_shmem(void) |
| 46 | +{ |
| 47 | +boolfound; |
| 48 | + |
| 49 | +LWLockAcquire(AddinShmemInitLock,LW_EXCLUSIVE); |
| 50 | +aqo_state=ShmemInitStruct("aqo",sizeof(AQOSharedState),&found); |
| 51 | +if (!found) |
| 52 | +{ |
| 53 | +/* First time through ... */ |
| 54 | +LWLockInitialize(&aqo_state->lock,LWLockNewTrancheId()); |
| 55 | +aqo_state->dsm_handler=DSM_HANDLE_INVALID; |
| 56 | +} |
| 57 | +LWLockRelease(AddinShmemInitLock); |
| 58 | + |
| 59 | +LWLockRegisterTranche(aqo_state->lock.tranche,"aqo"); |
| 60 | +on_shmem_exit(aqo_detach_shmem, (Datum)&aqo_state->dsm_handler); |
| 61 | +} |