//snippet-sourcedescription:[SubscribeLambda.java demonstrates how to send a confirmation message to a lambda function.]
//snippet-keyword:[Java]
//snippet-sourcesyntax:[java]
//snippet-keyword:[Code Sample]
//snippet-keyword:[Amazon Simple Notification Service]
//snippet-service:[sns]
//snippet-sourcetype:[full-example]
//snippet-sourcedate:[2019-07-20]
//snippet-sourceauthor:[scmacdon AWS]
/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */
//snippet-start:[sns.java2.SubscribeLambda.complete]

package com.example.sns;

//snippet-start:[sns.java2.SubscribeLambda.import]
import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.sns.SnsClient;
import software.amazon.awssdk.services.sns.model.SnsException;
import software.amazon.awssdk.services.sns.model.SubscribeRequest;
import software.amazon.awssdk.services.sns.model.SubscribeResponse;
//snippet-end:[sns.java2.SubscribeLambda.import]

public class SubscribeLambda {

    public static void main(String[] args) {
        final String USAGE = "\n" +
                "SubscribeLambda - subscribe an Amazon Lambda Function.\n" +
                "Usage: SubscribeLambda <topicArn> <lambdaArn>\n\n" +
                "Where:\n" +
                "  topicArn - the arn of the topic to subscribe.\n\n" +
                "  lambdaArn - the ARN of an AWS Lambda function.\n\n";

       if (args.length < 2) {
           System.out.println(USAGE);
           System.exit(1);
        }

        //snippet-start:[sns.java2.SubscribeLambda.main]
        String topicArn = args[0];
        String lambdaArn = args[1];

        SnsClient snsClient = SnsClient.builder()
                .region(Region.US_WEST_2)
                .build();

        try {

            SubscribeRequest request = SubscribeRequest.builder()
                .protocol("lambda")
                .endpoint(lambdaArn)
                .returnSubscriptionArn(true)
                .topicArn(topicArn)
                .build();

            SubscribeResponse result = snsClient.subscribe(request);

            System.out.println("Subscription ARN: " + result.subscriptionArn() + "\n\n Status was " + result.sdkHttpResponse().statusCode());


         } catch (SnsException e) {
            System.err.println(e.awsErrorDetails().errorMessage());
        System.exit(1);
    }
     //snippet-end:[sns.java2.SubscribeLambda.main]
    }
}
//snippet-end:[sns.java2.SubscribeLambda.complete]

@@ -966,61 +966,8 @@
 %0A */
-%0A//snippet-start:%5Bsns.java2.SubscribeLambda.complete%5D
 %0A%0Apa
@@ -1830,16 +1830,17 @@
 %0A       
+ 
 if (args
@@ -1850,24 +1850,25 @@
 ngth %3C 2) %7B%0A
+ 
            S
@@ -1896,32 +1896,33 @@
 GE);%0A           
+ 
 System.exit(1);%0A
@@ -1936,65 +1936,8 @@
  %7D%0A%0A
-        //snippet-start:%5Bsns.java2.SubscribeLambda.main%5D%0A
     
@@ -2115,32 +2115,317 @@
      .build();%0A%0A
+        String arnValue = subLambda(snsClient, topicArn, lambdaArn) ;%0A        System.out.println(%22Subscription ARN: %22 + arnValue);%0A    %7D%0A%0A    //snippet-start:%5Bsns.java2.SubscribeLambda.main%5D%0A    public static String subLambda(SnsClient snsClient, String topicArn, String lambdaArn) %7B%0A%0A
         try %7B%0A%0A 
@@ -2756,135 +2756,38 @@
     
-System.out.println(%22Subscription ARN: %22 + result.subscriptionArn() + %22%5Cn%5Cn Status was %22 + result.sdkHttpResponse().statusCode()
+return result.subscriptionArn(
 );%0A%0A
@@ -2919,17 +2919,40 @@
 1);%0A    
-%7D
+    %7D%0A        return %22%22;
 %0A     //
@@ -3008,55 +3008,4 @@
 %7D%0A%7D%0A
-//snippet-end:%5Bsns.java2.SubscribeLambda.complete%5D%0A

Update SubscribeLambda.java