Skip to content

Commit c7a5416

Browse files
committed
Use language enum instead of hardcoded 'java'
1 parent 747d0b8 commit c7a5416

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

sebs/benchmark.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,8 @@ def query_cache(self):
249249
self._is_cached_valid = False
250250

251251
def copy_code(self, output_dir):
252+
from sebs.faas.function import Language
253+
252254
FILES = {
253255
"python": ["*.py", "requirements.txt*"],
254256
"nodejs": ["*.js", "package.json"],
@@ -261,7 +263,7 @@ def copy_code(self, output_dir):
261263
shutil.copy2(os.path.join(path, f), output_dir)
262264

263265
# copy src folder of java (java benchmarks are maven project and need directories)
264-
if self.language_name == "java":
266+
if self.language == Language.JAVA:
265267
output_src_dir = os.path.join(output_dir, "src")
266268

267269
if os.path.exists(output_src_dir):
@@ -277,8 +279,8 @@ def copy_code(self, output_dir):
277279

278280
#This is for making jar file and add it to docker directory
279281
def add_java_output(self, code_dir):
280-
281-
if self.language_name == "java":
282+
from sebs.faas.function import Language
283+
if self.language == Language.JAVA:
282284

283285
# Step 1: Move Main.java o src directory
284286
src_dir = os.path.join(code_dir, "src", "main", "java")

0 commit comments

Comments
 (0)