MultiThread Basewrite -Java 语言-立哥开发

ackage net.rubyeye.memcached;
//Copyright 2008-2021 Jacky Zong .
//All rights reserved.
//但使龙城飞将在
//不教胡马度阴山  

import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.atomic.AtomicLong;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class BaseReadWriteThread extends Thread {

    private final Logger log;
    protected final int repeats;
    protected final CyclicBarrier barrier;
    protected final int offset;
    protected final int length;
    protected final AtomicLong miss;
    protected final AtomicLong fail;
    protected final AtomicLong hit;

    public BaseReadWriteThread(int repeats, CyclicBarrier barrier, int offset,
            int length, AtomicLong miss, AtomicLong fail, AtomicLong hit) {
        super();
        this.repeats = repeats;
        this.barrier = barrier;
        this.offset = offset;
        this.length = length;
        this.miss = miss;
        this.fail = fail;
        this.hit = hit;
        this.log = LoggerFactory.getLogger(BaseReadWriteThread.class);
    }

    public abstract boolean set(int i, String s) throws Exception;

    public abstract String get(int n) throws Exception;

    public void run() {

        int writeTimes = (int) (this.repeats * Constants.WRITE_RATE);
        try {
            barrier.await();
            int writeMax = offset + writeTimes;
            for (int i = offset; i <= writeMax; i++) {
                String s = StringGenerator.generate(i, length);
                if (!set(i, s)) {
                    log.error("set error");
                    System.exit(1);
                }
            }
            int countMax = (int) ((1 - Constants.WRITE_RATE) / Constants.WRITE_RATE);
            for (int count = 0; count < countMax; count++) {
                for (int i = offset; i <= writeMax; i++) {
                    String s = StringGenerator.generate(i, length);
                    String result = get(i);
                    if (result != null && !s.equals(result)) {
                        log.error("get error,expected " + s
                                + ",actual " + result);
                        fail.incrementAndGet();
                    } else if (result == null) {
                        miss.incrementAndGet();
                    } else if (result != null && result.equals(s)) {
                        hit.incrementAndGet();
                    }
                }
            }

            barrier.await();
        } catch (Exception e) {
            log.error(e.getMessage(), e);

        }
    }

}