カテゴリーアーカイブ: 日記

<?xml version="1.0" encoding="UTF-8"?>

<web-app>

  <display-name>Desyura</display-name>

  <servlet>

    <servlet-name>QuartzInitializer</servlet-name>

    <servlet-class>

    org.quartz.ee.servlet.QuartzInitializerServlet</servlet-class>

    <init-param>

      <param-name>shutdown-on-unload</param-name>

      <param-value>true</param-value>

    </init-param>

    <init-param>

      <param-name>start-scheduler-on-load</param-name>

      <param-value>false</param-value>

    </init-param>

    <load-on-startup>1</load-on-startup>

  </servlet>

  <servlet>

    <display-name>RobotInitializer</display-name>

    <servlet-name>RobotInitializer</servlet-name>

    <servlet-class>

    net.maple4ever.desyura.RobotInitializer</servlet-class>

    <load-on-startup>2</load-on-startup>

  </servlet>

</web-app>

package net.maple4ever.desyura;

import java.text.ParseException;

import java.util.Calendar;

import java.util.Locale;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServlet;

import org.quartz.CronTrigger;

import org.quartz.JobDetail;

import org.quartz.Scheduler;

import org.quartz.SchedulerException;

import org.quartz.Trigger;

import org.quartz.impl.StdSchedulerFactory;

public class RobotInitializer extends HttpServlet {

    public static final String START_TIME = "START_TIME";

    private static final long serialVersionUID = 1L;

    public RobotInitializer() {

        super();

    }

    @Override

    public void init() throws ServletException {

        // QuartzInitializerServlet で作成したスケジューラを取得

        Scheduler sched = null;

        try {

            sched = (new StdSchedulerFactory()).getScheduler();

        } catch (SchedulerException e) {

            throw new ServletException(e);

        }

        // Cron 形式のトリガー作成（とりあえず 5分）

        Trigger trigger = new CronTrigger("trigger1", "group1");

        try {

            ((CronTrigger) trigger).setCronExpression("0 0/5 * * * ?");

        } catch (ParseException e) {

            throw new ServletException(e);

        }

        // JobDetail にコールバッククラス登録

        JobDetail jobDetail = new JobDetail(

                "Twitter Bot"

                , "Twitter Bot"

                , TwitterBotService.class

                , true

                , true

                , true);

        // 起動時間を JobDataMap に格納（twitter のロケールも日本にすること）

        jobDetail.getJobDataMap().put(START_TIME,

                Calendar.getInstance(Locale.JAPAN));

        // スケジュール開始

        try {

            sched.scheduleJob(jobDetail, trigger);

            sched.start();

        } catch (SchedulerException e) {

            throw new ServletException(e);

        }

    }

}

package net.maple4ever.desyura;

import java.util.ArrayList;

import java.util.Calendar;

import java.util.Collections;

import java.util.List;

import java.util.Date;

import org.quartz.JobExecutionContext;

import org.quartz.StatefulJob;

import twitter4j.Status;

import twitter4j.Twitter;

import twitter4j.TwitterException;

public class TwitterBotService implements StatefulJob {

    // StatefulJob は同時実行されないが手抜き synchronized しておく

    static private List<Long> read = (List<Long>) Collections

            .synchronizedList(new ArrayList<Long>());

    static private final int PAGE_COUNT = 20;

    public void execute(JobExecutionContext context) {

        // スケジュールイベント

        System.out.println("Fire! " + context.getScheduledFireTime());

        // 初回起動時間取得

        Date fistTime = ((Calendar) context.getMergedJobDataMap().get(

                RobotInitializer.START_TIME)).getTime();

        // twitter 接続

        Twitter tw = new Twitter("ユーザ名", "パスワード");

        String tweet;

        try {

            // twitter から Mentions をもらう

            List<Status> status = tw.getMentions();

            for(Status stat : status) {

                // 起動時より前の返信は答えない

                if(fistTime.after(stat.getCreatedAt())) continue;

                // すでに返信していたら答えない

                if(read.contains(stat.getId())) continue;

                // 自分のスクリーンネームを相手に置換してつぶやく

                tweet = stat.getText().replaceAll(

                        stat.getInReplyToScreenName(),

                        stat.getUser().getScreenName());

                tw.updateStatus(tweet, stat.getId());

                System.out.println(tweet);

                // 読んだリストに入れておく

                read.add(stat.getId());

            }

            // getMentions() は 20件取得なのでそれを超えたら忘れる

            if(read.size() > PAGE_COUNT) {

                read.remove(0);

            }

        } catch (TwitterException e) {

            // 投稿失敗しても次に期待する

            System.out.println(e.getStackTrace());

        }

    }

}