blob: c42d21136a2d5daaff4da78be870fed9a462ed70 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
package com.ufund.api.ufundapi.persistence;
import java.io.File;
import java.io.IOException;
import java.util.Map;
import java.util.TreeMap;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.ufund.api.ufundapi.model.User;
@Component
public class UserFileDAO implements UserDAO {
private final Map<String, User> users; // cache
private final ObjectMapper objectMapper;
private final String filename;
public UserFileDAO(@Value("${users.file}") String filename, ObjectMapper objectMapper) throws IOException {
this.filename = filename;
this.objectMapper = objectMapper;
users = new TreeMap<>();
load(); // load the users from the file
}
/**
* Load changes from the json file
*
* @throws IOException Any IO issue with the file
*/
private void load() throws IOException {
users.clear();
User[] usersArray = objectMapper.readValue(new File(filename), User[].class);
for (User user : usersArray) {
users.put(user.getName(), user);
}
}
/**
* Saves the needs to json
*
* @return True if the save was successful, false otherwise
* @throws IOException If there was an IO issue saving the file
*/
private boolean save() throws IOException {
User[] userArray = getUserArray();
objectMapper.writeValue(new File(filename), userArray);
return true;
}
/**
* Return an array of the needs
*
* @return An array of all the needs
*/
private User[] getUserArray() {
return users.values().toArray(User[]::new);
}
@Override
public User[] getUsers() throws IOException {
synchronized (users) {
return getUserArray();
}
}
/**
* Return the user with the String name name or null otherwise
*
* @param name Name of desired user
*
* @return Desired user, null otherwise
* @throws IOException If there was an IO issue saving the file
*/
@Override
public User getUser(String name) throws IOException {
synchronized (users) {
return users.getOrDefault(name, null);
}
}
/**
* Return the user with the String name name or null otherwise
*
* @param name Name of desired user
*
* @return Desired user, null otherwise
* @throws IOException If there was an IO issue saving the file
*/
@Override
public User createUser(User user) throws IOException {
synchronized (users) {
User newUser = new User(user);
users.put(newUser.getName(), newUser);
save();
return newUser;
}
}
@Override
public User updateUser(User user) throws IOException {
synchronized (users) {
if (users.containsKey(user.getName())) {
users.put(user.getName(), user);
save();
return user;
} else {
return null;
}
}
}
@Override
public boolean deleteUser(String name) throws IOException {
synchronized (users) {
if (users.containsKey(name)) {
users.remove(name);
return save();
} else {
return false;
}
}
}
}
|
