workhour

package main

import "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")
	}
}