Update from https://github.com/cloudimpl/next-coder/commit/491c51f6ed4234faf7cffb88f48d5a182d2bf696

Author: CloudimplBot

polycode/client.go

@@ -232,6 +232,15 @@ type IncrementCounterRequest struct {
 	TTL   int64  `json:"TTL"`
 }
 
+type AcquireLockRequest struct {
+	Key string `json:"key"`
+	TTL int64  `json:"TTL"`
+}
+
+type ReleaseLockRequest struct {
+	Key string `json:"key"`
+}
+
 type IncrementCounterResponse struct {
 	Value       uint64 `json:"value"`
 	Incremented bool   `json:"incremented"`
@@ -420,6 +429,14 @@ func (sc *ServiceClient) EmitRealtimeEvent(sessionId string, req RealtimeEventEm
 	return executeApiWithoutResponse(sc.httpClient, sc.baseURL, sessionId, "v1/realtime/event/emit", req)
 }
 
+func (sc *ServiceClient) AcquireLock(sessionId string, req AcquireLockRequest) error {
+	return executeApiWithoutResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/lock/acquire", req)
+}
+
+func (sc *ServiceClient) ReleaseLock(sessionId string, req ReleaseLockRequest) error {
+	return executeApiWithoutResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/lock/release", req)
+}
+
 func (sc *ServiceClient) IncrementCounter(sessionId string, req IncrementCounterRequest) (IncrementCounterResponse, error) {
 	var res IncrementCounterResponse
 	err := executeApiWithResponse(sc.httpClient, sc.baseURL, sessionId, "v1/utils/counter/increment", req, &res)

polycode/context.go

@@ -38,6 +38,7 @@ type WorkflowContext interface {
 	Memo(getter func() (any, error)) Response
 	Signal(signalName string) Signal
 	ClientChannel(channelName string) ClientChannel
+	Lock(key string) Lock
 }
 
 type ApiContext interface {
@@ -178,6 +179,14 @@ func (s ContextImpl) ClientChannel(channelName string) ClientChannel {
 	}
 }
 
+func (s ContextImpl) Lock(key string) Lock {
+	return Lock{
+		client:    s.serviceClient,
+		sessionId: s.sessionId,
+		key:       key,
+	}
+}
+
 func (s ContextImpl) Counter(group string, name string, ttl int64) Counter {
 	return Counter{
 		client:    s.serviceClient,

polycode/lock.go

@@ -0,0 +1,28 @@
+package polycode
+
+import "time"
+
+type Lock struct {
+	client    *ServiceClient
+	sessionId string
+	key       string
+}
+
+func (l *Lock) Acquire(expireIn time.Duration) error {
+	ttl := time.Now().Unix() + int64(expireIn.Seconds())
+
+	req := AcquireLockRequest{
+		Key: l.key,
+		TTL: ttl,
+	}
+
+	return l.client.AcquireLock(l.sessionId, req)
+}
+
+func (l *Lock) Release() error {
+	req := ReleaseLockRequest{
+		Key: l.key,
+	}
+
+	return l.client.ReleaseLock(l.sessionId, req)
+}