|
- package handle
-
- import (
- "bufio"
- "context"
- "encoding/json"
- "errors"
- "fmt"
- "fts/config"
- "fts/etcdclient"
- "github.com/gorilla/websocket"
- _ "github.com/ipfs/go-ipfs-api"
- shell "github.com/ipfs/go-ipfs-api"
- "io"
- "io/ioutil"
- "log"
- "os"
- "os/exec"
- "path"
- "path/filepath"
- "strconv"
- "strings"
- "time"
- )
-
- var gobalLoginUserName string
-
- //key:filepath,value:hash
- var gobalFileMap = make(map[string] string)
- var gobalFileUpdateTimeMap = make(map[string] string)
-
- //var gobalFileChangeMap = make(map[string] string)
- var getLocalFileListDir string
- var gobalSubscriptionFileChangeSwitch int =0 //订阅文件变更开关
-
-
-
- var ipfsPath=os.Getenv("IPFS-PATH")
-
- /**
- 文件上传下载进度
- */
- type processStruct struct {
- Size string `json:"size"`
- CurrentSize string `json:"currentSize"`
- Unit string `json:"unit"`
- CurrentUnit string `json:"currentUnit"`
- Process float64 `json:"process"`
- Hash string `json:"hash"`
- CommitHistoryHash string `json:"commitHistoryHash"`
- }
-
- func main() {
-
-
- //config.InitConfig()
- //InitLocalWorkSpace("320872793405132801","test1")
- //
- //DownCommand("QmTp2hEo8eXRp6wg7jXv1BLCMh5a4F3B7buAUZNZUu772j","testOne","hello.txt","a/b/")
- //
- //UploadCommand("C:\\Users\\yuan_rh\\Downloads\\QmRzN7uW6HCVAkGMXNWv3rC9dqPJijvEgvtW6DKsQQE8Js","QmRzN7uW6HCVAkGMXNWv3rC9dqPJijvEgvtW6DKsQQE8Js","testOne","a/b/")
- //
- //GetLocalFileList("testOne")
- }
-
-
-
- /**
- 初始化本地工作目录
- @param userId 用户ID
- @param projectName 项目名称
- */
- func InitLocalWorkSpace(conn *websocket.Conn,userName,projectName string) (error){
-
- //空格路径处理
- ipfsPath=strings.Replace(os.Getenv("IPFS-PATH"),"\"","",1)+"\\ipfs.exe"
-
-
- //初始化当前登陆用户
- gobalLoginUserName = userName
- getLocalFileListDir = fmt.Sprint(config.LocalWorkSpaceDir+gobalLoginUserName+"\\"+projectName)
- // 检查本地目录是否存在
- var projectPath = config.LocalWorkSpaceDir+userName+"\\"+projectName
- _,err := os.Stat(projectPath)
- if err != nil {
- //创建文件目录
- os.MkdirAll(projectPath, os.ModePerm)
- }
-
- log.Println("切换本地工作目录至 "+projectPath)
-
- if err := conn.WriteMessage(websocket.TextMessage, []byte(fmt.Sprint(projectPath))); err != nil {
- log.Println(err)
- return err
- }
-
- return nil
- }
-
- func watchWalkfunc(filePath string, info os.FileInfo, err error) error {
-
- if info == nil{
- return nil
- }
-
- if info.IsDir()==true{
- //config.GobalWatch.Remove(filePath)
- err = config.GobalWatch.Add(filePath)
- if err != nil {
- log.Println(err)
- return err
- }
- }
- return nil
- }
-
- /**
- 初始化本地客户端配置
- */
- func InitClientConfig(ipfsApi,ipfsBootstrap string) error{
-
- //空格路径处理
- ipfsPath=strings.Replace(os.Getenv("IPFS-PATH"),"\"","",1)+"\\ipfs.exe"
-
-
- config.ServerIpfsUrl = ipfsApi
- log.Println("初始化客户端配置项IPFS—API:"+config.ServerIpfsUrl)
- ipfsBootstraps := strings.Split(ipfsBootstrap,";")
- for _, simpleBootstrap := range ipfsBootstraps {
- log.Println("增加引导节点:"+simpleBootstrap)
- cmd := exec.Command(ipfsPath,"bootstrap", "add", simpleBootstrap)
- err :=cmd.Run()
- if err!=nil{
- return err
- }
- }
- return nil
- }
-
- /**
- 下载指令
- @param hash ipfs哈希值
- @param projectName 项目名称
- @para fileName 文件名称
- @param dir 云文件目录
- */
- func DownCommand(conn *websocket.Conn, hash, projectName, fileName, dir string) error{
-
- absoluteDir := config.LocalWorkSpaceDir+gobalLoginUserName+"\\"+projectName+"\\"+dir
- //检查目录
- _,err := os.Stat(absoluteDir)
- if err != nil {
- //创建文件目录
- err = os.MkdirAll(absoluteDir, os.ModePerm)
- if err!=nil{
- log.Println(err)
- return err
- }
- }
-
-
- var downloading bool = false
-
- //检测文件打开状态
- //tfile,err := os.OpenFile(fmt.Sprint(absoluteDir+"\\"+fileName),os.O_RDWR,1)
- //
- //if err != nil && (!os.IsNotExist(err)) {
- //
- // log.Println("文件被占用,请关闭打开的软件")
- // if err := conn.WriteMessage(websocket.TextMessage, []byte("-2")); err != nil {
- // return err
- // }
- // return err
- //}
- //defer tfile.Close()
-
- //serverSh := shell.NewShell(config.ServerIpfsUrl)
- ////检测引导节点是否连接成功
- //isUp := serverSh.IsUp()
- //if !isUp {
- // if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- // return err
- // }
- // return nil
- //}
-
- cmd := exec.Command(ipfsPath,"get", hash,"-o",fmt.Sprint(absoluteDir+"\\"+fileName))
- progress := make(chan string,10000)
- var stdout, stderr []byte
- var errStdout, errStderr error
- stdoutIn, _ := cmd.StdoutPipe()
- stderrIn, _ := cmd.StderrPipe()
- cmd.Start()
- go func() {
- stdout, errStdout = copyAndCapture(os.Stdout, stdoutIn, progress)
- }()
- go func() {
- stderr, errStderr = copyAndCapture(os.Stderr, stderrIn, progress)
- }()
-
- go func(){
- millSeconds := time.Now().UnixNano() / 1e6
- for content := range progress { // 通道关闭后会退出for range循环
- current :=time.Now().UnixNano() / 1e6
-
- if current-millSeconds>500{
- projson,err := contentToJSONByte(content)
- if projson==nil && err==nil{
- continue
- }
-
- if err != nil {
- log.Printf("json.Marshal error %s\n", err)
- }
- millSeconds = current
- downloading = true
-
- if err := conn.WriteMessage(websocket.TextMessage, projson); err != nil {
- log.Println(err)
- break
- }
-
- }
- if strings.Index(content,"100.00%")!=-1{
-
- projson,err := contentToJSONByte(content)
- if projson==nil && err==nil{
- continue
- }
-
- if err != nil {
- log.Printf("json.Marshal error %s\n", err)
- }
- downloading = true
- if err := conn.WriteMessage(websocket.TextMessage, projson); err != nil {
- panic(err)
- }
- break
-
- }
- }
- }()
- log.Println("下载资源连接中...")
-
- //设置30秒连接超时
- go func() {
- index :=0
- for true{
- index++
- if downloading==true{
- return
- }
- time.Sleep(time.Duration(1)*time.Second)
- if downloading==false && index==30{
- err = cmd.Process.Kill()
- log.Println("进程连接超时30s已被Kill")
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return
- }
- return
- }
- }
- }()
-
- err = cmd.Wait()
- if err != nil {
- log.Printf("cmd.Run() failed with %s\n", err)
- }
- if errStdout != nil || errStderr != nil {
- log.Printf("failed to capture stdout or stderr\n")
- }
- outStr := string(stdout)
- log.Printf("out:%s", outStr)
-
- //TODO test 更新数据库hash
- time.Sleep(200*time.Millisecond)
- key := gobalLoginUserName+"\\"+projectName+"\\"+dir+"\\"+fileName
- err = etcdclient.ReplaceInto(key,hash)
- if err != nil {
- log.Println(err)
- return err
- }
- config.GobalWatchChannelMap[config.LocalWorkSpaceDir+gobalLoginUserName+"\\"+projectName] <- ";"
-
- if err==nil{
- log.Println("下载成功")
- }
-
- //time.Sleep(time.Duration(6)*time.Second)
-
- defer close(progress)
-
- return nil
- }
-
- func contentToJSONByte(content string) ([]byte,error){
- sts :=strings.Split(content," ")
- if len(sts)<8{
- log.Println("字符长度小于8")
- return nil,nil
- }
- var processFloat float64
- if (len(sts)==9 || len(sts)==8){
- processFloat,_ =strconv.ParseFloat(strings.Replace(sts[7],"%","",1), 64)
- }else{
- processFloat,_ =strconv.ParseFloat(strings.Replace(sts[8],"%","",1), 64)
- }
-
- if processFloat==0{
- //log.Println("当前进度0")
- return nil,nil
- }
-
- pro :=&processStruct{
- Size:sts[4],
- CurrentSize: sts[1],
- Unit: sts[2],
- CurrentUnit: sts[5],
- Process: processFloat,
- Hash: "",
-
- }
- projson,err :=json.Marshal(pro)
-
- return projson,err
- }
-
-
- /**
- 上传本地文件
- @param absolutePath 文件本地绝对路径
- @param fileName 文件名称
- @param projectName 项目名称
- @param dir 云文件目录
- */
- func UploadCommand(conn *websocket.Conn,absolutePath,fileName,projectName,dir,currentHistoryHash,note,creator string,milestone bool) error{
-
- //本地拷贝文件
- absoluteDir := config.LocalWorkSpaceDir+gobalLoginUserName+"\\"+projectName+"\\"+dir
- //检查目录
- _,err := os.Stat(absoluteDir)
- if err != nil {
- //创建文件目录
- err = os.MkdirAll(absoluteDir, os.ModePerm)
- if err!=nil{
- return err
- }
- }
-
- //检测文件打开状态
- ///*tfile,err := os.OpenFile(absolutePath,os.O_RDWR,1)
- //if err != nil {
- // log.Println("文件被占用,请关闭打开的软件")
- // if err := conn.WriteMessage(websocket.TextMessage, []byte("-2")); err != nil {
- // return err
- // }
- // return err
- //}
- //defer tfile.Close()*/
-
- serverSh := shell.NewShell(config.ServerIpfsUrl)
- //serverSh.SetTimeout(time.Duration(30)*time.Second)
- log.Println("检测引导节点存活情况"+config.ServerIpfsUrl)
- //检测引导节点是否连接成功
- isUp := serverSh.IsUp()
- if !isUp {
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return err
- }
- return nil
- }
-
-
- var uploading bool=false
-
- cmd := exec.Command(ipfsPath, "add",absolutePath)
- uploadProgress := make(chan string,10000)
- var stdout, stderr []byte
- var errStdout, errStderr error
- stdoutIn, _ := cmd.StdoutPipe()
- stderrIn, _ := cmd.StderrPipe()
- cmd.Start()
- go func() {
- stdout, errStdout = copyAndCapture(os.Stdout, stdoutIn, uploadProgress)
- }()
- go func() {
- stderr, errStderr = copyAndCapture(os.Stderr, stderrIn, uploadProgress)
- }()
-
- go func(){
- first := true
- millSeconds := time.Now().UnixNano() / 1e6
- current :=time.Now().UnixNano() / 1e6
- for content := range uploadProgress { // 通道关闭后会退出for range循环
- if first {
- projson,err := contentToJSONByte(content)
- if projson==nil && err==nil{
- continue
- }
-
- if err != nil {
- log.Println("json.Marshal error %s\n", err)
- }
- uploading=true
- if err := conn.WriteMessage(websocket.TextMessage, projson); err != nil {
- break
- }
- millSeconds = current
- first=false
- }
- if current-millSeconds>500{
- projson,err := contentToJSONByte(content)
- if projson==nil && err==nil{
- continue
- }
-
- if err != nil {
- log.Println("json.Marshal error %s\n", err)
- }
- uploading=true
- if err := conn.WriteMessage(websocket.TextMessage, projson); err != nil {
- break
- }
- millSeconds = current
- }
- if (strings.Index(content,"90.00%")!=-1){
-
- projson,err := contentToJSONByte(content)
- if projson==nil && err==nil{
- continue
- }
-
-
- if err != nil {
- log.Println("json.Marshal error %s\n", err)
- }
- uploading=true
- if err := conn.WriteMessage(websocket.TextMessage, projson); err != nil {
- panic(err)
- }
- break
- }
- }
- }()
-
- log.Println("文件上传中...")
-
- //设置30秒连接超时
- go func() {
- index :=0
- for true{
- index++
- if uploading==true{
- return
- }
- time.Sleep(time.Duration(1)*time.Second)
- if uploading==false && index==30{
- err = cmd.Process.Kill()
- log.Println("进程连接超时30s已被Kill")
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return
- }
- return
- }
- }
- }()
-
- err = cmd.Wait()
-
- if err != nil {
- log.Println("cmd.Run() failed with %s\n", err)
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return err
- }
- return err
- }
- if errStdout != nil || errStderr != nil {
- log.Println("failed to capture stdout or stderr\n")
- }
- outStr := string(stdout)
- log.Printf("out:%s", outStr)
- /*sh.Get(hash,absoluteDir+fileName)
- if err != nil {
- return "",err
- }*/
- defer close(uploadProgress)
-
- prog := new(processStruct)
- prog.Hash=strings.Split(outStr," ")[1]
- prog.Process=100.00
-
- sh := shell.NewShell(config.GobalIpfsUrl)
- //sh.SetTimeout(time.Duration(30)*time.Second)
- objectStat,err :=sh.ObjectStat(prog.Hash)
- if err != nil {
- log.Println(err)
- return err
- }
- prog.Size=strconv.Itoa(objectStat.CumulativeSize)
-
- //
- cmd = exec.Command(ipfsPath,"dht","provide",prog.Hash)
- err = cmd.Run()
- if err != nil {
- log.Println(err)
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return err
- }
- return err
- }
-
- //serverSh.SetTimeout(time.Duration(600)*time.Second)
- //判断备份节点对等节点是否已添加连接
- idOut,err :=sh.ID()
- localId :=idOut.ID
- swarmConnInfos,err :=serverSh.SwarmPeers(context.Background())
- hasConnect := false
- for _,swarmconn := range swarmConnInfos.Peers {
- if swarmconn.Peer==localId{
- hasConnect=true
- break
- }
- }
-
- if !hasConnect{
- log.Println("中继处理")
- swarmConnectAddr :="/ipfs/"+"12D3KooWER8uoGrzeLuJHTXoTMnR4jjHfpcA6ZcpXBaQNJvG5jMP"+"/p2p-circuit/ipfs/"+localId
- errT := serverSh.SwarmConnect(context.Background(),swarmConnectAddr)
- if errT!=nil{
- log.Println("中继失败,引导节点备份失败")
- log.Println(err)
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return err
- }
- }
- }
-
- err = serverSh.Pin(prog.Hash)
- if err!=nil{
- log.Println("引导节点备份失败")
- log.Println(err)
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return err
- }
- }
-
- log.Println("引导节点文件备份成功")
-
- //文件不存在则进行本地文件夹拷贝
- if !fileExist(fmt.Sprint((absoluteDir+"\\"+fileName))) {
- err = sh.Get(prog.Hash,fmt.Sprint((absoluteDir+"\\"+fileName)))
- if err != nil {
- log.Println(err)
- return err
- }
- }
-
- //记录历史
- filenameall := path.Base(fileName)
- filesuffix := path.Ext(fileName)
- fileprefix := filenameall[0:len(filenameall) - len(filesuffix)]
- commitFilePath := absoluteDir+"\\"+fileprefix+".commit"
-
- commitHistoryHash,err := commitRecord(commitFilePath,currentHistoryHash,prog.Hash,note,creator,milestone)
- if err != nil {
- log.Println("历史文件写入失败")
- log.Println(err)
- if err := conn.WriteMessage(websocket.TextMessage, []byte("-1")); err != nil {
- return err
- }
- return err
- }
-
- prog.CommitHistoryHash=commitHistoryHash
- projson,err :=json.Marshal(prog)
-
- if err := conn.WriteMessage(websocket.TextMessage, projson); err != nil {
- log.Println(err)
- return err
- }
-
- //更新数据库hash
- key := gobalLoginUserName+"\\"+projectName+"\\"+dir+"\\"+fileName
- err = etcdclient.ReplaceInto(key,prog.Hash+";0")
- if err != nil {
- log.Println(err)
- return err
- }
- config.GobalWatchChannelMap[config.LocalWorkSpaceDir+gobalLoginUserName+"\\"+projectName] <- ";"
- log.Println("上传成功")
-
- return nil
- }
-
- /**
- 记录提交记录
- */
- func commitRecord(path,currentHistoryHash,hash,note,creator string, milestone bool) (string,error){
-
- commitHistory := new(commitHistory)
-
- //历史文件不存在则创建
- localSh :=shell.NewShell(config.GobalIpfsUrl)
- localSh.SetTimeout(5*time.Second)
- if len(currentHistoryHash)!=0 {
- os.Remove(path)
- err := localSh.Get(currentHistoryHash,path)
- if err != nil {
- log.Println("历史版本管理文件下载失败")
- return "",err
- }
- }
-
- //初始化历史管理文件
- exist := fileExist(path)
- if !exist {
- commitFile,err := os.Create(path)
- if err != nil {
- log.Println("历史版本管理文件创建失败")
- return "",err
- }
- commitFile.Close()
- }
-
- //设置文件隐藏属性
- attribCmd :=exec.Command("attrib","+h",path)
- err :=attribCmd.Run()
- if err != nil {
- log.Println("设置文件隐藏属性失败")
- return "",err
- }
-
- //读取历史管理文件
- content ,err :=ioutil.ReadFile(path)
-
- if len(content)!=0{
- rows :=strings.Split(string(content),"\n")
- endRow :=rows[len(rows)-2]
- columns :=strings.Split(endRow,"\t")
- commitHistory.Version,_=strconv.Atoi(columns[6])
- commitHistory.Version++
- commitHistory.ParentHash=columns[1]
- }else{
- commitHistory.Version=1
- commitHistory.ParentHash="0000000000000000000000000000000000000000"
- }
- commitHistory.CurrentHash=hash
- commitHistory.Milestone = milestone
- commitHistory.Creator = creator
- commitHistory.Note = note
- commitHistory.CreateTime=time.Now().Unix()
-
- if commitHistory.ParentHash==commitHistory.CurrentHash{
- return commitHistory.CurrentHash,nil
- }
-
- file,err :=os.OpenFile(path,os.O_APPEND,0666)
- if err != nil{
- log.Println(err)
- return "",err
- }
- //写入历史管理文件
- w :=bufio.NewWriter(file)
- writeContent:=fmt.Sprintf("%v\t%v\t%v\t%v\t%v\t%v\t%v\t",commitHistory.ParentHash,commitHistory.CurrentHash,commitHistory.Creator,commitHistory.CreateTime,commitHistory.Note,commitHistory.Milestone,commitHistory.Version)
- fmt.Fprintln(w,writeContent)
- err = w.Flush()
- if err != nil {
- log.Println("历史版本管理文件写入失败")
- return "",err
- }
- file.Close()
-
- addFile,err :=os.Open(path)
- if err != nil{
- log.Println(err)
- return "",err
- }
- defer addFile.Close()
- //add 历史管理文件
- historyHash,err:=localSh.Add(addFile)
- if err != nil {
- log.Println("历史版本管理文件上传失败")
- return "",err
- }
-
- serverSh :=shell.NewShell(config.ServerIpfsUrl)
- serverSh.SetTimeout(60*time.Second)
- err = serverSh.Pin(historyHash)
- if err != nil {
- log.Println("历史版本管理文件备份失败")
- return "",err
- }
- return historyHash,nil
- }
-
- func copyAndCapture(w io.Writer, r io.Reader, progress chan string) ([]byte, error) {
- var out []byte
- buf := make([]byte, 1024, 1024)
- for {
- n, err := r.Read(buf[:])
- if n > 0 {
- d := buf[:n]
- out = append(out, d...)
- progress <- string(d)
-
- }
- if err != nil {
- // Read returns io.EOF at the end of file, which is not an error for us
- if err == io.EOF {
- err = nil
- }
- return out, err
- }
- }
- // never reached
- panic(true)
- return nil, nil
- }
-
- /**
- 单个文件信息
- */
- type simpleFileInfo struct {
- Name string `json:"name" `
- Extension string `json:"extension"`
- RelativePath string `json:"relativePath"`
- AbsolutePath string `json:"absolutePath"`
- }
-
- var gobalFolderFileMap map[string] *simpleFileInfo
- var gobalRelativePath string
- /**
- 获取指定目录或文件的文件信息,如果是目录递归获取文件信息
- @param id 文件id
- */
- func GetFolderFileInfo(conn *websocket.Conn,absolutePath string) error{
-
- gobalFolderFileMap = make(map[string] *simpleFileInfo)
- fileInfo,err :=os.Stat(absolutePath)
- if err!=nil{
- log.Println(err)
- return err
- }
-
- log.Println(filepath.Dir(absolutePath))
- //单个文件处理
-
- if !fileInfo.IsDir() {
- simpleFileInfo := new(simpleFileInfo)
- simpleFileInfo.Name=fileInfo.Name()
- simpleFileInfo.Extension=path.Ext(absolutePath)
- simpleFileInfo.RelativePath=""
- simpleFileInfo.AbsolutePath=absolutePath
- gobalFolderFileMap[absolutePath]=simpleFileInfo
- bytes,err :=json.Marshal(gobalFolderFileMap)
- if err != nil {
- log.Println(err)
- return err
- }
- if err := conn.WriteMessage(websocket.TextMessage, bytes); err != nil {
- log.Println(err)
- return err
- }
- return nil
- }
-
- //文件目录处理
- gobalRelativePath = filepath.Dir(absolutePath)
- err =filepath.Walk(absolutePath, myWalkfunc)
-
- if err != nil {
- log.Println(err)
- return err
- }
-
- bytes,err :=json.Marshal(gobalFolderFileMap)
- if err != nil {
- log.Println(err)
- return err
- }
- if err := conn.WriteMessage(websocket.TextMessage, bytes); err != nil {
- log.Println(err)
- return err
- }
-
- return nil
- }
-
- func myWalkfunc(path string, info os.FileInfo, err error) error {
-
- if info.IsDir()==false{
- simpleFileInfo := new(simpleFileInfo)
- simpleFileInfo.Name=info.Name()
- simpleFileInfo.Extension=filepath.Ext(path)
- simpleFileInfo.RelativePath=filepath.Dir(strings.Replace(path,gobalRelativePath,"",1))
- simpleFileInfo.AbsolutePath=path
- gobalFolderFileMap[path]=simpleFileInfo
- return nil
- }
-
- return nil
- }
-
- /**
- 本地文件是否存在
- */
- func fileExist(path string) bool {
- _, err := os.Lstat(path)
- return !os.IsNotExist(err)
- }
-
- /**
- 获取本地文件列表
- */
- func SubscriptionFileChange(conn *websocket.Conn, projectName string) error{
- getLocalFileListDir = fmt.Sprint(config.LocalWorkSpaceDir+gobalLoginUserName+"\\"+projectName)
- projectPath := getLocalFileListDir
- log.Println("切换文件列表:"+projectPath)
- keyPrefix := gobalLoginUserName+"\\"+projectName+"\\"
-
- //添加监控
- err := filepath.Walk(projectPath,watchWalkfunc)
- if err != nil {
- log.Println(err)
- return err
- }
- //初始化通道
- if config.GobalWatchChannelMap[projectPath] != nil {
- close(config.GobalWatchChannelMap[projectPath])
- }
- config.GobalWatchChannelMap[projectPath]=make(chan string,100)
- log.Println(projectPath+"添加文件监控")
- log.Println(config.GobalWatchChannelMap[projectPath])
-
- //定期校验缓存的本地文件状态
- dataMapa,err := etcdclient.QueryWithPrefix(keyPrefix)
- if err != nil {
- log.Println(err)
- }
- if dataMapa!=nil && len(dataMapa)>0{
- for k,_ := range dataMapa {
- if !fileExist(config.LocalWorkSpaceDir+k){
- err = etcdclient.DeleteWithPrefix(k)
- if err != nil {
- log.Println(err)
- }
- }
- }
- }
-
-
- //优先etcd查询
- dataMap,err := etcdclient.QueryWithPrefix(keyPrefix)
- if err != nil {
- log.Println(err)
- return err
- }
- if dataMap==nil || len(dataMap)==0{
- // 不存在则初始化进etcd
-
- err =filepath.Walk(getLocalFileListDir,walkfunc)
- //路径错误
- if err != nil {
- log.Println(err)
- if err := conn.WriteMessage(websocket.TextMessage, []byte("{}")); err != nil {
- log.Println(err)
- return err
- }
- }
-
- mapByte,err:=json.Marshal(gobalFileMap)
- if err != nil {
- log.Println(err)
- return err
- }
-
- if err := conn.WriteMessage(websocket.TextMessage, mapByte); err != nil {
- log.Println(err)
- return err
- }
-
- cacheMap := make(map[string] string)
- for k,v := range gobalFileMap {
- k := strings.Replace(k,config.LocalWorkSpaceDir,"",1)
- cacheMap[k]=v
-
- }
-
- //异步缓存
- //go func() {
- err = etcdclient.BatchAdd(cacheMap)
- if err != nil {
- log.Println(err)
- }
- //}()
-
- //log.Println("检查本地目录文件变更状态,执行睡眠时间1分钟")
-
- //清空gobalFileMap
- gobalFileMap = make(map[string] string)
- }
-
- err=sendFileListFromEtcd(keyPrefix,projectName,conn)
- if err != nil {
- log.Println(err)
- return err
- }
-
- //ch :=config.GobalWatchChannelMap[getLocalFileListDir]
- log.Println(config.GobalWatchChannelMap[getLocalFileListDir])
- for actionAndModifyFilePathStr :=range config.GobalWatchChannelMap[getLocalFileListDir] {
- log.Println(actionAndModifyFilePathStr)
- actionAndModifyFilePath := strings.Split(actionAndModifyFilePathStr,";")
- if actionAndModifyFilePath[0]=="remove"{
- k := strings.Replace(actionAndModifyFilePath[1],config.LocalWorkSpaceDir,"",1)
- queryMap,err :=etcdclient.QueryWithPrefix(k)
- if len(queryMap)==0{
- continue
- }
- err = etcdclient.DeleteWithPrefix(k)
- if err != nil {
- log.Println(err)
- }
- }else if actionAndModifyFilePath[0]=="write"{
- queryKey := strings.Replace(actionAndModifyFilePath[1],config.LocalWorkSpaceDir,"",1)
- querymap,err := etcdclient.QueryWithPrefix(queryKey)
- if err != nil {
- log.Println(err)
- continue
- }
- if len(querymap)==0{
- continue
- }
-
- oldValue := strings.Split(querymap[queryKey],";")
- newValue := oldValue[0]+";" +"1"
- err = etcdclient.ReplaceInto(queryKey,newValue)
- if err!=nil{
- log.Println(err)
- continue
- }
- }else if actionAndModifyFilePath[0]=="create"{
- queryKey := strings.Replace(actionAndModifyFilePath[1],config.LocalWorkSpaceDir,"",1)
- querymap,err := etcdclient.QueryWithPrefix(queryKey)
- if err != nil {
- log.Println(err)
- continue
- }
- if len(querymap)==0{
- continue
- }
-
- oldValue := strings.Split(querymap[queryKey],";")
- newValue := oldValue[0]+";" +"1"
- err = etcdclient.ReplaceInto(queryKey,newValue)
- if err!=nil{
- log.Println(err)
- continue
- }
- }
- err = sendFileListFromEtcd(keyPrefix,projectName,conn)
- if err != nil {
- log.Println(err)
- return err
- }
- }
-
- return nil
- }
-
- func sendFileListFromEtcd(keyPrefix,projectName string,conn *websocket.Conn) error{
- dataMap,err := etcdclient.QueryWithPrefix(keyPrefix)
- if err != nil {
- log.Println(err)
- return err
- }
- if dataMap!=nil && len(dataMap)>0{
-
- cacheMap := make(map[string] string)
- for k,v := range dataMap {
- //历史数据加默认值
- if len(strings.Split(v, ";"))==1{
- v=v+";0"
- }
- cacheMap[strings.Replace(k,gobalLoginUserName+"\\"+projectName+"\\","",1)]=v
- }
-
- mapByte,err:=json.Marshal(cacheMap)
- if err != nil {
- log.Println(err)
- return err
- }
-
- if err := conn.WriteMessage(websocket.TextMessage, mapByte); err != nil {
- log.Println(err)
- return err
- }
- return nil
- }else{
- log.Println("未查询到数据,keyPrefix:"+keyPrefix+",projectName:"+projectName)
- }
- return nil
- }
-
-
- /**
- 打开方式
- */
- func OpenFileWith(filePath string) error{
-
- //判断文件有效性
- _,err := os.Stat(filePath)
- if err!=nil{
- return err
- }
-
- //filePath = strings.Replace(filePath," ","~1",1)
-
- cmd := exec.Command("rundll32.exe","shell32.dll,OpenAs_RunDLL",filePath);
-
- err =cmd.Run()
- if err!=nil{
- log.Println(err)
- return err
- }
- return nil
- }
-
- /**
- 手动检查软件更新
- 0:不强制更新
- 1:强制更新
- */
- func CheckForUpdates(forceUpdate string) error{
-
- tszdir :=os.Getenv("TSZDIR")
-
- //空格路径处理
- ipfsPath=strings.Replace(os.Getenv("IPFS-PATH"),"\"","",1)+"\\stop.vbs"
-
- //判断文件有效性
- _,err := os.Stat(tszdir+config.UpdaterName)
- if err!=nil{
- return err
- }
-
- cmd := exec.Command(tszdir+config.UpdaterName,"/justcheck");
- err =cmd.Run()
- if err!=nil{
- log.Println(err)
- return err
- }
-
- cmd = exec.Command(tszdir+config.UpdaterName,"/checknow");
- err =cmd.Run()
- if err!=nil{
- log.Println(err)
- return err
- }
-
- //检测到更新 kill所有客户端进程
- log.Println("close all process")
- cmd = exec.Command("cmd.exe","/c",ipfsPath);
- err =cmd.Run()
- if err!=nil{
- log.Println(err)
- return err
- }
- return nil
-
- }
-
- func walkfunc(filePath string, info os.FileInfo, err error) error {
-
- if info == nil{
- return nil
- }
-
- if info.IsDir()==false{
- //历史文件不扫描
- if path.Ext(filePath)==".commit" {
- return nil
- }
-
- sh := shell.NewShell(config.GobalIpfsUrl)
- file,err :=os.Open(filePath)
-
-
- if err != nil{
- log.Println(err)
- return err
- }
-
- defer file.Close()
-
- hash,err :=sh.Add(file)
- if err != nil {
- log.Println(err)
- return err
- }
-
- dir :=strings.Replace(fmt.Sprint(filePath),fmt.Sprint(getLocalFileListDir+"\\"),"",1)
-
- gobalFileMap[dir]=hash
-
- }
- return nil
- }
-
- /**
- 查询历史文件
- path 文件路径
- hash 历史版本文件hash
- */
- func QueryCommitHistory(filePath,hash string) (map[int] *commitHistory,error){
-
- result := make(map[int] *commitHistory)
- //校验文件路径
- _,err :=os.Stat(filePath)
- if err != nil {
- log.Println("文件 "+filePath+"not found")
- return nil,errors.New("参数错误!")
- }
-
- if len(hash) == 0 {
- return result,nil
- }
-
- //根据hash更新文件
- localSh := shell.NewShell(config.GobalIpfsUrl)
- //连接失败判断
- localSh.SetTimeout(5*time.Second)
- ext :=path.Ext(filePath)
- commitFilePath :=strings.Replace(filePath,ext,".commit",1)
- os.Remove(commitFilePath)
- err = localSh.Get(hash,commitFilePath)
- if err!=nil {
- log.Println("文件"+hash+"下载失败")
-
- return result,errors.New("历史文件获取失败,请稍后重试")
- }
-
- //设置文件隐藏属性
- attribCmd :=exec.Command("attrib","+h",commitFilePath)
- err =attribCmd.Run()
- if err != nil {
- log.Println("设置文件隐藏属性失败")
- return result,err
- }
-
- //解析历史版本文件
- contentByte,err := ioutil.ReadFile(commitFilePath)
- content := string(contentByte)
- if content==""{
- return result,nil
- }
- rows :=strings.Split(content,"\n")
- length := len(rows)
- var index int = 0
- for i:=length-2;i>=0;i--{
- columns := strings.Split(rows[i],"\t")
- commitHistoryInstance := new(commitHistory)
- commitHistoryInstance.ParentHash =columns[0]
- commitHistoryInstance.CurrentHash=columns[1]
- commitHistoryInstance.Version,_=strconv.Atoi(columns[6])
- commitHistoryInstance.Milestone,_=strconv.ParseBool(columns[5])
- commitHistoryInstance.Creator=columns[2]
- commitHistoryInstance.CreateTime,_=strconv.ParseInt(columns[3], 10, 64)
- commitHistoryInstance.Note = columns[4]
- result[index]=commitHistoryInstance
- index++
- }
- return result,nil
- }
-
- /**
- 设定某个历史版本为里程碑版本
- @param filePath 文件绝对路径
- @param commitHistoryHash 历史版本管理文件hash
- @param hash 文件hash
- @param milestone 是否是里程碑
-
- */
- func EditCommitHistoryMilestoneHandler(filePath,commitHistoryHash,hash string,milestone bool) (string,error){
-
- //result := make(map[int] *commitHistory)
- //校验文件路径
- _,err :=os.Stat(filePath)
- if err != nil {
- log.Println("文件 "+filePath+"not found")
- return "",errors.New("参数错误!")
- }
-
- if len(commitHistoryHash) == 0 {
- log.Println("参数hash must not empty")
- return "",errors.New("参数错误!")
- }
-
-
- //根据hash更新文件
- localSh := shell.NewShell(config.GobalIpfsUrl)
- //连接失败判断
- localSh.SetTimeout(5*time.Second)
- ext :=path.Ext(filePath)
- commitFilePath :=strings.Replace(filePath,ext,".commit",1)
- os.Remove(commitFilePath)
- err = localSh.Get(commitHistoryHash,commitFilePath)
- if err!=nil {
- log.Println("文件"+commitHistoryHash+"下载失败")
- return "",errors.New("历史文件获取失败,请稍后重试")
- }
-
- //设置文件隐藏属性
- attribCmd :=exec.Command("attrib","+h",commitFilePath)
- err =attribCmd.Run()
- if err != nil {
- log.Println("设置文件隐藏属性失败")
- return "",err
- }
-
- //解析历史版本文件
- contentByte,err := ioutil.ReadFile(commitFilePath)
- content := string(contentByte)
- if content==""{
- return "",nil
- }
- rows :=strings.Split(content,"\n")
- length := len(rows)
- resultString :=""
- for i:=0;i<length-1;i++{
- columns := strings.Split(rows[i],"\t")
- if columns[1]==hash{
- resultString=resultString+fmt.Sprintf("%v\t%v\t%v\t%v\t%v\t%v\t%v\t\n",columns[0],columns[1],columns[2],columns[3],columns[4],milestone,columns[6])
- continue
- }
- resultString=resultString+fmt.Sprintf("%v\t%v\t%v\t%v\t%v\t%v\t%v\t\n",columns[0],columns[1],columns[2],columns[3],columns[4],columns[5],columns[6])
-
- }
-
- os.Remove(commitFilePath)
- fw, err := os.OpenFile(commitFilePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)//os.O_TRUNC清空文件重新写入,否则原文件内容可能残留
- w := bufio.NewWriter(fw)
- w.WriteString(resultString)
- if err != nil {
- log.Println(err)
- return "",err
- }
- w.Flush()
- fw.Close()
-
- addFile,err :=os.Open(commitFilePath)
- if err != nil{
- log.Println(err)
- return "",err
- }
- defer addFile.Close()
- //add 历史管理文件
- historyHash,err:=localSh.Add(addFile)
- if err != nil {
- log.Println("历史版本管理文件上传失败")
- return "",err
- }
-
- serverSh :=shell.NewShell(config.ServerIpfsUrl)
- serverSh.SetTimeout(5*time.Second)
- err = serverSh.Pin(historyHash)
- if err != nil {
- log.Println("历史版本管理文件备份失败")
- return "",err
- }
-
- return historyHash,nil
- }
-
- /*
- 提交历史
- */
- type commitHistory struct {
- ParentHash string `json:"parentHash"`
- CurrentHash string `json:"currentHash"`
- Creator string `json:"creator"`
- CreateTime int64 `json:"createTime"`
- Note string `json:"note"`
- Version int `json:"version"`
- Milestone bool `json:"milestone"`
- }
-
-
-
-
-
-
|