com.test.config.service.zkdao.AuthDao.java Source code

Introduction

Here is the source code for com.test.config.service.zkdao.AuthDao.java

Source

/**
 * Copyright 1999-2014 dangdang.com.
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *  
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.test.config.service.zkdao;

import java.util.Arrays;

import org.apache.zookeeper.data.Stat;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Charsets;
import com.google.common.base.Throwables;
import com.google.common.hash.HashCode;
import com.google.common.hash.HashFunction;
import com.google.common.hash.Hashing;

/**
 * 
 * Do access control with ACL instead of current manual implementation.
 * 
 * @author <a href="mailto:wangyuxuan@dangdang.com">Yuxuan Wang</a>
 * 
 */
public class AuthDao extends BaseDao implements IAuthDao {

    /**
     * 
     */
    private static final long serialVersionUID = 1L;

    private static final Logger LOGGER = LoggerFactory.getLogger(AuthDao.class);

    @Override
    public boolean checkAuth(String nodeName, String password) {
        byte[] hash = sha1Digest(password);
        boolean isPass = false;
        try {
            byte[] data = getClient().getData().forPath(nodeName);
            isPass = Arrays.equals(hash, data);
        } catch (Exception e) {
            Throwables.propagate(e);
        }
        return isPass;
    }

    @Override
    public boolean auth(String nodeName, String password) {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug("Try auth node {} with password ***", nodeName);
        }

        boolean suc = false;
        byte[] sha1Digest = sha1Digest(password);
        try {
            // ?
            Stat stat = getClient().checkExists().forPath(nodeName);
            if (stat == null) {
                LOGGER.info("Node not exists, create it.");
                getClient().create().creatingParentsIfNeeded().forPath(nodeName, sha1Digest);
                suc = true;
            } else {
                LOGGER.info("Node exists.");
                byte[] data = getClient().getData().forPath(nodeName);
                // ??
                if (data == null || (data.length == 1 && data[0] != (byte) 0)) {
                    getClient().setData().forPath(nodeName, sha1Digest);
                    suc = true;
                    LOGGER.info("Auth done.");
                } else {
                    LOGGER.info("Node has been authed, cannot do duplicated authentication.");
                }
            }

        } catch (Exception e) {
            Throwables.propagate(e);
        }
        return suc;
    }

    private byte[] sha1Digest(String text) {
        HashFunction hf = Hashing.sha1();
        HashCode hashCode = hf.hashString(text, Charsets.UTF_8);
        return hashCode.asBytes();
    }
}