IntegralRecordOperation.java

package com.lst.lc.hbase.service;

import java.util.ArrayList;
import java.util.List;

import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.client.ResultScanner;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;

import com.lst.lc.dao.OperationDao;
import com.lst.lc.hbase.model.IntegralRecord;
import com.lst.lc.utils.DateUtils;

@Service
public class IntegralRecordOperation {

        @Autowired
        private HbaseOperation mHbaseOperation;
        @Autowired
        private ModelMapping modelMapping;

        @Autowired
        @Qualifier("operationDao")
        private OperationDao operationDao;

        public void add(IntegralRecord integralRecord) {
                if ((integralRecord == null)
                                || (integralRecord.getKey().equals(""))) {
                        // 传值有问题，处理一下
                        return;
                }
                mHbaseOperation.insertData("lc_integralRecord",
                                integralRecord.getKey(), "attr", "date",
                                integralRecord.getDate());
                mHbaseOperation.insertData("lc_integralRecord",
                                integralRecord.getKey(), "var", "integral",
                                integralRecord.getIntegral());
        }

        public IntegralRecord get(String key) {
                Result rs = mHbaseOperation.queryByRowKey("lc_integralRecord",
                                key);
                return modelMapping.integralRecordMapping(rs, key);
        }

        public List<IntegralRecord> getRecent(String email, int day) {
                ResultScanner rs = mHbaseOperation.queryIntegral(email, day);
                List<IntegralRecord> lists = modelMapping
                                .integralListMapping(rs);
                IntegralRecord record = get(email + DateUtils.getCurrentDate());
                if (lists == null) {
                        lists = new ArrayList<IntegralRecord>();
                }
                if (record == null) {
                        record = new IntegralRecord(email, "0");
                }
                lists.add(record);
                if (lists.size() < day) {
                        List<IntegralRecord> records = new ArrayList<IntegralRecord>();
                        for (int i = day - 1; i > 0; i--) {
                                String before = DateUtils.getBefore(i);
                                String key = email + before;
                                boolean flag = false;
                                for (int j = 0; j < lists.size(); j++) {
                                        if (lists.get(j).getKey().equals(key)) {
                                                flag = true;
                                                records.add(lists.get(j));
                                                lists.remove(j);
                                                break;
                                        }
                                }
                                if (!flag) {
                                        IntegralRecord record1 = new IntegralRecord(
                                                        key,
                                                        DateUtils.getBeforeForShow(i),
                                                        "0");
                                        records.add(record1);
                                }

                        }
                        records.add(record);
                        return records;
                }
                lists.add(record);
                return lists;
        }

        public void update(String email, String operation) {
                int integral = operationDao.getIntegral(operation);
                String key = email + DateUtils.getCurrentDate();
                IntegralRecord record = get(key);
                IntegralRecord todayRecord = null;
                if (record == null) {
                        todayRecord = new IntegralRecord(email,
                                        String.valueOf(integral));
                } else {
                        int add;
                        if (record.getIntegral() == null
                                        || record.getIntegral().equals("")) {
                                add = 0;
                        } else {
                                add = Integer.parseInt(record.getIntegral());
                        }
                        todayRecord = new IntegralRecord(email,
                                        String.valueOf(integral + add));
                }
                add(todayRecord);
        }

        public void updateTest(String date, int integral, String key) {
                IntegralRecord record = get(key);
                IntegralRecord todayRecord = null;
                System.out.println(key);
                if (record == null) {
                        todayRecord = new IntegralRecord();
                        todayRecord.setKey(key);
                        todayRecord.setIntegral(String.valueOf(integral));
                        todayRecord.setDate(date);

                } else {
                        int add;
                        if (record.getIntegral() == null
                                        || record.getIntegral().equals("")) {
                                add = 0;
                        } else {
                                add = Integer.parseInt(record.getIntegral());
                        }
                        todayRecord = new IntegralRecord();
                        todayRecord.setKey(key);
                        todayRecord.setIntegral(String.valueOf(integral + add));
                        todayRecord.setDate(date);
                }
                add(todayRecord);
        }
}