() {
304 |
305 | @Override
306 | public String action(Jedis jedis) {
307 | return jedis.getSet(key, value);
308 | }
309 | });
310 | }
311 |
312 | /**
313 | * Increment the number stored at key by one. If the key does not exist or
314 | * contains a value of a wrong type, set the key to the value of "0" before
315 | * to perform the increment operation.
316 | *
317 | * INCR commands are limited to 64 bit signed integers.
318 | *
319 | * Note: this is actually a string operation, that is, in Redis there are
320 | * not "integer" types. Simply the string stored at the key is parsed as a
321 | * base 10 64 bit signed integer, incremented, and then converted back as a
322 | * string.
323 | *
324 | * @return Integer reply, this commands will reply with the new value of key
325 | * after the increment.
326 | */
327 | public Long incr(final String key) {
328 | return execute(new JedisAction() {
329 | @Override
330 | public Long action(Jedis jedis) {
331 | return jedis.incr(key);
332 | }
333 | });
334 | }
335 |
336 | public Long incrBy(final String key, final long increment) {
337 | return execute(new JedisAction() {
338 | @Override
339 | public Long action(Jedis jedis) {
340 | return jedis.incrBy(key, increment);
341 | }
342 | });
343 | }
344 |
345 | public Double incrByFloat(final String key, final double increment) {
346 | return execute(new JedisAction() {
347 | @Override
348 | public Double action(Jedis jedis) {
349 | return jedis.incrByFloat(key, increment);
350 | }
351 | });
352 | }
353 |
354 | /**
355 | * Decrement the number stored at key by one. If the key does not exist or
356 | * contains a value of a wrong type, set the key to the value of "0" before
357 | * to perform the decrement operation.
358 | */
359 | public Long decr(final String key) {
360 | return execute(new JedisAction() {
361 | @Override
362 | public Long action(Jedis jedis) {
363 | return jedis.decr(key);
364 | }
365 | });
366 | }
367 |
368 | public Long decrBy(final String key, final long decrement) {
369 | return execute(new JedisAction() {
370 | @Override
371 | public Long action(Jedis jedis) {
372 | return jedis.decrBy(key, decrement);
373 | }
374 | });
375 | }
376 |
377 | // / Hash Actions ///
378 | /**
379 | * If key holds a hash, retrieve the value associated to the specified
380 | * field.
381 | *
382 | * If the field is not found or the key does not exist, a special 'nil'
383 | * value is returned.
384 | */
385 | public String hget(final String key, final String fieldName) {
386 | return execute(new JedisAction() {
387 | @Override
388 | public String action(Jedis jedis) {
389 | return jedis.hget(key, fieldName);
390 | }
391 | });
392 | }
393 |
394 | public List hmget(final String key, final String... fieldsNames) {
395 | return execute(new JedisAction>() {
396 | @Override
397 | public List action(Jedis jedis) {
398 | return jedis.hmget(key, fieldsNames);
399 | }
400 | });
401 | }
402 |
403 | public Map hgetAll(final String key) {
404 | return execute(new JedisAction>() {
405 | @Override
406 | public Map action(Jedis jedis) {
407 | return jedis.hgetAll(key);
408 | }
409 | });
410 | }
411 |
412 | public void hset(final String key, final String fieldName,
413 | final String value) {
414 | execute(new JedisActionNoResult() {
415 |
416 | @Override
417 | public void action(Jedis jedis) {
418 | jedis.hset(key, fieldName, value);
419 | }
420 | });
421 | }
422 |
423 | public void hmset(final String key, final Map map) {
424 | execute(new JedisActionNoResult() {
425 |
426 | @Override
427 | public void action(Jedis jedis) {
428 | jedis.hmset(key, map);
429 | }
430 | });
431 |
432 | }
433 |
434 | public Boolean hsetnx(final String key, final String fieldName,
435 | final String value) {
436 | return execute(new JedisAction() {
437 |
438 | @Override
439 | public Boolean action(Jedis jedis) {
440 | return jedis.hsetnx(key, fieldName, value) == 1 ? true : false;
441 | }
442 | });
443 | }
444 |
445 | public Long hincrBy(final String key, final String fieldName,
446 | final long increment) {
447 | return execute(new JedisAction() {
448 | @Override
449 | public Long action(Jedis jedis) {
450 | return jedis.hincrBy(key, fieldName, increment);
451 | }
452 | });
453 | }
454 |
455 | public Double hincrByFloat(final String key, final String fieldName,
456 | final double increment) {
457 | return execute(new JedisAction() {
458 | @Override
459 | public Double action(Jedis jedis) {
460 | return jedis.hincrByFloat(key, fieldName, increment);
461 | }
462 | });
463 | }
464 |
465 | public Long hdel(final String key, final String... fieldsNames) {
466 | return execute(new JedisAction() {
467 | @Override
468 | public Long action(Jedis jedis) {
469 | return jedis.hdel(key, fieldsNames);
470 | }
471 | });
472 | }
473 |
474 | public Boolean hexists(final String key, final String fieldName) {
475 | return execute(new JedisAction() {
476 | @Override
477 | public Boolean action(Jedis jedis) {
478 | return jedis.hexists(key, fieldName);
479 | }
480 | });
481 | }
482 |
483 | public Set hkeys(final String key) {
484 | return execute(new JedisAction>() {
485 | @Override
486 | public Set action(Jedis jedis) {
487 | return jedis.hkeys(key);
488 | }
489 | });
490 | }
491 |
492 | public Long hlen(final String key) {
493 | return execute(new JedisAction() {
494 | @Override
495 | public Long action(Jedis jedis) {
496 | return jedis.hlen(key);
497 | }
498 | });
499 | }
500 |
501 | // / List Actions ///
502 |
503 | public Long lpush(final String key, final String... values) {
504 | return execute(new JedisAction() {
505 | @Override
506 | public Long action(Jedis jedis) {
507 | return jedis.lpush(key, values);
508 | }
509 | });
510 | }
511 |
512 | public String rpop(final String key) {
513 | return execute(new JedisAction() {
514 |
515 | @Override
516 | public String action(Jedis jedis) {
517 | return jedis.rpop(key);
518 | }
519 | });
520 | }
521 |
522 | public String brpop(final String key) {
523 | return execute(new JedisAction() {
524 |
525 | @Override
526 | public String action(Jedis jedis) {
527 | List nameValuePair = jedis.brpop(key);
528 | if (nameValuePair != null) {
529 | return nameValuePair.get(1);
530 | } else {
531 | return null;
532 | }
533 | }
534 | });
535 | }
536 |
537 | public String brpop(final int timeout, final String key) {
538 | return execute(new JedisAction() {
539 |
540 | @Override
541 | public String action(Jedis jedis) {
542 | List nameValuePair = jedis.brpop(timeout, key);
543 | if (nameValuePair != null) {
544 | return nameValuePair.get(1);
545 | } else {
546 | return null;
547 | }
548 | }
549 | });
550 | }
551 |
552 | /**
553 | * Not support for sharding.
554 | */
555 | public String rpoplpush(final String sourceKey, final String destinationKey) {
556 | return execute(new JedisAction() {
557 |
558 | @Override
559 | public String action(Jedis jedis) {
560 | return jedis.rpoplpush(sourceKey, destinationKey);
561 | }
562 | });
563 | }
564 |
565 | /**
566 | * Not support for sharding.
567 | */
568 | public String brpoplpush(final String source, final String destination,
569 | final int timeout) {
570 | return execute(new JedisAction() {
571 |
572 | @Override
573 | public String action(Jedis jedis) {
574 | return jedis.brpoplpush(source, destination, timeout);
575 | }
576 | });
577 | }
578 |
579 | public Long llen(final String key) {
580 | return execute(new JedisAction() {
581 |
582 | @Override
583 | public Long action(Jedis jedis) {
584 | return jedis.llen(key);
585 | }
586 | });
587 | }
588 |
589 | public String lindex(final String key, final long index) {
590 | return execute(new JedisAction() {
591 |
592 | @Override
593 | public String action(Jedis jedis) {
594 | return jedis.lindex(key, index);
595 | }
596 | });
597 | }
598 |
599 | public List lrange(final String key, final int start, final int end) {
600 | return execute(new JedisAction>() {
601 |
602 | @Override
603 | public List action(Jedis jedis) {
604 | return jedis.lrange(key, start, end);
605 | }
606 | });
607 | }
608 |
609 | public void ltrim(final String key, final int start, final int end) {
610 | execute(new JedisActionNoResult() {
611 | @Override
612 | public void action(Jedis jedis) {
613 | jedis.ltrim(key, start, end);
614 | }
615 | });
616 | }
617 |
618 | public void ltrimFromLeft(final String key, final int size) {
619 | execute(new JedisActionNoResult() {
620 | @Override
621 | public void action(Jedis jedis) {
622 | jedis.ltrim(key, 0, size - 1);
623 | }
624 | });
625 | }
626 |
627 | public Boolean lremFirst(final String key, final String value) {
628 | return execute(new JedisAction() {
629 | @Override
630 | public Boolean action(Jedis jedis) {
631 | Long count = jedis.lrem(key, 1, value);
632 | return (count == 1);
633 | }
634 | });
635 | }
636 |
637 | public Boolean lremAll(final String key, final String value) {
638 | return execute(new JedisAction() {
639 | @Override
640 | public Boolean action(Jedis jedis) {
641 | Long count = jedis.lrem(key, 0, value);
642 | return (count > 0);
643 | }
644 | });
645 | }
646 |
647 | // / Set Actions ///
648 | public Boolean sadd(final String key, final String member) {
649 | return execute(new JedisAction() {
650 |
651 | @Override
652 | public Boolean action(Jedis jedis) {
653 | return jedis.sadd(key, member) == 1 ? true : false;
654 | }
655 | });
656 | }
657 |
658 | public Set smembers(final String key) {
659 | return execute(new JedisAction>() {
660 |
661 | @Override
662 | public Set action(Jedis jedis) {
663 | return jedis.smembers(key);
664 | }
665 | });
666 | }
667 |
668 | // / Ordered Set Actions ///
669 | /**
670 | * return true for add new element, false for only update the score.
671 | */
672 | public Boolean zadd(final String key, final double score,
673 | final String member) {
674 | return execute(new JedisAction() {
675 |
676 | @Override
677 | public Boolean action(Jedis jedis) {
678 | return jedis.zadd(key, score, member) == 1 ? true : false;
679 | }
680 | });
681 | }
682 |
683 | public Double zscore(final String key, final String member) {
684 | return execute(new JedisAction() {
685 |
686 | @Override
687 | public Double action(Jedis jedis) {
688 | return jedis.zscore(key, member);
689 | }
690 | });
691 | }
692 |
693 | public Long zrank(final String key, final String member) {
694 | return execute(new JedisAction() {
695 |
696 | @Override
697 | public Long action(Jedis jedis) {
698 | return jedis.zrank(key, member);
699 | }
700 | });
701 | }
702 |
703 | public Long zrevrank(final String key, final String member) {
704 | return execute(new JedisAction() {
705 |
706 | @Override
707 | public Long action(Jedis jedis) {
708 | return jedis.zrevrank(key, member);
709 | }
710 | });
711 | }
712 |
713 | public Long zcount(final String key, final double min, final double max) {
714 | return execute(new JedisAction() {
715 |
716 | @Override
717 | public Long action(Jedis jedis) {
718 | return jedis.zcount(key, min, max);
719 | }
720 | });
721 | }
722 |
723 | public Set zrange(final String key, final int start, final int end) {
724 | return execute(new JedisAction>() {
725 |
726 | @Override
727 | public Set action(Jedis jedis) {
728 | return jedis.zrange(key, start, end);
729 | }
730 | });
731 | }
732 |
733 | public Set zrangeWithScores(final String key, final int start,
734 | final int end) {
735 | return execute(new JedisAction>() {
736 |
737 | @Override
738 | public Set action(Jedis jedis) {
739 | return jedis.zrangeWithScores(key, start, end);
740 | }
741 | });
742 | }
743 |
744 | public Set zrevrange(final String key, final int start,
745 | final int end) {
746 | return execute(new JedisAction>() {
747 |
748 | @Override
749 | public Set action(Jedis jedis) {
750 | return jedis.zrevrange(key, start, end);
751 | }
752 | });
753 | }
754 |
755 | public Set zrevrangeWithScores(final String key, final int start,
756 | final int end) {
757 | return execute(new JedisAction>() {
758 |
759 | @Override
760 | public Set action(Jedis jedis) {
761 | return jedis.zrevrangeWithScores(key, start, end);
762 | }
763 | });
764 | }
765 |
766 | public Set zrangeByScore(final String key, final double min,
767 | final double max) {
768 | return execute(new JedisAction>() {
769 |
770 | @Override
771 | public Set action(Jedis jedis) {
772 | return jedis.zrangeByScore(key, min, max);
773 | }
774 | });
775 | }
776 |
777 | public Set zrangeByScoreWithScores(final String key,
778 | final double min, final double max) {
779 | return execute(new JedisAction>() {
780 |
781 | @Override
782 | public Set action(Jedis jedis) {
783 | return jedis.zrangeByScoreWithScores(key, min, max);
784 | }
785 | });
786 | }
787 |
788 | public Set zrevrangeByScore(final String key, final double max,
789 | final double min) {
790 | return execute(new JedisAction>() {
791 |
792 | @Override
793 | public Set action(Jedis jedis) {
794 | return jedis.zrevrangeByScore(key, max, min);
795 | }
796 | });
797 | }
798 |
799 | public Set zrevrangeByScoreWithScores(final String key,
800 | final double max, final double min) {
801 | return execute(new JedisAction>() {
802 |
803 | @Override
804 | public Set action(Jedis jedis) {
805 | return jedis.zrevrangeByScoreWithScores(key, max, min);
806 | }
807 | });
808 | }
809 |
810 | public Boolean zrem(final String key, final String member) {
811 | return execute(new JedisAction() {
812 |
813 | @Override
814 | public Boolean action(Jedis jedis) {
815 | return jedis.zrem(key, member) == 1 ? true : false;
816 | }
817 | });
818 | }
819 |
820 | public Long zremByScore(final String key, final double start,
821 | final double end) {
822 | return execute(new JedisAction() {
823 |
824 | @Override
825 | public Long action(Jedis jedis) {
826 | return jedis.zremrangeByScore(key, start, end);
827 | }
828 | });
829 | }
830 |
831 | public Long zremByRank(final String key, final long start, final long end) {
832 | return execute(new JedisAction() {
833 |
834 | @Override
835 | public Long action(Jedis jedis) {
836 | return jedis.zremrangeByRank(key, start, end);
837 | }
838 | });
839 | }
840 |
841 | public Long zcard(final String key) {
842 | return execute(new JedisAction() {
843 |
844 | @Override
845 | public Long action(Jedis jedis) {
846 | return jedis.zcard(key);
847 | }
848 | });
849 | }
850 | }
851 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/nosql/JedisUtils.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.nosql;
2 |
3 | import redis.clients.jedis.Jedis;
4 | import redis.clients.jedis.JedisPool;
5 | import redis.clients.jedis.exceptions.JedisException;
6 | import cn.flaty.NettyPush.nosql.JedisTemplate.JedisAction;
7 |
8 | public class JedisUtils {
9 |
10 | private static final String OK_CODE = "OK";
11 | private static final String OK_MULTI_CODE = "+OK";
12 |
13 | /**
14 | * 判断 返回值是否ok.
15 | */
16 | public static boolean isStatusOk(String status) {
17 | return (status != null)
18 | && (OK_CODE.equals(status) || OK_MULTI_CODE.equals(status));
19 | }
20 |
21 | /**
22 | * Ping the jedis instance, return true is the result is PONG.
23 | */
24 | public static boolean ping(JedisPool pool) {
25 | JedisTemplate template = new JedisTemplate(pool);
26 | try {
27 | String result = template.execute(new JedisAction() {
28 | @Override
29 | public String action(Jedis jedis) {
30 | return jedis.ping();
31 | }
32 | });
33 | return (result != null) && result.equals("PONG");
34 | } catch (JedisException e) {
35 | return false;
36 | }
37 | }
38 |
39 | }
40 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/nosql/RedisClientRepository.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.nosql;
2 |
3 | import org.slf4j.Logger;
4 | import org.slf4j.LoggerFactory;
5 | import org.springframework.beans.factory.annotation.Autowired;
6 | import org.springframework.stereotype.Repository;
7 |
8 | import cn.flaty.NettyPush.entity.packet.ClientPacket;
9 |
10 | /**
11 | *
12 | *
13 | * did -> 设备ID
14 | * appkey -> app唯一标识码
15 | * appClientUId = md5(appKey+did)
16 | *
17 | * client:[id] -> [appClientUId] **[String]** **expire[x second]**
18 | * [appkey] -> id **[set]**
19 | * [appkey]:[os] -> id **[set]**
20 | * [appkey]:[appVer] -> id **[set]**
21 | * [appkey]:[label] -> id **[set]**
22 | *
23 | *
24 | *
25 | * @author flatychen
26 | *
27 | */
28 | @Repository
29 | public class RedisClientRepository {
30 |
31 | private static int clientExpireTime = 30;
32 | private static String clientsKeyPrefix = "clients:";
33 | private static String clientKeyPrefix = "client:";
34 |
35 | // @Autowired
36 | private JedisTemplate jedisTemplate;
37 |
38 | private static Logger logger = LoggerFactory
39 | .getLogger(RedisClientRepository.class);
40 |
41 | /**
42 | *
43 | *
44 | * @param client
45 | * @return
46 | * @author flatychen
47 | */
48 | public void saveClient(ClientPacket client) {
49 | jedisTemplate.hset(clientsKeyPrefix + client.getAppKey(),
50 | client.getRedisField(), client.getAppClientUID());
51 | jedisTemplate.setex(clientKeyPrefix + client.getAppClientUID(), "1",
52 | clientExpireTime);
53 | }
54 |
55 | /**
56 | *
57 | *
58 | * @param client
59 | * @return
60 | * @author flatychen
61 | */
62 | public void touchClient(ClientPacket client) {
63 | jedisTemplate.setex(clientKeyPrefix + client.getAppClientUID(), "1",
64 | clientExpireTime);
65 | }
66 |
67 | }
68 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/repository/ClientRepository.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.repository;
2 |
3 | import java.util.List;
4 |
5 | import org.apache.commons.lang.time.DateUtils;
6 |
7 | import cn.flaty.NettyPush.entity.packet.ClientPacket;
8 | import cn.flaty.NettyPush.entity.persitence.Client;
9 |
10 | public interface ClientRepository {
11 |
12 | @SuppressWarnings("deprecation")
13 | public static long client_db_live_time = DateUtils.MILLIS_IN_SECOND * 40;
14 |
15 | public boolean insertClient(ClientPacket c);
16 |
17 | public Client getClient(String did);
18 |
19 | public List queryClients(String appKey);
20 |
21 | public boolean touchClient(ClientPacket c);
22 |
23 | public boolean delExpireClient();
24 |
25 | }
26 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/repository/H2ClientRepository.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.repository;
2 |
3 | import java.util.Date;
4 | import java.util.List;
5 |
6 | import org.springframework.beans.factory.annotation.Autowired;
7 | import org.springframework.stereotype.Repository;
8 |
9 | import cn.flaty.NettyPush.entity.packet.ClientPacket;
10 | import cn.flaty.NettyPush.entity.persitence.Client;
11 |
12 | @Repository
13 | public class H2ClientRepository implements ClientRepository {
14 |
15 | @Autowired
16 | private JdbcTemplateWrapper jdbc;
17 |
18 | @Override
19 | public boolean insertClient(ClientPacket c) {
20 | // 设置过期时间
21 | long expireTime = new Date(new Date().getTime()
22 | + ClientRepository.client_db_live_time).getTime();
23 | String sql = " insert into tb_online_client(appkey,did,expireTime,appVer,os) values(?,?,?,?,?)";
24 | return jdbc.saveORUpdate(sql, new Object[] { c.getAppKey(), c.getDid(),
25 | expireTime, c.getAppVer(), c.getOs() }) == 1;
26 | }
27 |
28 | @Override
29 | public List queryClients(String appKey) {
30 | String sql = " select appkey,did,expireTime from tb_online_client where appkey = ? and expireTime > ? ";
31 | return jdbc
32 | .queryForBeanList(sql, Client.class, new Object[] { appKey ,new Date().getTime()});
33 | }
34 |
35 | @Override
36 | public boolean delExpireClient() {
37 | String sql = " delete from tb_online_client where expireTime < ? ";
38 | return jdbc.saveORUpdate(sql, new Object[] { new Date().getTime() }) >= 1;
39 | }
40 |
41 | @Override
42 | public boolean touchClient(ClientPacket c) {
43 | String sql = " update tb_online_client set expireTime = ? where did = ?";
44 | return jdbc.saveORUpdate(sql,
45 | new Object[] { new Date().getTime() + ClientRepository.client_db_live_time , c.getDid() }) == 1;
46 | }
47 |
48 | @Override
49 | public Client getClient(String did) {
50 | String sql = " select appkey,did,expireTime from tb_online_client where did = ?";
51 | return jdbc.queryForBean(sql, Client.class, new Object[] { did });
52 | }
53 |
54 | }
55 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/repository/JdbcTemplateWrapper.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.repository;
2 |
3 | import java.util.Arrays;
4 | import java.util.List;
5 |
6 | import org.slf4j.Logger;
7 | import org.slf4j.LoggerFactory;
8 | import org.springframework.beans.factory.annotation.Autowired;
9 | import org.springframework.dao.DataAccessException;
10 | import org.springframework.jdbc.core.BeanPropertyRowMapper;
11 | import org.springframework.jdbc.core.JdbcTemplate;
12 | import org.springframework.stereotype.Component;
13 |
14 | /**
15 | * Spring jdbcTemplate 包装器类，提供基本SQL操作
16 | *
17 | * @author flatychen
18 | *
19 | */
20 |
21 | @Component
22 | public class JdbcTemplateWrapper {
23 |
24 | private static Logger log = LoggerFactory
25 | .getLogger(JdbcTemplateWrapper.class);
26 |
27 | @Autowired
28 | private JdbcTemplate jdbcTemplate;
29 |
30 | public void setJdbcTemplate(JdbcTemplate jdbcTemplate) {
31 | this.jdbcTemplate = jdbcTemplate;
32 | }
33 |
34 | private final List query(String sql, Class clazz, Object args[]) {
35 | List l = null;
36 | try {
37 | log.info("====>>query{ sql:{}, args:{} }", sql,
38 | Arrays.toString(args));
39 | l = this.jdbcTemplate.query(sql, args,
40 | new BeanPropertyRowMapper(clazz));
41 | } catch (DataAccessException e) {
42 | e.printStackTrace();
43 | }
44 | return l;
45 |
46 | }
47 |
48 | /**
49 | * class得到对象
50 | *
51 | * @param sql
52 | * @param clazz
53 | * @param args
54 | * @return
55 | * @author flatychen
56 | * @date 2014-1-10
57 | */
58 | public final T queryForBean(String sql, Class clazz, Object args[]) {
59 | List l = this.query(sql, clazz, args);
60 | if ((l != null) && (l.size() == 1)) {
61 | return l.get(0);
62 | } else {
63 |
64 | }
65 | return null;
66 | }
67 |
68 | /**
69 | * class得到对象list
70 | *
71 | * @param sql
72 | * @param clazz
73 | * @param args
74 | * @return
75 | * @author flatychen
76 | * @date 2014-1-10
77 | */
78 | public final List queryForBeanList(String sql, Class clazz,
79 | Object args[]) {
80 | return this.query(sql, clazz, args);
81 |
82 | }
83 |
84 | /**
85 | * 返回 long --> count(1)
86 | *
87 | * @param sql
88 | * @param args
89 | * @return
90 | * @author flatychen
91 | * @date 2014-1-10
92 | */
93 | public final long queryForLong(String sql, Object args[]) {
94 | log.info("====>>queryForLong[ sql:[ {} ] , args:[ {} ] ]", sql,
95 | Arrays.toString(args));
96 | return this.jdbcTemplate.queryForLong(sql, args);
97 |
98 | }
99 |
100 | /**
101 | * 返回 int -->count(1)
102 | *
103 | * @param sql
104 | * @param args
105 | * @return
106 | * @author flatychen
107 | * @date 2014-1-10
108 | */
109 | public final int queryForInt(String sql, Object args[]) {
110 | log.info("====>>queryForInt[ sql:[ {} ] , args:[ {} ] ]", sql,
111 | Arrays.toString(args));
112 | return this.jdbcTemplate.queryForInt(sql, args);
113 |
114 | }
115 |
116 | /**
117 | * 返回 object eg: Integer,Object
118 | *
119 | * @param sql
120 | * @param args
121 | * @return
122 | * @author flatychen
123 | * @param
124 | * @date 2014-1-10
125 | */
126 | public final T queryForObject(String sql, Class clazz, Object args[]) {
127 | log.info("====>>queryForObject[ sql:[{}] , args:[{}] ]", sql,
128 | Arrays.toString(args));
129 | return this.jdbcTemplate.queryForObject(sql, clazz, args);
130 |
131 | }
132 |
133 | /**
134 | *
135 | * @author flatychen
136 | * @param sql
137 | * @param args
138 | * @return
139 | * @date 2014-1-12
140 | */
141 | public int saveORUpdate(String sql, Object args[]) {
142 | log.info("====>>saveORUpdate[ sql:[ {} ] , args:[ {} ] ]", sql,
143 | Arrays.toString(args));
144 | int num = 0;
145 | try {
146 | num = jdbcTemplate.update(sql, args);
147 | } catch (DataAccessException e) {
148 | e.printStackTrace();
149 | throw e;
150 | }
151 | return num;
152 | }
153 |
154 | /**
155 | * 批量更新
156 | *
157 | * @author flatychen
158 | * @param sql
159 | * @param listArgs
160 | * @return
161 | */
162 | public int batchUpdate(String sql, List listArgs) {
163 | log.info("====>>batchInsert[ sql:[ {} ] , args:[ {} ] ]", sql,
164 | Arrays.toString(listArgs.toArray()));
165 | int num[] = null;
166 | try {
167 | num = jdbcTemplate.batchUpdate(sql, listArgs);
168 | } catch (DataAccessException e) {
169 | e.printStackTrace();
170 | throw e;
171 | }
172 | return num.length;
173 | }
174 |
175 | }
176 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/repository/MysqlClientRepository.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.repository;
2 |
3 | import java.util.List;
4 |
5 | import org.springframework.beans.factory.annotation.Autowired;
6 |
7 | import cn.flaty.NettyPush.entity.packet.ClientPacket;
8 | import cn.flaty.NettyPush.entity.persitence.Client;
9 |
10 | //@Repository
11 | public class MysqlClientRepository implements ClientRepository {
12 |
13 | @Autowired
14 | private JdbcTemplateWrapper jdbc;
15 |
16 | @Override
17 | public boolean insertClient(ClientPacket c) {
18 | // TODO Auto-generated method stub
19 | return false;
20 | }
21 |
22 | @Override
23 | public Client getClient(String did) {
24 | // TODO Auto-generated method stub
25 | return null;
26 | }
27 |
28 | @Override
29 | public List queryClients(String appKey) {
30 | // TODO Auto-generated method stub
31 | return null;
32 | }
33 |
34 |
35 | @Override
36 | public boolean delExpireClient() {
37 | // TODO Auto-generated method stub
38 | return false;
39 | }
40 |
41 | @Override
42 | public boolean touchClient(ClientPacket c) {
43 | // TODO Auto-generated method stub
44 | return false;
45 | }
46 |
47 | }
48 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/DefaultListener.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server;
2 |
3 | import io.netty.bootstrap.ServerBootstrap;
4 | import io.netty.buffer.PooledByteBufAllocator;
5 | import io.netty.channel.Channel;
6 | import io.netty.channel.ChannelFuture;
7 | import io.netty.channel.ChannelHandler;
8 | import io.netty.channel.ChannelInitializer;
9 | import io.netty.channel.ChannelOption;
10 | import io.netty.channel.EventLoopGroup;
11 | import io.netty.channel.MessageSizeEstimator;
12 | import io.netty.channel.MessageSizeEstimator.Handle;
13 | import io.netty.channel.nio.NioEventLoopGroup;
14 | import io.netty.channel.socket.nio.NioServerSocketChannel;
15 | import io.netty.handler.logging.LogLevel;
16 | import io.netty.handler.logging.LoggingHandler;
17 |
18 | import java.text.MessageFormat;
19 |
20 | import org.slf4j.Logger;
21 | import org.slf4j.LoggerFactory;
22 | import org.springframework.beans.factory.annotation.Required;
23 |
24 | import cn.flaty.NettyPush.utils.RuntimeUtils;
25 |
26 | /**
27 | *
28 | * 默认监听器
29 | * @author flaty
30 | *
31 | */
32 |
33 | public class DefaultListener implements Listener{
34 |
35 | private Logger log = LoggerFactory.getLogger(DefaultListener.class);
36 |
37 | private ChannelInitializer channelInitializer;
38 |
39 | private int port;
40 |
41 | private String host;
42 |
43 |
44 | public void start() {
45 | // acceptor 线程，用于监听 相当于 bio accept,并分发至worker线程
46 | EventLoopGroup acceptor = new NioEventLoopGroup(1);
47 | // work线程，用于处理，取当前服务器 cpu 数量 * 2
48 | EventLoopGroup worker = new NioEventLoopGroup(RuntimeUtils.getProcessors() * 2);
49 | try {
50 | // 启动器
51 | ServerBootstrap sbs = new ServerBootstrap();
52 | sbs.group(acceptor, worker).channel(NioServerSocketChannel.class)
53 | // .childOption(ChannelOption.SO_KEEPALIVE, true)
54 | .childOption(ChannelOption.TCP_NODELAY, true) // 禁止 Nagle算法
55 | // .childOption(ChannelOption.SO_REUSEADDR, true)
56 | .childOption(ChannelOption.ALLOCATOR, PooledByteBufAllocator.DEFAULT) // 使用内存池 回收堆外内存
57 | // .childOption(ChannelOption.MESSAGE_SIZE_ESTIMATOR, new MessageSizeEstimator() {
58 | // @Override
59 | // public Handle newHandle() {
60 | // return new Handle() {
61 | // @Override
62 | // public int size(Object msg) {
63 | // return 256;
64 | // }
65 | // };
66 | // }
67 | // }) // 使用内存池 回收堆外内存
68 |
69 | .localAddress(host, port)
70 | .handler(new LoggingHandler(LogLevel.INFO))
71 | .childHandler(channelInitializer);
72 | log.info(MessageFormat.format("---> server start and listen on port:{0}",this.port+""));
73 | ChannelFuture f = sbs.bind().sync();
74 | f.channel().closeFuture().sync();
75 | } catch (InterruptedException e) {
76 | log.error(e.getMessage());
77 | e.printStackTrace();
78 | } finally{
79 | acceptor.shutdownGracefully();
80 | worker.shutdownGracefully();
81 | }
82 | }
83 |
84 | @Required
85 | public void setChannelInitializer(ChannelInitializer channelInitializer) {
86 | this.channelInitializer = channelInitializer;
87 | }
88 |
89 | @Required
90 | public void setPort(int port) {
91 | this.port = port;
92 | }
93 |
94 | @Required
95 | public void setHost(String host) {
96 | this.host = host;
97 | }
98 |
99 |
100 |
101 | }
102 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/Listener.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server;
2 |
3 |
4 | public interface Listener {
5 |
6 |
7 | /**
8 | * 启动服务器监听
9 | */
10 | public void start();
11 |
12 | }
13 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/channelInitializer/AbstractChannelInitializer.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server.channelInitializer;
2 |
3 | import io.netty.channel.Channel;
4 | import io.netty.channel.ChannelHandler;
5 | import io.netty.channel.ChannelInitializer;
6 | import io.netty.channel.ChannelPipeline;
7 |
8 | /**
9 | *
10 | * 抽像ChannelInitializer
11 | *
12 | * 用于组织不同的channel
13 | *
14 | * @author flatychen
15 | *
16 | */
17 | public abstract class AbstractChannelInitializer extends ChannelInitializer {
18 |
19 | private ChannelHandler[] inHandlers;
20 |
21 | private ChannelHandler[] outHandlers;
22 |
23 | public void setInHandlers(ChannelHandler[] inHandlers) {
24 | this.inHandlers = inHandlers;
25 | }
26 |
27 | public void setOutHandlers(ChannelHandler[] outHandlers) {
28 | this.outHandlers = outHandlers;
29 | }
30 |
31 | @Override
32 | protected void initChannel(Channel ch) throws Exception {
33 | ChannelPipeline pipeline = ch.pipeline();
34 | if (inHandlers != null) {
35 | pipeline.addFirst(inHandlers);
36 | }
37 | this.initPipeline(pipeline);
38 | if (outHandlers != null) {
39 | // pipeline.addl(outHandlers);
40 | }
41 |
42 | }
43 |
44 | protected abstract void initPipeline(ChannelPipeline pipeline);
45 |
46 | }
47 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/channelInitializer/PushChannelInitializer.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server.channelInitializer;
2 |
3 | import io.netty.channel.ChannelPipeline;
4 |
5 | import org.springframework.beans.factory.annotation.Required;
6 |
7 | import cn.flaty.NettyPush.server.codec.push.DeserializeHandler;
8 | import cn.flaty.NettyPush.server.codec.push.MergeFrameEncoder;
9 | import cn.flaty.NettyPush.server.codec.push.PushFrameDecoder;
10 | import cn.flaty.NettyPush.server.codec.push.PushFrameEncoder;
11 | import cn.flaty.NettyPush.server.codec.push.SplitFrameDecoder;
12 | import cn.flaty.NettyPush.server.frame.FrameHead;
13 |
14 | /**
15 | *
16 | * 推送channel pipeline
17 | *
18 | * @author flatychen
19 | *
20 | */
21 | public class PushChannelInitializer extends AbstractChannelInitializer {
22 |
23 | private FrameHead frameHeader;
24 |
25 | @Override
26 | protected void initPipeline(ChannelPipeline pipeline) {
27 |
28 | // 切包,粘包
29 | pipeline.addLast(new SplitFrameDecoder(frameHeader));
30 | // 解码
31 | pipeline.addLast(new PushFrameDecoder(frameHeader));
32 |
33 | // 编码器
34 | // 粘包
35 | pipeline.addLast(new MergeFrameEncoder(frameHeader));
36 | // 编码
37 | pipeline.addLast(new PushFrameEncoder(frameHeader));
38 |
39 | // 业务处理
40 | pipeline.addLast(new DeserializeHandler());
41 |
42 | }
43 |
44 | @Required
45 | public void setFrameHeader(FrameHead frameHeader) {
46 | this.frameHeader = frameHeader;
47 | }
48 |
49 | }
50 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/codec/push/DeserializeHandler.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server.codec.push;
2 |
3 | import io.netty.channel.ChannelHandlerContext;
4 | import io.netty.channel.SimpleChannelInboundHandler;
5 |
6 | import org.slf4j.Logger;
7 | import org.slf4j.LoggerFactory;
8 |
9 | import cn.flaty.NettyPush.server.conn.NettyConnection;
10 | import cn.flaty.NettyPush.services.ClientDispacherService;
11 | import cn.flaty.NettyPush.utils.beanFactoryUtils;
12 |
13 | /**
14 | *
15 | * 业务处理.
16 | *
17 | * @author flatychen
18 | *
19 | */
20 | public class DeserializeHandler extends SimpleChannelInboundHandler {
21 |
22 | private ClientDispacherService deserialize;
23 |
24 | private Logger log = LoggerFactory.getLogger(DeserializeHandler.class);
25 |
26 | public DeserializeHandler() {
27 | super();
28 | deserialize = beanFactoryUtils.getClientDispacherService();
29 | }
30 |
31 | @Override
32 | protected void channelRead0(ChannelHandlerContext ctx, String msg)
33 | throws Exception {
34 | log.info("receive msg:{}", msg);
35 | // 分发
36 | NettyConnection conn = new NettyConnection(ctx);
37 | deserialize.dispacher(conn, msg);
38 | }
39 |
40 | }
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/codec/push/MergeFrameEncoder.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server.codec.push;
2 |
3 | import io.netty.handler.codec.LengthFieldPrepender;
4 | import cn.flaty.NettyPush.server.frame.FrameHead;
5 |
6 |
7 | /**
8 | * @author flaty
9 | *
10 | */
11 | public class MergeFrameEncoder extends LengthFieldPrepender {
12 |
13 | public MergeFrameEncoder(FrameHead frameHead){
14 | super(frameHead.byteLength(),- frameHead.headLength(),false);
15 | }
16 |
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/java/cn/flaty/NettyPush/server/codec/push/PushFrameDecoder.java:
--------------------------------------------------------------------------------
1 | package cn.flaty.NettyPush.server.codec.push;
2 |
3 | import io.netty.buffer.ByteBuf;
4 | import io.netty.channel.ChannelHandlerContext;
5 | import io.netty.handler.codec.ByteToMessageDecoder;
6 |
7 | import java.net.InetSocketAddress;
8 | import java.util.List;
9 |
10 | import org.slf4j.Logger;
11 | import org.slf4j.LoggerFactory;
12 |
13 | import cn.flaty.NettyPush.server.frame.FrameHead;
14 | import cn.flaty.NettyPush.server.frame.SimplePushInFrame;
15 | import cn.flaty.NettyPush.utils.AssertUtils;
16 |
17 | /**
18 | *
19 | * 根据包头解码
20 | *
21 | * @author flatychen
22 | *
23 | */
24 | public class PushFrameDecoder extends ByteToMessageDecoder {
25 |
26 | private Logger log = LoggerFactory.getLogger(PushFrameDecoder.class);
27 |
28 | private FrameHead frameHead;
29 |
30 | public PushFrameDecoder(FrameHead frameHead) {
31 | super();
32 | AssertUtils.notNull(frameHead, " frameHead 不能为空 ");
33 | this.frameHead = frameHead;
34 | }
35 |
36 | @Override
37 | protected void decode(ChannelHandlerContext ctx, ByteBuf in,
38 | List