fix: make initializeClient a pure function (#498)

* fix: make initializeClient a pure function * fix: make initializeClient a pure function
aws · Jul 29, 2024 · c5e2ee3 · c5e2ee3
1 parent 70467ba
commit c5e2ee3
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/...m/amazonaws/services/lambda/runtime/api/client/runtimeapi/LambdaRuntimeApiClientImpl.java b/...m/amazonaws/services/lambda/runtime/api/client/runtimeapi/LambdaRuntimeApiClientImpl.java
@@ -38,7 +38,7 @@ public LambdaRuntimeApiClientImpl(String hostnameAndPort) {
         Objects.requireNonNull(hostnameAndPort, "hostnameAndPort cannot be null");
         this.baseUrl = "http://" + hostnameAndPort;
         this.invocationEndpoint = this.baseUrl + "/2018-06-01/runtime/invocation/";
-        NativeClient.init();
+        NativeClient.init(hostnameAndPort);
     }
 
     @Override

diff --git a/...c/main/java/com/amazonaws/services/lambda/runtime/api/client/runtimeapi/NativeClient.java b/...c/main/java/com/amazonaws/services/lambda/runtime/api/client/runtimeapi/NativeClient.java
@@ -14,12 +14,12 @@
  * interactions with the Runtime API.
  */
 class NativeClient {
-    static void init() {
+    static void init(String awsLambdaRuntimeApi) {
         JniHelper.load();
-        initializeClient(USER_AGENT.getBytes());
+        initializeClient(USER_AGENT.getBytes(), awsLambdaRuntimeApi.getBytes());
     }
 
-    static native void initializeClient(byte[] userAgent);
+    static native void initializeClient(byte[] userAgent, byte[] awsLambdaRuntimeApi);
 
     static native InvocationRequest next();
 

diff --git a/...src/main/jni/com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient.cpp b/...src/main/jni/com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient.cpp
@@ -69,9 +69,9 @@ static std::string toNativeString(JNIEnv *env, jbyteArray jArray) {
   return nativeString;
 }
 
-JNIEXPORT void JNICALL Java_com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient_initializeClient(JNIEnv *env, jobject thisObject, jbyteArray userAgent) {
+JNIEXPORT void JNICALL Java_com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient_initializeClient(JNIEnv *env, jobject thisObject, jbyteArray userAgent, jbyteArray awsLambdaRuntimeApi) {
   std::string user_agent = toNativeString(env, userAgent);
-  std::string endpoint(getenv("AWS_LAMBDA_RUNTIME_API"));
+  std::string endpoint = toNativeString(env, awsLambdaRuntimeApi);
   CLIENT = new aws::lambda_runtime::runtime(endpoint, user_agent);
 }
 

diff --git a/...t/src/main/jni/com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient.h b/...t/src/main/jni/com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient.h
@@ -11,7 +11,7 @@ extern "C" {
 #endif
 
 JNIEXPORT void JNICALL Java_com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient_initializeClient
-  (JNIEnv *, jobject, jbyteArray);
+  (JNIEnv *, jobject, jbyteArray, jbyteArray);
 
 JNIEXPORT jobject JNICALL Java_com_amazonaws_services_lambda_runtime_api_client_runtimeapi_NativeClient_next
   (JNIEnv *, jobject);