Example #1
0
        public async ValueTask <Tokens[]> Selecttoken(SelectTokenMode Mode)
        {
            DataTable Table;

            Tokens[] ret;

            string cmdstr = @"SELECT
user_id, token, token_secret
FROM token
NATURAL JOIN crawlprocess
";

            switch (Mode)
            {
            case SelectTokenMode.CurrentProcess:
                cmdstr += "WHERE pid = @pid;"; break;

            case SelectTokenMode.RestInStreamer:
                cmdstr += "WHERE rest_needed IS TRUE;"; break;

            case SelectTokenMode.All:
                cmdstr += ";"; break;
            }
            using (MySqlCommand cmd = new MySqlCommand(cmdstr))
            {
                cmd.Parameters.Add("@pid", MySqlDbType.Int32).Value = Selfpid;
                Table = await SelectTable(cmd, IsolationLevel.ReadUncommitted).ConfigureAwait(false);
            }
            if (Table == null)
            {
                return(new Tokens[0]);
            }
            try
            {
                ret = new Tokens[Table.Rows.Count];
                for (int i = 0; i < Table.Rows.Count; i++)
                {
                    ret[i] = (Tokens.Create(config.token.ConsumerKey,
                                            config.token.ConsumerSecret,
                                            Table.Rows[i].Field <string>(1),
                                            Table.Rows[i].Field <string>(2),
                                            Table.Rows[i].Field <long>(0)));
                }
                return(ret);
            }
            finally { Table.Clear(); Table.Dispose(); }
        }
Example #2
0
        ///<summary>いろいろ条件付きでTokenを取得する
        ///Token.UserIdはちゃんとセットされる</summary>
        public async Task <IEnumerable <UserStreamerSetting> > SelectUserStreamerSetting(SelectTokenMode Mode)
        {
            string cmdstr = @"SELECT
user_id, token, token_secret, rest_my_tweet, last_status_id
FROM token
JOIN crawlprocess USING (user_id)
";

            switch (Mode)
            {
            case SelectTokenMode.CurrentProcess:
                cmdstr += "WHERE pid = @pid;"; break;

            case SelectTokenMode.All:
                cmdstr += ";"; break;
            }
            using (var cmd = new MySqlCommand(cmdstr))
            {
                cmd.Parameters.Add("@pid", MySqlDbType.Int32).Value = Selfpid;

                var ret = new List <UserStreamerSetting>();
                if (await ExecuteReader(cmd, (r) =>
                {
                    ret.Add(new UserStreamerSetting()
                    {
                        Token = Tokens.Create(config.token.ConsumerKey, config.token.ConsumerSecret, r.GetString(1), r.GetString(2), r.GetInt64(0)),
                        rest_my_tweet = r.GetBoolean(3),
                        last_status_id = r.GetInt64(4)
                    });
                }, IsolationLevel.ReadUncommitted).ConfigureAwait(false))
                {
                    return(ret.ToArray());
                }
                else
                {
                    return(Enumerable.Empty <UserStreamerSetting>());
                }                                                       //一応全取得に成功しない限り返さない
            }
        }