IM服务端

hefeixia

2021-02-18 5b8c95c760840f09910730943b21391e47187315

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
package chat.message;
 
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicInteger;
 
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
 
public class MessageRunner {
 
    private static final Logger LOG = LoggerFactory.getLogger(MessageRunner.class);
    //private ScheduledExecutorService mScheduledExecutorService = Executors.newScheduledThreadPool(10);
    private ScheduledExecutorService executor;
    private final AtomicInteger runCounter;
    
    public MessageRunner(ScheduledExecutorService executor) {
        this.runCounter = new AtomicInteger();
        this.executor = executor;
    }
    
//    private ThreadFactory threadFactory = new ThreadFactory() {
//        @Override
//        public Thread newThread(final Runnable r) {
//            return new Thread() {
//                @Override
//                public void run() {
//                    r.run();
//                }
//            };
//        }
//    };
    
    public void execute(Runnable task) {
        int startCount = runCounter.incrementAndGet();
        LOG.debug("Submit task and current task count {}", startCount);
        final long startTime = System.currentTimeMillis();
        executor.execute(() -> {
            try {
                 task.run();
            } finally {
                 int endCount = runCounter.decrementAndGet();
                 LOG.debug("Finish task and current task count {} use time {}", endCount, System.currentTimeMillis()-startTime);
            }
        });
    }
    
    public void shutdown() {
        executor.isShutdown();
    }
    
//    public void publishToMembers(MessageRecord message, List<Member> memberList, User sender) {
//        mScheduledExecutorService.schedule(threadFactory.newThread(new Runnable() {
//            @Override
//            public void run() {
//               exeMessage(user);
//            }
//        }), 100, TimeUnit.MILLISECONDS);
//    }
}