package com.zdemo.redis;

import com.zdemo.IProducer;
import org.redkale.convert.json.JsonConvert;
import org.redkale.service.Service;
import org.redkale.util.AnyValue;
import org.redkale.util.AutoLoad;

import javax.annotation.Resource;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.util.logging.Level;

@AutoLoad(false)
public class RedisProducer implements IProducer, Service {

    @Resource(name = "property.redis.host")
    private String host = "127.0.0.1";
    @Resource(name = "property.redis.password")
    private String password = "";
    @Resource(name = "property.redis.port")
    private int port = 6379;

    private OutputStreamWriter osw;

    @Override
    public void init(AnyValue config) {
        try {
            Socket client = new Socket();
            client.connect(new InetSocketAddress(host, port));
            client.setKeepAlive(true);

            osw = new OutputStreamWriter(client.getOutputStream());
            osw.write("AUTH " + password + "\r\n");
            osw.flush();
        } catch (IOException e) {
            logger.log(Level.WARNING, "", e);
        }
    }

    @Override
    public boolean publish(String topic, Object v) {
        try {
            osw.write("PUBLISH " + topic + " '" + toStr(v) + "' \r\n");
            osw.flush();
            return true;
        } catch (IOException e) {
            logger.log(Level.WARNING, "", e);
        }
        return false;
    }

    private String toStr(Object v) {
        if (v instanceof String) {
            return (String) v;
        }
        return JsonConvert.root().convertTo(v);
    }
}