com.github.liyp.cassandra.readafterwrite.WriteOpt.java Source code

Introduction

Here is the source code for com.github.liyp.cassandra.readafterwrite.WriteOpt.java

Source

/*
 * Copyright  2017 liyp (liyp.yunpeng@gmail.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.github.liyp.cassandra.readafterwrite;

import java.util.Date;
import java.util.concurrent.atomic.AtomicBoolean;

import org.apache.commons.lang3.RandomStringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class WriteOpt implements Runnable {

    private final static Logger logger = LoggerFactory.getLogger(WriteOpt.class);

    public static volatile AtomicBoolean readyToWrite = new AtomicBoolean(true);

    public static volatile String pk;
    public static volatile String ck;
    public static volatile String data;
    public static volatile Date date;

    private RAWBeanDao dao;

    public WriteOpt(RAWBeanDao dao) {
        this.dao = dao;
    }

    private void write() {
        pk = RandomStringUtils.randomNumeric(9);
        ck = RandomStringUtils.randomAlphabetic(9);
        data = "data_" + RandomStringUtils.randomAlphanumeric(30);
        date = new Date();
        RAWBean bean = new RAWBean(pk, ck, data, date);
        dao.getMappingManager().mapper(RAWBean.class).save(bean);
    }

    public void run() {
        System.out.println("w." + readyToWrite.get());
        if (readyToWrite.get()) {
            write();
            readyToWrite.compareAndSet(true, false);
        } else {
            RAWBean bean = dao.getMappingManager().mapper(RAWBean.class).get(pk, ck);
            if (bean == null || !bean.getData().equals(data)) {
                System.out.println(String.format("R-A-W error. curr pk=%s ck=%s data=%s date=%s | {}", bean));
            }
        }
    }

}