/
command_file_head.go
70 lines (57 loc) · 1.37 KB
/
command_file_head.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package beans
import (
"bufio"
"fmt"
"os"
"strconv"
"strings"
)
/********************************************************************
created: 2020-07-26
author: lixianmin
Copyright (C) - All Rights Reserved
*********************************************************************/
func ReadFileHead(note string, texts []string, maxNum int) string {
if len(texts) == 1 {
return note
}
var args, err = parseReadFileArgs(texts, maxNum)
if err != nil {
return note
}
var lines = readHeadLines(args)
var message = fmt.Sprintf("<br> 返回行数:%d <br>", len(lines)) + strings.Join(lines, "<br>")
return message
}
func readHeadLines(args ReadFileArgs) []string {
var fin, err = os.Open(args.FullPath)
if err != nil {
return nil
}
defer fin.Close()
var reader = bufio.NewReader(fin)
var lines = make([]string, 0, args.Num)
var skipLines = args.StartLine - 1
for i := 0; i < skipLines; i++ {
var _, err = reader.ReadString('\n')
if err != nil {
return lines
}
}
var filter = strings.ToLower(args.Filter)
var lineNum = skipLines
var counter = 0
for counter < args.Num {
var line, err = reader.ReadString('\n')
if err != nil {
break
}
lineNum += 1
if filter == "" || strings.Contains(strings.ToLower(line), filter) {
var item = strconv.Itoa(lineNum) + " " + line
lines = append(lines, item)
counter += 1
}
}
return lines
}