vnodeWrite.c 2.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

H
refact  
Hongze Cheng 已提交
16 17
#include "vnodeDef.h"

H
refact  
Hongze Cheng 已提交
18
int vnodeProcessWMsgs(SVnode *pVnode, SArray *pMsgs) {
H
more  
Hongze Cheng 已提交
19 20
  SRpcMsg *  pMsg;
  SVnodeReq *pVnodeReq;
H
more  
Hongze Cheng 已提交
21

H
more  
Hongze Cheng 已提交
22 23
  for (int i = 0; i < taosArrayGetSize(pMsgs); i++) {
    pMsg = *(SRpcMsg **)taosArrayGet(pMsgs, i);
H
more  
Hongze Cheng 已提交
24

H
more  
Hongze Cheng 已提交
25
    // ser request version
H
Hongze Cheng 已提交
26
    void *   pBuf = pMsg->pCont;
H
more  
Hongze Cheng 已提交
27
    uint64_t ver = pVnode->state.processed++;
H
Hongze Cheng 已提交
28
    taosEncodeFixedU64(&pBuf, ver);
H
more  
Hongze Cheng 已提交
29

H
more  
Hongze Cheng 已提交
30
    if (walWrite(pVnode->pWal, ver, pMsg->pCont, pMsg->contLen) < 0) {
H
more  
Hongze Cheng 已提交
31 32
      // TODO: handle error
    }
H
more  
Hongze Cheng 已提交
33 34
  }

H
more  
Hongze Cheng 已提交
35
  walFsync(pVnode->pWal, false);
H
refact  
Hongze Cheng 已提交
36

H
more  
Hongze Cheng 已提交
37 38 39
  // Apply each request now
  for (int i = 0; i < taosArrayGetSize(pMsgs); i++) {
    pMsg = *(SRpcMsg **)taosArrayGet(pMsgs, i);
H
more  
Hongze Cheng 已提交
40
    SVnodeReq vReq;
H
refact  
Hongze Cheng 已提交
41

H
more  
Hongze Cheng 已提交
42 43 44 45 46
    // Apply the request
    {
      void *ptr = vnodeMalloc(pVnode, pMsg->contLen);
      if (ptr == NULL) {
        // TODO: handle error
H
more  
Hongze Cheng 已提交
47
      }
H
more  
Hongze Cheng 已提交
48

H
more  
Hongze Cheng 已提交
49 50
      // TODO: copy here need to be extended
      memcpy(ptr, pMsg->pCont, pMsg->contLen);
H
more  
Hongze Cheng 已提交
51

H
Hongze Cheng 已提交
52
      // todo: change the interface here
H
more  
Hongze Cheng 已提交
53 54 55
      uint64_t ver;
      taosDecodeFixedU64(pMsg->pCont, &ver);
      if (tqPushMsg(pVnode->pTq, ptr, ver) < 0) {
H
more  
Hongze Cheng 已提交
56 57 58
        // TODO: handle error
      }

H
more  
Hongze Cheng 已提交
59 60
      vnodeParseReq(pMsg->pCont, &vReq, pMsg->msgType);

H
more  
Hongze Cheng 已提交
61 62
      switch (pMsg->msgType) {
        case TSDB_MSG_TYPE_CREATE_TABLE:
H
more  
Hongze Cheng 已提交
63
          if (metaCreateTable(pVnode->pMeta, &(vReq.ctReq)) < 0) {
H
more  
Hongze Cheng 已提交
64 65 66 67 68 69
            // TODO: handle error
          }

          // TODO: maybe need to clear the requst struct
          break;
        case TSDB_MSG_TYPE_DROP_TABLE:
H
more  
Hongze Cheng 已提交
70
          if (metaDropTable(pVnode->pMeta, vReq.dtReq.uid) < 0) {
H
more  
Hongze Cheng 已提交
71 72
            // TODO: handle error
          }
H
more  
Hongze Cheng 已提交
73 74 75 76 77 78
          break;
        case TSDB_MSG_TYPE_SUBMIT:
          /* code */
          break;
        default:
          break;
H
more  
Hongze Cheng 已提交
79
      }
H
more  
Hongze Cheng 已提交
80

H
more  
Hongze Cheng 已提交
81 82
      pVnode->state.applied = ver;
    }
H
more  
Hongze Cheng 已提交
83 84 85 86 87

    // Check if it needs to commit
    if (vnodeShouldCommit(pVnode)) {
      if (vnodeAsyncCommit(pVnode) < 0) {
        // TODO: handle error
H
more  
Hongze Cheng 已提交
88
      }
H
more  
Hongze Cheng 已提交
89
    }
H
refact  
Hongze Cheng 已提交
90 91
  }

H
more  
Hongze Cheng 已提交
92
  return 0;
H
more  
Hongze Cheng 已提交
93 94
}

H
more  
Hongze Cheng 已提交
95 96 97 98 99
int vnodeApplyWMsg(SVnode *pVnode, SRpcMsg *pMsg, SRpcMsg **pRsp) {
  // TODO
  return 0;
}

H
Hongze Cheng 已提交
100
/* ------------------------ STATIC METHODS ------------------------ */