Go语言实现工时计算

package mainimport "flag"import "fmt"import "bufio"import "os"import "strconv"import "strings"var infile *string = flag.String("i", "before.txt", "Work hours to convert")var outfile *string = flag.String("o", "after.txt", "Work hours converted")// 声明一个变量，用来临时记录新一行的数据var newLine string// 声明一个slice，用他来保存所有新行数据var newLines []string/** * 请手动去掉有请假或迟到、早退的行（这些情况需要特殊处理） * 计算正常上下班，当天没有请假、早退情况的上班时间（单位：小时），保留到小数点后面两位数字。 * 计算之前的文件请命名为before.txt，放在该代码同级目录下。 * go语言排序 */func main() {    // 初始化保存新行数据的slice，由于每个月最多31天，所以给它分配初始大小为32，多一行打印平均时间    newLines = make([]string, 32)    flag.Parse()    if infile != nil {        fmt.Println("infile=", *infile, "outfile=", *outfile)    }    // 打开未处理的文件    file, err := os.Open(*infile)    if err != nil {        fmt.Println("打开文件失败", infile)        return    }    defer file.Close()    br := bufio.NewReader(file)    sumHour := 0.0    index := 0    for {        // 每次读取一行，都让新行归位为空        newLine = ""        // 读取一行数据        line, isPrefix, readLineErr := br.ReadLine()        if readLineErr != nil {            // 已经读完了            break        }        if isPrefix {            // 行太长了，无法处理，退出程序            fmt.Println("行太长了")            return        }        // 字符数组转为字符串        str := string(line)        // 按制表符分割一行数据，第0个字符串为日期，第1个数据为上班时间，第2个数据为下班        strs := strings.Split(str, "\t")        // 让新行先记录下日期        newLine = strs[0] + " "        // 处理上班时间，根据冒号进行分割，前者为小时，后者为分钟        start := strings.Split(strs[1], ":")        startHour, startHourErr := strconv.Atoi(start[0])        if startHourErr != nil {            // 转换上班小时的时候出错了            fmt.Println("转换上班的小时出现错误")            return        }        startMinute, startMinuteErr := strconv.Atoi(start[1])        if startMinuteErr != nil {            // 转换上班分钟的时候出错了            fmt.Println("转换上班的分钟出现错误")            return        }        // 处理下班时间，根据冒号进行分割，前者为小时，后者为分钟        end := strings.Split(strs[2], ":")        endHour, endHourErr := strconv.Atoi(end[0])        if endHourErr != nil {            // 转换下班小时的时候出错了            fmt.Println("转换下班的小时出现错误")            return        }        endMinute, endMinuteErr := strconv.Atoi(end[1])        if endMinuteErr != nil {            // 转换下班分钟的时候出错了            fmt.Println("转换下班的分钟出现错误")            return        }        // 过滤不符合要求的时间（有迟到）        if startHour > 9 {            // 迟到            fmt.Println("请去除迟到的时间")            return        }        // 过滤不符合要求的时间（有早退）        if endHour == 17 && endMinute < 30 || endHour < 17 {            // 早退            fmt.Println("请去除早退的时间")            return        }        // 小数形式的每天上班时间        currentDayHour := 0.0        // 上班时间转换为小数形式        startFloat := float64(float64(startHour) + float64(startMinute)/60)        // 小数形式的下班时间        endFloat := 0.0        // 由于5:30到6:00是吃饭时间，不计入工时，所以要分情况处理        if endHour < 18 {            // 处于下午5点半到6点之间，相当于17:30下班，也就是下班点为17.5            endFloat = 17.5        } else {            // 6点之后下班，直接减去30/60小时            endFloat = float64(float64(endHour) + float64(endMinute)/60 - 0.5)        }        // 减去1.5小时的中午休息时间就是当天上班时间        currentDayHour = endFloat - startFloat - 1.5        sumHour += currentDayHour        // 保存新一行数据到slice中currentDayHour        newLines[index] = newLine + strconv.FormatFloat(currentDayHour, 'f', 2, 64)        index++    }    average := sumHour / float64(index)    newLines[index] = "平均每天上班时间：" + strconv.FormatFloat(average, 'f', 2, 64) + "小时"    // 保存文件    afterFile, afterErr := os.Create(*outfile)    if afterErr != nil {        fmt.Println("创建输出文件失败")        return    }    defer afterFile.Close()    // 将slice倒转过来输出，让时间从小到大    for i := len(newLines) - 1; i >= 0; i-- {        afterFile.WriteString(newLines[i] + "\r\n")    }}