4dkankan-project-tieta/src/main/java/com/fdkankan/redis/util/RedisClient.java

package com.fdkankan.redis.util;

import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.fdkankan.project.tieta.httpclient.HttpClient;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.logging.Log;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

@Slf4j
@Component
public class RedisClient {

    public final static String url_str_add = "/redis/string/add/";

    public final static String url_str_get = "/redis/string/get/";

    public final static String url_str_del = "/redis/string/delete";

    public final static String URL_REDISSON_LOCKLEASETIME = "/redis/redisson/lockLeaseTime";

    public final static String URL_REDISSON_UNLOCK = "/redis/redisson/unlock";

    public final static String field_status = "status";

    @Value("${redis.host}")
    private String host;

    @Value("${tieta.sysCode}")
    private String sysCode;

    @Value("${tieta.serverName}")
    private String serverName;

    @Resource
    private HttpClient httpClient;


     public void add(String key, String value){
         String url = host + url_str_add;
         Map<String, Object> params = new HashMap<>();
         params.put("key", this.genKey(key));
         params.put("value", value);
         JSONObject jsonObject = httpClient.postJson(url, params);
         if(Objects.isNull(jsonObject) || !"0".equals(jsonObject.getString(field_status))){
             throw new RuntimeException("redis add string error");
         }
     }

    public String get(String key){
        String url = host + url_str_get;
        Map<String, Object> params = new HashMap<>();
        params.put("key", this.genKey(key));
        JSONObject jsonObject = httpClient.postJson(url, params);
        if(Objects.isNull(jsonObject) || !"0".equals(jsonObject.getString(field_status))){
            throw new RuntimeException("redis get string error");
        }
        return jsonObject.getString("data");
    }

    public String del(String key){
        String url = host + url_str_del;
        Map<String, Object> params = new HashMap<>();
        params.put("key", this.genKey(key));
        JSONObject jsonObject = httpClient.postJson(url, params);
        if(Objects.isNull(jsonObject) || !"0".equals(jsonObject.getString(field_status))){
            throw new RuntimeException("redis del string error");
        }
        return jsonObject.getString("data");
    }

    public Map<String, Object> lockLeaseTime(String key, long expirTime){
        Map<String, Object> result = new HashMap<>();
        result.put("lock", false);
        String url = host + URL_REDISSON_LOCKLEASETIME;
        Map<String, Object> params = new HashMap<>();
        params.put("lockName", this.genKey(key));
        params.put("leaseTime", expirTime);
        JSONObject jsonObject = httpClient.postJson(url, params);
        log.info("获取分布式锁,key:{}, result:{}", this.genKey(key), jsonObject);
        if(Objects.nonNull(jsonObject) && "0".equals(jsonObject.getString(field_status))){
            result.put("lock", true);
            result.put("uniqueFlag", jsonObject.getString("data"));
        }
        return result;
    }

    public boolean unLock(String key, String uniqueFlag){
        String url = host + URL_REDISSON_UNLOCK;
        Map<String, Object> params = new HashMap<>();
        params.put("lockName", this.genKey(key));
        params.put("uniqueFlag", uniqueFlag);
        JSONObject jsonObject = httpClient.postJson(url, params);
        log.info("释放分布式锁,key:{},uniqueFlag:{}, result:{}", this.genKey(key), uniqueFlag, jsonObject);
        if(Objects.nonNull(jsonObject) && "0".equals(jsonObject.getString(field_status))){
            return true;
        }
        return false;
    }

    private String genKey(String key){
         return sysCode + "_" + serverName + "_" + key;
    }
}