优化拉流代理的添加

648540858

2023-07-14 a9f88be8c592664332de2587f9304e371ab34192

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 com.genersoft.iot.vmp.gb28181.session; 
 
import com.genersoft.iot.vmp.common.InviteSessionType; 
import com.genersoft.iot.vmp.common.VideoManagerConstants; 
import com.genersoft.iot.vmp.conf.UserSetting; 
import com.genersoft.iot.vmp.gb28181.bean.SipTransactionInfo; 
import com.genersoft.iot.vmp.gb28181.bean.SsrcTransaction; 
import com.genersoft.iot.vmp.utils.JsonUtil; 
import com.genersoft.iot.vmp.utils.redis.RedisUtil; 
import gov.nist.javax.sip.message.SIPResponse; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.data.redis.core.RedisTemplate; 
import org.springframework.stereotype.Component; 
import org.springframework.util.ObjectUtils; 
 
import java.util.ArrayList; 
import java.util.List; 
 
/** 
 * 视频流session管理器，管理视频预览、预览回放的通信句柄 
 */ 
@Component 
public class VideoStreamSessionManager { 
 
    @Autowired 
    private UserSetting userSetting; 
 
    @Autowired 
    private RedisTemplate<Object, Object> redisTemplate; 
 
    /** 
     * 添加一个点播/回放的事务信息 
     * 后续可以通过流Id/callID 
     * @param deviceId 设备ID 
     * @param channelId 通道ID 
     * @param callId 一次请求的CallID 
     * @param stream 流名称 
     * @param mediaServerId 所使用的流媒体ID 
     * @param response 回复 
     */ 
    public void put(String deviceId, String channelId, String callId, String stream, String ssrc, String mediaServerId, SIPResponse response, InviteSessionType type){ 
        SsrcTransaction ssrcTransaction = new SsrcTransaction(); 
        ssrcTransaction.setDeviceId(deviceId); 
        ssrcTransaction.setChannelId(channelId); 
        ssrcTransaction.setStream(stream); 
        ssrcTransaction.setSipTransactionInfo(new SipTransactionInfo(response)); 
        ssrcTransaction.setCallId(callId); 
        ssrcTransaction.setSsrc(ssrc); 
        ssrcTransaction.setMediaServerId(mediaServerId); 
        ssrcTransaction.setType(type); 
 
        redisTemplate.opsForValue().set(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() 
                + "_" +  deviceId + "_" + channelId + "_" + callId + "_" + stream, ssrcTransaction); 
    } 
 
    public SsrcTransaction getSsrcTransaction(String deviceId, String channelId, String callId, String stream){ 
 
        if (ObjectUtils.isEmpty(deviceId)) { 
            deviceId ="*"; 
        } 
        if (ObjectUtils.isEmpty(channelId)) { 
            channelId ="*"; 
        } 
        if (ObjectUtils.isEmpty(callId)) { 
            callId ="*"; 
        } 
        if (ObjectUtils.isEmpty(stream)) { 
            stream ="*"; 
        } 
        String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_" + deviceId + "_" + channelId + "_" + callId+ "_" + stream; 
        List<Object> scanResult = RedisUtil.scan(redisTemplate, key); 
        if (scanResult.size() == 0) { 
            return null; 
        } 
        return (SsrcTransaction)redisTemplate.opsForValue().get(scanResult.get(0)); 
    } 
 
    public List<SsrcTransaction> getSsrcTransactionForAll(String deviceId, String channelId, String callId, String stream){ 
        if (ObjectUtils.isEmpty(deviceId)) { 
            deviceId ="*"; 
        } 
        if (ObjectUtils.isEmpty(channelId)) { 
            channelId ="*"; 
        } 
        if (ObjectUtils.isEmpty(callId)) { 
            callId ="*"; 
        } 
        if (ObjectUtils.isEmpty(stream)) { 
            stream ="*"; 
        } 
        String key = VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_" + deviceId + "_" + channelId + "_" + callId+ "_" + stream; 
        List<Object> scanResult = RedisUtil.scan(redisTemplate, key); 
        if (scanResult.size() == 0) { 
            return null; 
        } 
        List<SsrcTransaction> result = new ArrayList<>(); 
        for (Object keyObj : scanResult) { 
            result.add((SsrcTransaction)redisTemplate.opsForValue().get(keyObj)); 
        } 
        return result; 
    } 
 
    public String getMediaServerId(String deviceId, String channelId, String stream){ 
        SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream); 
        if (ssrcTransaction == null) { 
            return null; 
        } 
        return ssrcTransaction.getMediaServerId(); 
    } 
 
    public String getSSRC(String deviceId, String channelId, String stream){ 
        SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream); 
        if (ssrcTransaction == null) { 
            return null; 
        } 
        return ssrcTransaction.getSsrc(); 
    } 
     
    public void remove(String deviceId, String channelId, String stream) { 
        SsrcTransaction ssrcTransaction = getSsrcTransaction(deviceId, channelId, null, stream); 
        if (ssrcTransaction == null) { 
            return; 
        } 
        redisTemplate.delete(VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX + userSetting.getServerId() + "_" 
                +  deviceId + "_" + channelId + "_" + ssrcTransaction.getCallId() + "_" + ssrcTransaction.getStream()); 
    } 
 
 
    public List<SsrcTransaction> getAllSsrc() { 
        List<Object> ssrcTransactionKeys = RedisUtil.scan(redisTemplate, String.format("%s_*_*_*_*", VideoManagerConstants.MEDIA_TRANSACTION_USED_PREFIX+ userSetting.getServerId())); 
        List<SsrcTransaction> result= new ArrayList<>(); 
        for (Object ssrcTransactionKey : ssrcTransactionKeys) { 
            String key = (String) ssrcTransactionKey; 
            SsrcTransaction ssrcTransaction = JsonUtil.redisJsonToObject(redisTemplate, key, SsrcTransaction.class); 
            result.add(ssrcTransaction); 
        } 
        return result; 
    } 
}