Skip to content

Commit

Permalink
Amendments for example
Browse files Browse the repository at this point in the history
  • Loading branch information
tmdiep committed Jul 2, 2021
1 parent def076e commit dfead22
Showing 1 changed file with 9 additions and 10 deletions.
19 changes: 9 additions & 10 deletions samples/snippets/seek_lite_subscription_example.py
Expand Up @@ -14,7 +14,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

"""This application demonstrates how to invoke an out-of-band seek for a
"""This application demonstrates how to initiate an out-of-band seek for a
subscription with the Pub/Sub Lite API.
"""

Expand All @@ -24,7 +24,7 @@
def seek_lite_subscription(project_number, cloud_region, zone_id, subscription_id, target, wait_for_operation):
# [START pubsublite_seek_subscription]
from datetime import datetime
from google.api_core.exceptions import NotFound, GoogleAPICallError
from google.api_core.exceptions import NotFound
from google.cloud.pubsublite import AdminClient
from google.cloud.pubsublite.types import CloudRegion, CloudZone, SubscriptionPath, BacklogLocation, PublishTime

Expand All @@ -34,7 +34,7 @@ def seek_lite_subscription(project_number, cloud_region, zone_id, subscription_i
# zone_id = "a"
# subscription_id = "your-subscription-id"
# target = "BEGINNING"
# wait_for_operation = 1
# wait_for_operation = True

cloud_region = CloudRegion(cloud_region)
location = CloudZone(cloud_region, zone_id)
Expand All @@ -54,16 +54,14 @@ def seek_lite_subscription(project_number, cloud_region, zone_id, subscription_i
print(f"Metadata:\n{seek_operation.metadata}")
except NotFound:
print(f"{subscription_path} not found.")
return

# Note: In order for the operation to complete, a subscriber must be
# Note: In order for the operation to complete, subscribers must be
# receiving messages for the subscription.
if wait_for_operation:
print("Waiting for operation to complete...")
try:
seek_operation.result()
print(f"Operation completed. Metadata:\n{seek_operation.metadata}")
except GoogleAPICallError as e:
print(e)
unused = seek_operation.result()
print(f"Operation completed. Metadata:\n{seek_operation.metadata}")
# [END pubsublite_seek_subscription]


Expand All @@ -81,5 +79,6 @@ def seek_lite_subscription(project_number, cloud_region, zone_id, subscription_i
args = parser.parse_args()

seek_lite_subscription(
args.project_number, args.cloud_region, args.zone_id, args.subscription_id, args.target, args.wait_for_operation
args.project_number, args.cloud_region, args.zone_id,
args.subscription_id, args.target, args.wait_for_operation
)

0 comments on commit dfead22

Please sign in to comment.