io.github.hpsocket.soa.starter.leaf.service.impl.GlobalIdServiceImpl Maven / Gradle / Ivy

Go to download
package io.github.hpsocket.soa.starter.leaf.service.impl;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;

import io.github.hpsocket.soa.framework.core.exception.ServiceException;
import io.github.hpsocket.soa.framework.leaf.common.BatchResult;
import io.github.hpsocket.soa.framework.leaf.common.Result;
import io.github.hpsocket.soa.framework.leaf.common.Status;
import io.github.hpsocket.soa.framework.leaf.service.IdGen;
import lombok.Getter;
import io.github.hpsocket.soa.framework.leaf.service.GlobalIdService;

@Getter
public class GlobalIdServiceImpl implements GlobalIdService
{
    @Autowired(required = false)
    @Qualifier(LEAF_SNOWFLAKE_ID_GENERATOR_BEAN)
    IdGen leafSnowflakeIdGenerator;

    @Autowired(required = false)
    @Qualifier(LEAF_SEGMENT_ID_GENERATOR_BEAN)
    IdGen leafSegmentIdGenerator;
    
    @Override
    public boolean isSnowflakeIdSupported()
    {
        return (getLeafSnowflakeIdGenerator() != null);
    }
    
    @Override
    public boolean isSegmentIdSupported()
    {
        return (getLeafSegmentIdGenerator() != null);
    }

    @Override
    public long getSnowflakeId()
    {
        Result rs = leafSnowflakeIdGenerator.get(null);
        
        return checkResult(rs);
    }
    
    @Override
    public long getSegmentId(String tag)
    {
        Result rs = leafSegmentIdGenerator.get(tag);
        
        return checkResult(rs);
    }

    @Override
    public long getUniqueSegmentId()
    {
        return getSegmentId(UNIQUE_SEGMENT_TAG);
    }
    
    @Override
    public int getIntSegmentId(String tag)
    {
        long id = getSegmentId(tag);
        checkIntId(id);
        
        return (int)(id);
    }

    @Override
    public long[] getBatchSnowflakeId(int batchSize)
    {
        BatchResult brs = leafSnowflakeIdGenerator.getBatch(null, batchSize);
        
        return checkBatchResult(brs);
    }
    
    @Override
    public long[] getBatchSegmentId(String tag, int batchSize)
    {
        BatchResult brs = leafSegmentIdGenerator.getBatch(tag, batchSize);
        
        return checkBatchResult(brs);
    }
    
    @Override
    public long[] getBatchUniqueSegmentId(int batchSize)
    {
        return getBatchSegmentId(UNIQUE_SEGMENT_TAG, batchSize);
    }
    
    @Override
    public int[] getBatchIntSegmentId(String tag, int batchSize)
    {
        long[] ids   = getBatchSegmentId(tag, batchSize);
        int[] intIds = new int[ids.length];
        
        if(ids.length > 0)
        {
            checkIntId(ids[ids.length - 1]);
            
            for(int i = 0; i < ids.length; i++)
                intIds[i] = (int)(ids[i]);
        }
        
        return intIds;
    }
    
    @Override
    public SnowflakeId decodeSnowflakeId(long snowflakeId)
    {
        return decodeSnowflakeId(snowflakeId, IdGen.DEFAULT_SNOWFLAKE_TWEPOCK);
    }
    
    @Override
    public SnowflakeId decodeSnowflakeId(long snowflakeId, long twepoch)
    {
        long originTimestamp = (snowflakeId >> 22) + twepoch;
        int workerId = (int)((snowflakeId >> 12) ^ (snowflakeId >> 22 << 10));
        int sequence = (int)(snowflakeId ^ (snowflakeId >> 12 << 12));
        
        return new SnowflakeId(originTimestamp, workerId, sequence);
    }

    private long checkResult(Result rs)
    {
        long id = rs.getId();
        
        if(rs.getStatus() != Status.SUCCESS)
            throw new ServiceException(String.format("generate global ID fail -> (code: %d)", id), ServiceException.INNER_API_CALL_FAIL);
            
        return id;
    }

    private long[] checkBatchResult(BatchResult brs)
    {
        long[] ids = brs.getIds();
        
        if(brs.getStatus() != Status.SUCCESS)
        {
            long code = (ids != null && ids.length > 0) ? ids[0] : -100;
            throw new ServiceException(String.format("generate batch global ID fail -> (code: %d)", code), ServiceException.INNER_API_CALL_FAIL);
        }
            
        return ids;
    }
    
    private void checkIntId(long id)
    {
        if(id > Integer.MAX_VALUE)
            throw new ServiceException(String.format("generate integer global ID fail -> (id: %d) greater than Integer.MAX_VALUE", id), ServiceException.INNER_API_CALL_FAIL);
    }

}