2
peternameyakj
2024-08-14 9bcf85f8d8f9b503e386a67924f5943cd77bd813
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
133
134
135
136
137
138
139
package project.item.internal;
 
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
 
import org.apache.commons.collections.CollectionUtils;
 
import com.mysql.cj.util.StringUtils;
 
import kernel.web.ApplicationUtil;
import project.data.DataService;
import project.data.model.Realtime;
import project.item.ItemRedisKeys;
import project.item.ItemUserOptionalService;
import project.item.model.ItemUserOptional;
import project.redis.RedisHandler;
 
public class ItemUserOptionalServiceImpl implements ItemUserOptionalService {
 
    private RedisHandler redisHandler;
    
    private DataService dataService;
 
    @Override
    public List<ItemUserOptional> cacheListByPartyId(String partyId) {
        Map<String, ItemUserOptional> map = (Map<String, ItemUserOptional>) redisHandler.get(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + partyId);
        return map == null ? null : new ArrayList<ItemUserOptional>(map.values());
    }
 
    public List<Map<String, Object>> cacheListDataByPartyId(String partyId, String symbol) {
        List<Map<String, Object>> result = new ArrayList<Map<String, Object>>();
        List<ItemUserOptional> list = new ArrayList<>();
        List<ItemUserOptional> lists = cacheListByPartyId(partyId);
        if (!StringUtils.isNullOrEmpty(symbol)) {
            for (ItemUserOptional op : lists) {
                if (op.getSymbol().equals(symbol)) {
                    list.add(op);
                }
            }
        }else {
            list = lists;
        }
        
        if (CollectionUtils.isEmpty(list)) return result;
 
        List<String> symbols = new ArrayList<String>();
        for (ItemUserOptional itemUserOptional : list) {
            symbols.add(itemUserOptional.getSymbol());
        }
        if (CollectionUtils.isEmpty(symbols))
            return new ArrayList<Map<String, Object>>();
 
        List<Realtime> realtimes = dataService.realtime(String.join(",", symbols));
        for (Realtime realtime : realtimes) {
            Map<String, Object> bind = bind(realtime);
            result.add(bind);
        }
        return result;
    }
 
    public Map<String, Object> bind(Realtime realtime) {
        Map<String, Object> map = new HashMap<String, Object>();
        map.put("amount", String.valueOf(realtime.getAmount()));
        map.put("change_ratio", realtime.getChange_ratio());
        map.put("close", String.valueOf(realtime.getClose()));
        map.put("current_time", realtime.getCurrent_time());
        map.put("high", String.valueOf(realtime.getHigh()));
        map.put("low", String.valueOf(realtime.getLow()));
        map.put("name", realtime.getName());
        map.put("open", String.valueOf(realtime.getOpen()));
        map.put("symbol", String.valueOf(realtime.getSymbol()));
        map.put("ts", realtime.getTs());
        map.put("volume", String.valueOf(realtime.getVolume()));
        return map;
    }
 
    @Override
    public void update(ItemUserOptional entity) {
        ApplicationUtil.executeUpdate(entity, "WHERE UUID=?", new Object[] {entity.getId()});
        
        Map<String, ItemUserOptional> map = (Map<String, ItemUserOptional>) redisHandler.get(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + entity.getPartyId().toString());
        
        if (map == null) map = new ConcurrentHashMap<String, ItemUserOptional>();
        
        map.put(entity.getSymbol(), entity);
        redisHandler.setSync(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + entity.getPartyId().toString(), map);
    }
 
    @Override
    public void save(ItemUserOptional entity) {
        Map<String, ItemUserOptional> map = (Map<String, ItemUserOptional>) redisHandler.get(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + entity.getPartyId().toString());
        
        if (map == null) {
            map = new ConcurrentHashMap<String, ItemUserOptional>();
        } else if (map.containsKey(entity.getSymbol())) { // 已经添加的则直接返回
            return;
        }
        
        entity.setId(ApplicationUtil.getCurrentTimeUUID());
        ApplicationUtil.executeInsert(entity);
        
        map.put(entity.getSymbol(), entity);
        redisHandler.setSync(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + entity.getPartyId().toString(), map);
    }
 
    public void delete(String partyId, String symbol) {
        Map<String, ItemUserOptional> map = (Map<String, ItemUserOptional>) redisHandler.get(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + partyId);
 
        if (null!=map && !map.isEmpty()) {
            ItemUserOptional itemUserOptional = map.remove(symbol);
            if (itemUserOptional == null) return;
            
            ApplicationUtil.executeDelete(ItemUserOptional.class, "WHERE UUID=?", new Object[] {itemUserOptional.getId()});
            redisHandler.setSync(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + partyId, map);
        }
    }
 
    public void delete(ItemUserOptional entity) {
        ApplicationUtil.executeDelete(ItemUserOptional.class, "WHERE UUID=?", new Object[] {entity.getId()});
        
        Map<String, ItemUserOptional> map = (Map<String, ItemUserOptional>) redisHandler.get(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + entity.getPartyId().toString());
 
        if (null!=map && !map.isEmpty()) {
            map.remove(entity.getSymbol());
            redisHandler.setSync(ItemRedisKeys.ITEM_USER_OPTIONAL_MAP_PARTY_ID + entity.getPartyId().toString(), map);
        }
    }
 
    public void setRedisHandler(RedisHandler redisHandler) {
        this.redisHandler = redisHandler;
    }
 
    public void setDataService(DataService dataService) {
        this.dataService = dataService;
    }
}