首页 > 代码库 > jedis例子
jedis例子
@Test public void testDiscoverNodesAutomatically(){ Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort("127.0.0.1",7379)); JedisCluster jc=new JedisCluster(jedisClusterNode); assertEquals(3,jc.getClusterNodes().size());}
@Test public void testCalculateConnectionPerSlot(){ Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort("127.0.0.1",7379)); JedisCluster jc=new JedisCluster(jedisClusterNode); jc.set("foo","bar"); jc.set("test","test"); assertEquals("bar",node3.get("foo")); assertEquals("test",node2.get("test"));}
@Test public void testRecalculateSlotsWhenMoved() throws InterruptedException { Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort("127.0.0.1",7379)); JedisCluster jc=new JedisCluster(jedisClusterNode); int slot51=JedisClusterCRC16.getSlot("51"); node2.clusterDelSlots(slot51); node3.clusterDelSlots(slot51); node3.clusterAddSlots(slot51); JedisClusterTestUtil.waitForClusterReady(node1,node2,node3); jc.set("51","foo"); assertEquals("foo",jc.get("51"));}
@Test public void testAskResponse() throws InterruptedException { Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort("127.0.0.1",7379)); JedisCluster jc=new JedisCluster(jedisClusterNode); int slot51=JedisClusterCRC16.getSlot("51"); node3.clusterSetSlotImporting(slot51,JedisClusterTestUtil.getNodeId(node2.clusterNodes())); node2.clusterSetSlotMigrating(slot51,JedisClusterTestUtil.getNodeId(node3.clusterNodes())); jc.set("51","foo"); assertEquals("foo",jc.get("51"));}
@Test(expected=JedisClusterException.class) public void testThrowExceptionWithoutKey(){ Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort("127.0.0.1",7379)); JedisCluster jc=new JedisCluster(jedisClusterNode); jc.ping();}
@Test(expected=JedisClusterMaxRedirectionsException.class) public void testRedisClusterMaxRedirections(){ Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort("127.0.0.1",7379)); JedisCluster jc=new JedisCluster(jedisClusterNode); int slot51=JedisClusterCRC16.getSlot("51"); node2.clusterSetSlotMigrating(slot51,JedisClusterTestUtil.getNodeId(node3.clusterNodes())); jc.set("51","foo");}
@Test public void testClusterCountKeysInSlot(){ Set<HostAndPort> jedisClusterNode=new HashSet<HostAndPort>(); jedisClusterNode.add(new HostAndPort(nodeInfo1.getHost(),nodeInfo1.getPort())); JedisCluster jc=new JedisCluster(jedisClusterNode); for (int index=0; index < 5; index++) { jc.set("foo{bar}" + index,"hello"); } int slot=JedisClusterCRC16.getSlot("foo{bar}"); assertEquals(5,node1.clusterCountKeysInSlot(slot).intValue());}
private void setSuperConnectionHandler(JedisClusterConnectionHandler handler){ try { Field connectionHandlerField=JedisCluster.class.getDeclaredField("connectionHandler"); connectionHandlerField.setAccessible(true); connectionHandlerField.set(this,handler); } catch ( Exception e) { e.printStackTrace(); }}
/** * Store a job in Redis * @param jobDetail the {@link JobDetail} object to be stored * @param replaceExisting if true, any existing job with the same group and name as the given job will be overwritten * @param jedis a thread-safe Redis connection * @throws ObjectAlreadyExistsException */@Override @SuppressWarnings("unchecked") public void storeJob(JobDetail jobDetail,boolean replaceExisting,JedisCluster jedis) throws ObjectAlreadyExistsException { final String jobHashKey=redisSchema.jobHashKey(jobDetail.getKey()); final String jobDataMapHashKey=redisSchema.jobDataMapHashKey(jobDetail.getKey()); final String jobGroupSetKey=redisSchema.jobGroupSetKey(jobDetail.getKey()); if (!replaceExisting && jedis.exists(jobHashKey)) { throw new ObjectAlreadyExistsException(jobDetail); } jedis.hmset(jobHashKey,(Map<String,String>)mapper.convertValue(jobDetail,new TypeReference<HashMap<String,String>>(){ })); if (jobDetail.getJobDataMap() != null && !jobDetail.getJobDataMap().isEmpty()) { jedis.hmset(jobDataMapHashKey,getStringDataMap(jobDetail.getJobDataMap())); } jedis.sadd(redisSchema.jobsSet(),jobHashKey); jedis.sadd(redisSchema.jobGroupsSet(),jobGroupSetKey); jedis.sadd(jobGroupSetKey,jobHashKey);}
/** * Remove the given job from Redis * @param jobKey the job to be removed * @param jedis a thread-safe Redis connection * @return true if the job was removed; false if it did not exist */@Override public boolean removeJob(JobKey jobKey,JedisCluster jedis) throws JobPersistenceException { final String jobHashKey=redisSchema.jobHashKey(jobKey); final String jobDataMapHashKey=redisSchema.jobDataMapHashKey(jobKey); final String jobGroupSetKey=redisSchema.jobGroupSetKey(jobKey); final String jobTriggerSetKey=redisSchema.jobTriggersSetKey(jobKey); Long delJobHashKeyResponse=jedis.del(jobHashKey); jedis.del(jobDataMapHashKey); jedis.srem(redisSchema.jobsSet(),jobHashKey); jedis.srem(jobGroupSetKey,jobHashKey); Set<String> jobTriggerSetResponse=jedis.smembers(jobTriggerSetKey); jedis.del(jobTriggerSetKey); Long jobGroupSetSizeResponse=jedis.scard(jobGroupSetKey); if (jobGroupSetSizeResponse == 0) { jedis.srem(redisSchema.jobGroupsSet(),jobGroupSetKey); } for ( String triggerHashKey : jobTriggerSetResponse) { final TriggerKey triggerKey=redisSchema.triggerKey(triggerHashKey); final String triggerGroupSetKey=redisSchema.triggerGroupSetKey(triggerKey); unsetTriggerState(triggerHashKey,jedis); jedis.srem(redisSchema.triggersSet(),triggerHashKey); jedis.srem(redisSchema.triggerGroupsSet(),triggerGroupSetKey); jedis.srem(triggerGroupSetKey,triggerHashKey); jedis.del(triggerHashKey); } return delJobHashKeyResponse == 1;}
/** * Remove (delete) the <code> {@link Trigger}</code> with the given key. * @param triggerKey the key of the trigger to be removed * @param removeNonDurableJob if true, the job associated with the given trigger will be removed if it is non-durableand has no other triggers * @param jedis a thread-safe Redis connection * @return true if the trigger was found and removed */@Override protected boolean removeTrigger(TriggerKey triggerKey,boolean removeNonDurableJob,JedisCluster jedis) throws JobPersistenceException, ClassNotFoundException { final String triggerHashKey=redisSchema.triggerHashKey(triggerKey); final String triggerGroupSetKey=redisSchema.triggerGroupSetKey(triggerKey); if (!jedis.exists(triggerHashKey)) { return false; } OperableTrigger trigger=retrieveTrigger(triggerKey,jedis); final String jobHashKey=redisSchema.jobHashKey(trigger.getJobKey()); final String jobTriggerSetKey=redisSchema.jobTriggersSetKey(trigger.getJobKey()); jedis.srem(redisSchema.triggersSet(),triggerHashKey); jedis.srem(triggerGroupSetKey,triggerHashKey); jedis.srem(jobTriggerSetKey,triggerHashKey); if (jedis.scard(triggerGroupSetKey) == 0) { jedis.srem(redisSchema.triggerGroupsSet(),triggerGroupSetKey); } if (removeNonDurableJob) { Long jobTriggerSetKeySizeResponse=jedis.scard(jobTriggerSetKey); Boolean jobExistsResponse=jedis.exists(jobHashKey); if (jobTriggerSetKeySizeResponse == 0 && jobExistsResponse) { JobDetail job=retrieveJob(trigger.getJobKey(),jedis); if (!job.isDurable()) { removeJob(job.getKey(),jedis); signaler.notifySchedulerListenersJobDeleted(job.getKey()); } } } if (isNullOrEmpty(trigger.getCalendarName())) { jedis.srem(redisSchema.calendarTriggersSetKey(trigger.getCalendarName()),triggerHashKey); } unsetTriggerState(triggerHashKey,jedis); jedis.del(triggerHashKey); return true;}
/** * Unsets the state of the given trigger key by removing the trigger from all trigger state sets. * @param triggerHashKey the redis key of the desired trigger hash * @param jedis a thread-safe Redis connection * @return true if the trigger was removed, false if the trigger was stateless * @throws JobPersistenceException if the unset operation failed */@Override public boolean unsetTriggerState(String triggerHashKey,JedisCluster jedis) throws JobPersistenceException { boolean removed=false; List<Long> responses=new ArrayList<>(RedisTriggerState.values().length); for ( RedisTriggerState state : RedisTriggerState.values()) { responses.add(jedis.zrem(redisSchema.triggerStateKey(state),triggerHashKey)); } for ( Long response : responses) { removed=response == 1; if (removed) { jedis.del(redisSchema.triggerLockKey(redisSchema.triggerKey(triggerHashKey))); break; } } return removed;}
/** * Store a {@link Calendar} * @param name the name of the calendar * @param calendar the calendar object to be stored * @param replaceExisting if true, any existing calendar with the same name will be overwritten * @param updateTriggers if true, any existing triggers associated with the calendar will be updated * @param jedis a thread-safe Redis connection * @throws JobPersistenceException */@Override public void storeCalendar(String name,Calendar calendar,boolean replaceExisting,boolean updateTriggers,JedisCluster jedis) throws JobPersistenceException { final String calendarHashKey=redisSchema.calendarHashKey(name); if (!replaceExisting && jedis.exists(calendarHashKey)) { throw new ObjectAlreadyExistsException(String.format("Calendar with key %s already exists.",calendarHashKey)); } Map<String,String> calendarMap=new HashMap<>(); calendarMap.put(CALENDAR_CLASS,calendar.getClass().getName()); try { calendarMap.put(CALENDAR_JSON,mapper.writeValueAsString(calendar)); } catch ( JsonProcessingException e) { throw new JobPersistenceException("Unable to serialize calendar.",e); } jedis.hmset(calendarHashKey,calendarMap); jedis.sadd(redisSchema.calendarsSet(),calendarHashKey); if (updateTriggers) { final String calendarTriggersSetKey=redisSchema.calendarTriggersSetKey(name); Set<String> triggerHashKeys=jedis.smembers(calendarTriggersSetKey); for ( String triggerHashKey : triggerHashKeys) { OperableTrigger trigger=retrieveTrigger(redisSchema.triggerKey(triggerHashKey),jedis); long removed=jedis.zrem(redisSchema.triggerStateKey(RedisTriggerState.WAITING),triggerHashKey); trigger.updateWithNewCalendar(calendar,misfireThreshold); if (removed == 1) { setTriggerState(RedisTriggerState.WAITING,(double)trigger.getNextFireTime().getTime(),triggerHashKey,jedis); } } }}
/** * Remove (delete) the <code> {@link Calendar}</code> with the given name. * @param calendarName the name of the calendar to be removed * @param jedis a thread-safe Redis connection * @return true if a calendar with the given name was found and removed */@Override public boolean removeCalendar(String calendarName,JedisCluster jedis) throws JobPersistenceException { final String calendarTriggersSetKey=redisSchema.calendarTriggersSetKey(calendarName); if (jedis.scard(calendarTriggersSetKey) > 0) { throw new JobPersistenceException(String.format("There are triggers pointing to calendar %s, so it cannot be removed.",calendarName)); } final String calendarHashKey=redisSchema.calendarHashKey(calendarName); Long deleteResponse=jedis.del(calendarHashKey); jedis.srem(redisSchema.calendarsSet(),calendarHashKey); return deleteResponse == 1;}
/** * Get the keys of all of the <code> {@link Job}</code> s that have the given group name. * @param matcher the matcher with which to compare group names * @param jedis a thread-safe Redis connection * @return the set of all JobKeys which have the given group name */@Override public Set<JobKey> getJobKeys(GroupMatcher<JobKey> matcher,JedisCluster jedis){ Set<JobKey> jobKeys=new HashSet<>(); if (matcher.getCompareWithOperator() == StringMatcher.StringOperatorName.EQUALS) { final String jobGroupSetKey=redisSchema.jobGroupSetKey(new JobKey("",matcher.getCompareToValue())); final Set<String> jobs=jedis.smembers(jobGroupSetKey); if (jobs != null) { for ( final String job : jobs) { jobKeys.add(redisSchema.jobKey(job)); } } } else { List<Set<String>> jobGroups=new ArrayList<>(); for ( final String jobGroupSetKey : jedis.smembers(redisSchema.jobGroupsSet())) { if (matcher.getCompareWithOperator().evaluate(redisSchema.jobGroup(jobGroupSetKey),matcher.getCompareToValue())) { jobGroups.add(jedis.smembers(jobGroupSetKey)); } } for ( Set<String> jobGroup : jobGroups) { if (jobGroup != null) { for ( final String job : jobGroup) { jobKeys.add(redisSchema.jobKey(job)); } } } } return jobKeys;}
jedis例子
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。