搭建基于hyperledger fabric的联盟社区（七） --升级chaincode

上个版本的chaincode有很多功能不完备，所以要部署新版本的chaincode。Fabric支持在保留现有状态的前提对chaincode进行升级。

一.新版chaincode

新版本的chaincode增加的功能如下：

1.增加了数据追溯功能，在社区用户发起transaction时，chaincode将自动在用户证书中提取用户信息，将其存储在帖子的字段里。

2.加入了敏感词监管功能，敏感词字典和敏感词过滤功能在外部提供，chaincode通过http api（post 请求）调用服务。

3.加入了模糊查询功能。

chaincode代码：

package main

import (

    "bytes"

    "encoding/json"

    "fmt"

    "strconv"

    "strings"

    "net/http"

    "io/ioutil"

    "crypto/x509"

    "encoding/pem"

    "github.com/hyperledger/fabric/core/chaincode/shim"

    sc "github.com/hyperledger/fabric/protos/peer"

)

type SmartContract struct {

}

type Post struct {

    Id   string `json:"id"`

    OriginalWebsite  string `json:"originalwebsite"`

    OriginalID string `json:"originalid"`

    Title  string `json:"title"`

    Content  string `json:"content"`

    AuthorId  string `json:"authorid"`

    PublishTime  string `json:"publishtime"`

    UpdateTime  string `json:"updatetime"`

    Category  string `json:"category"`

    SourceId  string `json:"sourceid"`

    Labels  string `json:"labels"`

    Follower_num  int `json:"follower_num"`

    Browse_num  int `json:"browse_num"`

    Star_num  int `json:"star_num"`

    UserName  string `json:"username"`

}

type PostLength struct {

    Length int `json:"length"`

}

func (s *SmartContract) Init(APIstub shim.ChaincodeStubInterface) sc.Response {

    return shim.Success(nil)

}

func (s *SmartContract) Invoke(APIstub shim.ChaincodeStubInterface) sc.Response {

    function, args := APIstub.GetFunctionAndParameters()

    if function == "queryPost" {

        return s.queryPost(APIstub, args)

    } else if function == "initLedger" {

        return s.initLedger(APIstub)

    }  else if function == "addPost" {

        return s.addPost(APIstub, args)

    } else if function == "updatePost" {

        return s.updatePost(APIstub, args)

    } else if function == "richQueryPosts" {

        return s.richQueryPosts(APIstub, args)

    } else if function == "getPostNum" {

        return s.getPostNum(APIstub, args)

    }

    return shim.Error("Invalid Smart Contract function name.")

}

func (s *SmartContract) queryPost(APIstub shim.ChaincodeStubInterface, args []string) sc.Response {

    if len(args) != 1 {

        return shim.Error("Incorrect number of arguments. Expecting 1")

    }

    postAsBytes, _ := APIstub.GetState(args[0])

    return shim.Success(postAsBytes)

}

func (s *SmartContract) initLedger(APIstub shim.ChaincodeStubInterface) sc.Response {

    creatorByte,_:= APIstub.GetCreator()

   certStart := bytes.IndexAny(creatorByte, "-----BEGIN")

   if certStart == -1 {

      fmt.Errorf("No certificate found")

   }

   certText := creatorByte[certStart:]

   bl, _ := pem.Decode(certText)

   if bl == nil {

      fmt.Errorf("Could not decode the PEM structure")

   }

   cert, err := x509.ParseCertificate(bl.Bytes)

   if err != nil {

      fmt.Errorf("ParseCertificate failed")

   }

   uname:=cert.Subject.CommonName

    posts := []Post{

        Post{Id: "1", OriginalWebsite: "b", OriginalID: "c", Title: "如何学习人工智能呢？",Content:"好好学习",AuthorId:"f",PublishTime:"g",UpdateTime:"h",Category:"i",SourceId:"j",Labels:"k",Follower_num:100,Browse_num:200,Star_num:300,UserName:uname},

        Post{Id: "2", OriginalWebsite: "bb", OriginalID: "bb", Title: "目前大数据有什么用呢？",Content:"没用",AuthorId:"ff",PublishTime:"gg",UpdateTime:"hh",Category:"ii",SourceId:"jj",Labels:"kk",Follower_num:400,Browse_num:500,Star_num:600,UserName:uname},

    }

    length := PostLength{Length:len(posts)}

    lengthAsBytes,_ := json.Marshal(length)

    APIstub.PutState("POSTLENGTH",lengthAsBytes)

    i := 0

    for i < len(posts) {

        fmt.Println("i is ", i)

        postAsBytes, _ := json.Marshal(posts[i])

        APIstub.PutState("POST"+strconv.Itoa(i), postAsBytes)

        fmt.Println("Added", posts[i])

        i = i + 1

    }

    return shim.Success(nil)

}

func (s *SmartContract) addPost(APIstub shim.ChaincodeStubInterface, args []string) sc.Response {

    if len(args) != 13 {

        return shim.Error("Incorrect number of arguments. Expecting 13")

    }

    filteredtitle := sensitiveSupervision(args[2])

    filteredcontent := sensitiveSupervision(args[3])

    creatorByte,_:= APIstub.GetCreator()

   certStart := bytes.IndexAny(creatorByte, "-----BEGIN")

   if certStart == -1 {

      fmt.Errorf("No certificate found")

   }

   certText := creatorByte[certStart:]

   bl, _ := pem.Decode(certText)

   if bl == nil {

      fmt.Errorf("Could not decode the PEM structure")

   }

   cert, err := x509.ParseCertificate(bl.Bytes)

   if err != nil {

      fmt.Errorf("ParseCertificate failed")

   }

   uname:=cert.Subject.CommonName

    args10,error := strconv.Atoi(args[10])

    args11,error := strconv.Atoi(args[11])

    args12,error := strconv.Atoi(args[12])

    if error != nil{

     fmt.Println("String conversion integer failed!")

    }

    lengthAsBytes, _ := APIstub.GetState("POSTLENGTH")

    length := PostLength{}

    json.Unmarshal(lengthAsBytes,&length)

    newlength := length.Length+1

    var post = Post{Id: strconv.Itoa(newlength), OriginalWebsite: args[0], OriginalID: args[1], Title: filteredtitle,Content:filteredcontent,AuthorId:args[4],PublishTime:args[5],UpdateTime:args[6],Category:args[7],SourceId:args[8],Labels:args[9],Follower_num:args10,Browse_num:args11,Star_num:args12,UserName:uname}

    postAsBytes, _ := json.Marshal(post)

    APIstub.PutState("POST"+strconv.Itoa(newlength), postAsBytes)

    length.Length = newlength

    lengthAsBytes,_ = json.Marshal(length)

    APIstub.PutState("POSTLENGTH",lengthAsBytes)

    return shim.Success(lengthAsBytes)

}

func (s *SmartContract) updatePost(APIstub shim.ChaincodeStubInterface, args []string) sc.Response {

    if len(args) != 14 {

        return shim.Error("Incorrect number of arguments. Expecting 14")

    }

    filteredtitle := sensitiveSupervision(args[3])

    filteredcontent := sensitiveSupervision(args[4])

    creatorByte,_:= APIstub.GetCreator()

   certStart := bytes.IndexAny(creatorByte, "-----BEGIN")

   if certStart == -1 {

      fmt.Errorf("No certificate found")

   }

   certText := creatorByte[certStart:]

   bl, _ := pem.Decode(certText)

   if bl == nil {

      fmt.Errorf("Could not decode the PEM structure")

   }

   cert, err := x509.ParseCertificate(bl.Bytes)

   if err != nil {

      fmt.Errorf("ParseCertificate failed")

   }

   uname:=cert.Subject.CommonName

    args11,error := strconv.Atoi(args[11])

    args12,error := strconv.Atoi(args[12])

    args13,error := strconv.Atoi(args[13])

    if error != nil{

     fmt.Println("String conversion integer failed!")

    }

    var post = Post{Id: args[0], OriginalWebsite: args[1], OriginalID: args[2], Title: filteredtitle,Content:filteredcontent,AuthorId:args[5],PublishTime:args[6],UpdateTime:args[7],Category:args[8],SourceId:args[9],Labels:args[10],Follower_num:args11,Browse_num:args12,Star_num:args13,UserName:uname}

    postAsBytes, _ := json.Marshal(post)

    APIstub.PutState("POST"+args[0], postAsBytes)

    return shim.Success(nil)

}

func (s *SmartContract) richQueryPosts(APIstub shim.ChaincodeStubInterface, args []string) sc.Response {

    if len(args) != 3 {

        return shim.Error("Incorrect number of arguments. Expecting 3")

    }

    var queryString string

    if args[1] == "0" {

        queryString = fmt.Sprintf("{\"selector\":{\"%s\":\"%s\"}}", args[0],args[2])

    } else if args[1] == "1" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$gt\":%s}}}", args[0],args[2])

    } else if args[1] == "2" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$gte\":%s}}}", args[0],args[2])

    } else if args[1] == "3" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$lt\":%s}}}", args[0],args[2])

    } else if args[1] == "4" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$lte\":%s}}}", args[0],args[2])

    } else if args[1] == "5" {

      between := strings.Split(args[2], ",")

      queryString = fmt.Sprintf("{\"selector\":{\"$and\":[{\"%s\":{\"$gte\":%s}},{\"%s\":{\"$lte\":%s}}]}}", args[0],between[0],args[0],between[1])

    } else if args[1] == "6" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$regex\":\"(?i)%s\"}}}", args[0],args[2])

    }  else {

        return shim.Error("Incorrect number of arguments. Expecting 0~6")

    }

    resultsIterator, err := APIstub.GetQueryResult(queryString)

    if err != nil {

        return shim.Error(err.Error())

    }

    defer resultsIterator.Close()

    var buffer bytes.Buffer

    buffer.WriteString("[")

    bArrayMemberAlreadyWritten := false

    for resultsIterator.HasNext() {

        queryResponse, err := resultsIterator.Next()

        if err != nil {

            return shim.Error(err.Error())

        }

        if bArrayMemberAlreadyWritten == true {

            buffer.WriteString(",")

        }

        buffer.WriteString("{\"Key\":")

        buffer.WriteString("\"")

        buffer.WriteString(queryResponse.Key)

        buffer.WriteString("\"")

        buffer.WriteString(", \"Record\":")

        buffer.WriteString(string(queryResponse.Value))

        buffer.WriteString("}")

        bArrayMemberAlreadyWritten = true

    }

    buffer.WriteString("]")

    fmt.Printf("- richQueryPosts:\n%s\n", buffer.String())

    return shim.Success(buffer.Bytes())

}

func  (s *SmartContract) getPostNum(APIstub shim.ChaincodeStubInterface, args []string) sc.Response {

    if len(args) != 3 {

        return shim.Error("Incorrect number of arguments. Expecting 3")

    }

   var queryString string

    if args[1] == "0" {

        queryString = fmt.Sprintf("{\"selector\":{\"%s\":\"%s\"}}", args[0],args[2])

    } else if args[1] == "1" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$gt\":%s}}}", args[0],args[2])

    } else if args[1] == "2" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$gte\":%s}}}", args[0],args[2])

    } else if args[1] == "3" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$lt\":%s}}}", args[0],args[2])

    } else if args[1] == "4" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$lte\":%s}}}", args[0],args[2])

    } else if args[1] == "5" {

      between := strings.Split(args[2], ",")

      queryString = fmt.Sprintf("{\"selector\":{\"$and\":[{\"%s\":{\"$gte\":%s}},{\"%s\":{\"$lte\":%s}}]}}", args[0],between[0],args[0],between[1])

    } else if args[1] == "6" {

      queryString = fmt.Sprintf("{\"selector\":{\"%s\":{\"$regex\":\"(?i)%s\"}}}", args[0],args[2])

    } else {

        return shim.Error("Incorrect number of arguments. Expecting 0~6")

    }

    resultsIterator, err := APIstub.GetQueryResult(queryString)

    if err != nil {

        return shim.Error(err.Error())

    }

    defer resultsIterator.Close()

    i := 0

    for resultsIterator.HasNext() {

      resultsIterator.Next()

      i = i + 1

    }

    fmt.Printf("- getPostNum:\n%s\n", strconv.Itoa(i))

    return shim.Success([]byte(strconv.Itoa(i)))

}

func sensitiveSupervision(arg string) string {

  quertString := fmt.Sprintf("{\"content\":\"%s\"}",arg)

    resp, err := http.Post("http://敏感监管url",

        "application/x-www-form-urlencoded",

        strings.NewReader(quertString))

    if err != nil {

        fmt.Println(err)

    }

    defer resp.Body.Close()

    body, err := ioutil.ReadAll(resp.Body)

    if err != nil {

    }

    return string(body)

}

func main() {

    err := shim.Start(new(SmartContract))

    if err != nil {

        fmt.Printf("Error creating new Smart Contract: %s", err)

    }

}

二.升级链码步骤

2.1 从宿主机将chaincode拷贝进容器

docker cp community2. cli:/opt/gopath/src/github.com/hyperledger/fabric/examples/chaincode/go

2.2 安装新版本chaincode，打包到peer节点

docker exec -it cli bash

peer chaincode install -n mycc -v 2.0 -p github.com/hyperledger/fabric/examples/chaincode/go/community2.

2.3 升级chaincode

peer chaincode upgrade -o orderer.example.com: --tls true --cafile /opt/gopath/src/github.com/hyperledger/fabric/peer/crypto/ordererOrganizations/example.com/orderers/orderer.example.com/msp/tlscacerts/tlsca.example.com-cert.pem -C mychannel -n mycc -v 2.0 -c '{"Args":["init"]}' -P "OR      ('Org1MSP.member','Org2MSP.member')"

这样chaincode就升级完毕了。

巴特西