Skip to content

Commit 29f11b5

Browse files
Sagar-S-RCopilot
andauthored
Update caching/src/main/java/com/iluwatar/caching/database/MongoDb.java
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 18b076e commit 29f11b5

1 file changed

Lines changed: 12 additions & 7 deletions

File tree

  • caching/src/main/java/com/iluwatar/caching/database

caching/src/main/java/com/iluwatar/caching/database/MongoDb.java

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,24 +45,29 @@
4545
public class MongoDb implements DbManager {
4646
private static final String DATABASE_NAME = "admin";
4747
private static final String MONGO_USER = getEnvOrDefault("MONGO_USER", "root");
48-
private static final String MONGO_PASSWORD = getEnvOrDefault("MONGO_PASSWORD", "rootpassword");
48+
private static final String MONGO_PASSWORD = getRequiredEnv("MONGO_PASSWORD");
4949
private MongoClient client;
5050
private MongoDatabase db;
5151

52-
/**
53-
* Helper method to get environment variable with fallback. In production, MONGO_PASSWORD MUST be
54-
* set as environment variable.
55-
*/
52+
/** Helper method to get environment variable with fallback for non-secret values. */
5653
private static String getEnvOrDefault(String key, String defaultValue) {
5754
String value = System.getenv(key);
5855
if (value == null || value.isEmpty()) {
59-
LOGGER.warn(
60-
"Environment variable {} not set. Using default (NOT RECOMMENDED FOR PRODUCTION)", key);
56+
LOGGER.warn("Environment variable {} not set. Using default value {}", key, defaultValue);
6157
return defaultValue;
6258
}
6359
return value;
6460
}
6561

62+
/** Helper method to get required environment variables for secrets. */
63+
private static String getRequiredEnv(String key) {
64+
String value = System.getenv(key);
65+
if (value == null || value.isEmpty()) {
66+
throw new IllegalStateException(
67+
String.format("Required environment variable %s is not set or empty", key));
68+
}
69+
return value;
70+
}
6671
void setDb(MongoDatabase db) {
6772
this.db = db;
6873
}

0 commit comments

Comments
 (0)