X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=lib%2FPublicInbox%2FLock.pm;h=0ee2a8bd60bd964a13c0c24f958f43e0ca9a6b5b;hb=7d2e572aca7297ea2015d2b6e7c71b672521ec82;hp=7fd1774584c3df93b80ebad3f2e6f5ad4bdc8229;hpb=fa7bab31c9e5ded5c832b42de56f48dbdf9de972;p=public-inbox.git
diff --git a/lib/PublicInbox/Lock.pm b/lib/PublicInbox/Lock.pm
index 7fd17745..0ee2a8bd 100644
--- a/lib/PublicInbox/Lock.pm
+++ b/lib/PublicInbox/Lock.pm
@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2020 all contributors
+# Copyright (C) 2018-2021 all contributors
# License: AGPL-3.0+
# Base class for per-inbox locking
@@ -16,7 +16,7 @@ sub lock_acquire {
my $lock_path = $self->{lock_path};
croak 'already locked '.($lock_path // '(undef)') if $self->{lockfh};
return unless defined($lock_path);
- sysopen(my $lockfh, $lock_path, O_WRONLY|O_CREAT) or
+ sysopen(my $lockfh, $lock_path, O_RDWR|O_CREAT) or
croak "failed to open $lock_path: $!\n";
flock($lockfh, LOCK_EX) or croak "lock $lock_path failed: $!\n";
$self->{lockfh} = $lockfh;
@@ -35,9 +35,26 @@ sub lock_release {
# caller must use return value
sub lock_for_scope {
- my ($self) = @_;
- $self->lock_acquire;
- PublicInbox::OnDestroy->new(\&lock_release, $self);
+ my ($self, @single_pid) = @_;
+ lock_acquire($self) or return; # lock_path not set
+ PublicInbox::OnDestroy->new(@single_pid, \&lock_release, $self);
+}
+
+sub lock_acquire_fast {
+ $_[0]->{lockfh} or return lock_acquire($_[0]);
+ flock($_[0]->{lockfh}, LOCK_EX) or croak "lock (fast) failed: $!";
+}
+
+sub lock_release_fast {
+ flock($_[0]->{lockfh} // return, LOCK_UN) or
+ croak "unlock (fast) $_[0]->{lock_path}: $!";
+}
+
+# caller must use return value
+sub lock_for_scope_fast {
+ my ($self, @single_pid) = @_;
+ lock_acquire_fast($self) or return; # lock_path not set
+ PublicInbox::OnDestroy->new(@single_pid, \&lock_release_fast, $self);
}
1;