FEATURE: Add gat command

add get and touch command

docs/ascii-protocol/ch04-command-key-value.md

@@ -110,3 +110,30 @@ decr <key> <delta> [<flags> <exptime> <initial>] [noreply]\r\n
 | "TYPE_MISMATCH"      | 해당 아이템이 key-value 타입이 아님
 | "CLIENT_ERROR"       | 클라이언트에서 잘못된 질의를 했음을 의미. 이어 나오는 문자열을 통해 오류의 원인을 파악 가능. 예) invalid numeric delta argument, cannot increment or decrement non-numeric value
 | "SERVER ERROR"       | 서버 측의 오류로 연산하지 못했음을 의미. 이어 나오는 문자열을 통해 오류의 원인을 파악 가능. 예) out of memory
+
+## gat/gats (get and touch)
+
+gat, gats 명령은 item을 가져옴과 동시에 exptime을 재설정한다.
+
+```
+gat <exptime> <key>*\r\n
+gats <exptime> <key>*\r\n
+```
+
+- \<key\>* - 대상 item의 key string 복수개의 key를 공백을 두고 지정할 수 있다.
+- \<exptime\> - 재설정할 expiretime 값 이다.
+
+gat 명령이 정상 수행되었을 경우, Response string 은 아래와 같이 구성된다.
+
+- key hit된 아이템 정보를 모두 출력
+- key miss된 아이템은 별도 response 없이 생략
+- 응답의 끝에 "END\r\n" 출력
+
+gats 명령의 경우 cas value 도 같이 출력된다.
+
+실패시 string은 아래와 같다.
+
+| Response String      | 설명                     |
+|----------------------|------------------------ |
+| "CLIENT_ERROR"       | 클라이언트에서 잘못된 질의를 했음을 의미. 이어 나오는 문자열을 통해 오류의 원인을 파악 가능. 예) bad command line format
+| "SERVER ERROR"       | 서버 측의 오류로 조회하지 못했음을 의미. 이어 나오는 문자열을 통해 오류의 원인을 파악 가능. 예) out of memory writing get response

memcached.c

@@ -8345,11 +8345,21 @@ static void process_stats_command(conn *c, token_t *tokens, const size_t ntokens
 }
 
 /* ntokens is overwritten here... shrug.. */
-static inline void process_get_command(conn *c, token_t *tokens, size_t ntokens, bool return_cas)
+static inline void process_get_command(conn *c, token_t *tokens, size_t ntokens, bool return_cas, bool should_touch)
 {
     assert(c != NULL);
     token_t *key_token = &tokens[KEY_TOKEN];
     ENGINE_ERROR_CODE ret = ENGINE_SUCCESS;
+    int64_t exptime = 0;
+
+    if (should_touch) {
+        // For get and touch commands, use first token as exptime
+        if (!safe_strtoll(tokens[1].value, &exptime)) {
+            out_string(c, "CLIENT_ERROR bad value");
+            return;
+        }
+        key_token++;
+    }
 
     do {
         while (key_token->length != 0) {
@@ -8362,6 +8372,19 @@ static inline void process_get_command(conn *c, token_t *tokens, size_t ntokens,
             if (ret != ENGINE_SUCCESS) {
                 break; /* ret == ENGINE_ENOMEM */
             }
+            if (should_touch) {
+                ENGINE_ITEM_ATTR attr_id = ATTR_EXPIRETIME;
+                ENGINE_ERROR_CODE gat_ret = ENGINE_SUCCESS;
+                item_attr attr_data;
+
+                attr_data.exptime = realtime(exptime);
+                gat_ret = mc_engine.v1->setattr(mc_engine.v0, c, key_token->value, key_token->length,
+                                                &attr_id, 1, &attr_data, 0);
+                CONN_CHECK_AND_SET_EWOULDBLOCK(gat_ret, c);
+                if (settings.detail_enabled) {
+                    stats_prefix_record_setattr(key_token->value, key_token->length);
+                }
+            }
             key_token++;
         }
         if (ret != ENGINE_SUCCESS) break;
@@ -13124,11 +13147,19 @@ static void process_command_ascii(conn *c, char *command, int cmdlen)
 
     if ((ntokens >= 3) && (strcmp(tokens[COMMAND_TOKEN].value, "get") == 0))
     {
-        process_get_command(c, tokens, ntokens, false);
+        process_get_command(c, tokens, ntokens, false, false);
     }
     else if ((ntokens >= 3) && (strcmp(tokens[COMMAND_TOKEN].value, "gets") == 0))
     {
-        process_get_command(c, tokens, ntokens, true);
+        process_get_command(c, tokens, ntokens, true, false);
+    }
+    else if ((ntokens >= 3) && (strcmp(tokens[COMMAND_TOKEN].value, "gat") == 0))
+    {
+        process_get_command(c, tokens, ntokens, false, true);
+    }
+    else if ((ntokens >= 3) && (strcmp(tokens[COMMAND_TOKEN].value, "gats") == 0))
+    {
+        process_get_command(c, tokens, ntokens, true, true);
     }
     else if ((ntokens == 4) && (strcmp(tokens[COMMAND_TOKEN].value, "mget") == 0))
     {

t/gat.t

@@ -0,0 +1,95 @@
+#!/usr/bin/perl
+
+use strict;
+use Test::More tests =>19;
+use FindBin qw($Bin);
+use lib "$Bin/lib";
+use MemcachedTest;
+
+my $engine = shift;
+my $server = get_memcached($engine);
+my $sock = $server->sock;
+
+my $cmd;
+my $val;
+my $rst;
+my $expire;
+
+# Initialize
+$cmd = "set key 0 0 5"; $val = "datum"; $rst = "STORED";
+mem_cmd_is($sock, $cmd, $val, $rst);
+$cmd = "lop create lkey 0 0 1 error"; $rst = "CREATED";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "sop create skey 0 0 1 error"; $rst = "CREATED";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "mop create mkey 0 0 1 error"; $rst = "CREATED";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "bop create bkey 0 0 1 error"; $rst = "CREATED";
+mem_cmd_is($sock, $cmd, "", $rst);
+
+# Success Cases
+# key value
+$cmd = "gat 1 key";
+$rst = "VALUE key 0 5\n"
+     . "datum\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr key expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+# gat key with KV and collection
+$cmd = "gat 1 key lkey skey mkey bkey";
+$rst = "VALUE key 0 5\n"
+     . "datum\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr key expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr lkey expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr skey expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr mkey expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr bkey expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+# gats command
+$cmd = "gats 1 key";
+$rst = "VALUE key 0 5 1\n"
+     . "datum\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "getattr key expiretime";
+$rst = "ATTR expiretime=1\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+
+# Fail Cases
+# bad value
+$cmd = "set key 0 0 5"; $val = "datum"; $rst = "STORED";
+mem_cmd_is($sock, $cmd, $val, $rst);
+$cmd = "gat str key"; $rst = "CLIENT_ERROR invalid exptime argument";
+mem_cmd_is($sock, $cmd, "", $rst);
+# not exist key
+$expire = time() - 1;
+$cmd = "gat $expire key";
+$rst = "VALUE key 0 5\n"
+     . "datum\n"
+     . "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+$cmd = "gat 1 key"; $rst = "END";
+mem_cmd_is($sock, $cmd, "", $rst);
+
+# after test
+release_memcached($engine, $server);